Repository navigation

Commit cf47da7

committed

Fix cuTENSORMp CUDA runtime preload

CUDA 12's libcutensorMp.so.2 directly depends on libcudart.so.12, but its wheel RUNPATH does not include the cuda_runtime package. Pathfinder can therefore find and preload cuTENSOR, cuBLASLt, and NCCL yet still fail to dlopen cuTENSORMp in a wheel-only environment unless libcudart is already available from an ambient toolkit. Add cudart to the descriptor dependency list so Pathfinder resolves the packaged CUDA runtime before opening cuTENSORMp. This makes CUDA 12 wheel-only loading independent of LD_LIBRARY_PATH while preserving CUDA 13 and local-toolkit behavior.

1 parent 75f221c commit cf47da7Copy full SHA for cf47da7

1 file changed

cuda_pathfinder/cuda/pathfinder/_dynamic_libs
- descriptor_catalog.py

`‎cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py‎`

-Original file line number
+Diff line change
         packaged_with="other",
         linux_sonames=("libcutensorMp.so.2",),
         site_packages_linux=("cutensor/lib",),
 -        dependencies=("cutensor", "cublasLt", "nccl"),
 +        dependencies=("cutensor", "cublasLt", "cudart", "nccl"),
     ),
     DescriptorSpec(
         name="nccl",

Comments

(0)