jax
080294cf - Load CUDA libraries up front with cdll.LoadLibrary().

Commit
1 year ago
Load CUDA libraries up front with cdll.LoadLibrary(). Works around a bug in CUDNN where it does not declare or locate its own NVRTC dependency. However, this approach should also improve our error messages in general when we load NVIDIA's libraries, so do it across the board for all libraries we use.
Author
Committer
Parents
Loading