Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix cuTensor, cuTensorNet and cuStateVec when using local Toolkit #2274

Merged
merged 1 commit into from
Feb 23, 2024

Conversation

bjoe2k4
Copy link
Contributor

@bjoe2k4 bjoe2k4 commented Feb 23, 2024

Currently when trying to use cuTENSOR with a local version of the CUDA toolkit, this fails with:

ERROR: LoadError: InitError: UndefVarError: `CUDA_Runtime` not defined

due to #2058 being incomplete.

Versions used:

  • Julia 1.10.1
  • Cuda Toolkit v12.3.2 loaded via Lmod on a HPC cluster

There are some more occurances of CUDA_Runtime in CUDA.jl, but I haven't had any errors due to these: https://github.com/search?q=repo%3AJuliaGPU%2FCUDA.jl%20CUDA_Runtime.&type=code

I have tested this PR by using cuTENSOR (also locally provided as Lmod module), and fixed cuTensorNet (currently not compatible with cuTensor v2?) and cuStateVec due to similarity without testing.

@maleadt
Copy link
Member

maleadt commented Feb 23, 2024

Thanks. The original idea was to conditionally define CUDA_Runtime, as is done in CUDA.jl itself, but apparently I forgot about that in the subpackages.

@maleadt
Copy link
Member

maleadt commented Feb 23, 2024

cuTensorNet (currently not compatible with cuTensor v2?)

That's correct; there's been no release (by NVIDIA) yet of a cuQuantum compatible with CUTENSOR v2. The JLLs should encode those compatibility constraints, but in your case you'll have to manage that yourself.

@maleadt maleadt merged commit 7a9a0a2 into JuliaGPU:master Feb 23, 2024
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants