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
-
I am trying to perform a contraction on a relatively large TN but I get a
MemoryLimitExceeded
when callingcontract_path
. From my understanding in this discussion, slicing should enable larger contractions but even after trying different settings foroptimize_options
I 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