Are there contractions simply too large to work on given memory ? #167
Unanswered
0livier1O1
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I am trying to perform a contraction on a relatively large TN but I get a
MemoryLimitExceededwhen callingcontract_path. From my understanding in this discussion, slicing should enable larger contractions but even after trying different settings foroptimize_optionsI can't get the contraction to fit on memory. Are there contractions that simply need more GPU memory and cannot be sliced sufficiently small to fit?This is the error
and the runnable file to reproduce it can be found here: https://github.com/0livier1O1/tff/blob/move_to_gpu/tn_contraction.py
I am quite new to GPUs and TN in general.
Beta Was this translation helpful? Give feedback.
All reactions