Closed zjwegert closed 3 years ago
Which version of CUDA.jl is this, I've never seen this before, I guess something has changed in CUTENSOR or the way it is wrapped by CUDA.jl
Maybe @maleadt has some time to chime in?
The CUTENSOR handle has changed in size, so this is invalid now: https://github.com/Jutho/TensorOperations.jl/blob/6e9fa69244210565a46edb473f18ddc3174beb5a/src/implementation/cuarray.jl#L202
Turns out the handles didn't need to be zero-initialized anyway, so I just removed that: https://github.com/JuliaGPU/CUDA.jl/commit/bdab0d5363d31ee22a6532a6a806b32ad614110d
That's quick! Thanks, this is very helpful. I will change my custom wrappers based on the way it is done in your recent implementation. Do I have to be worried about backwards compatibility, or will this also work on older versions of CUDA.jl?
Nothing has changed on the side of the C wrappers themselves, only the size of the handle there has changed (in https://github.com/JuliaGPU/CUDA.jl/commit/ab741155990c98ff628bb184cc7af6b73a211089#diff-0cb7d3454209a70f1e260802e7e8c6a979f4b721cf5f9585cbaac8b92256a6e5, for CUTENSOR 1.3), so this should be backwards compatible with your code that's calling those wrappers directly.
Ok, this should be fixed in TensorOperations.jl 3.2.1 (coincidentally the same as the current CUDA version), which is scheduled to be merged in the next round. Feel free to reopen if the problem is not solved completely or resurfaces.
Great! Thank you
Hi, I'm having trouble getting the code working for tensor contractions on a GPU. See below:
produces the error:
Any suggestions?