Skip to content

Commit a533b15

Browse files
yangcalrwgk
andauthored
Pathfinder: add support on cutensorMp library loading (#2374)
* NEW: add support on cutensorMp library loading * 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. --------- Co-authored-by: Ralf W. Grosse-Kunstleve <rgrossekunst@nvidia.com>
1 parent e2e816a commit a533b15

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -375,6 +375,13 @@ class DescriptorSpec:
375375
site_packages_windows=("cutensor/bin",),
376376
dependencies=("cutensor", "cublasLt"),
377377
),
378+
DescriptorSpec(
379+
name="cutensorMp",
380+
packaged_with="other",
381+
linux_sonames=("libcutensorMp.so.2",),
382+
site_packages_linux=("cutensor/lib",),
383+
dependencies=("cutensor", "cublasLt", "cudart", "nccl"),
384+
),
378385
DescriptorSpec(
379386
name="nccl",
380387
packaged_with="other",

0 commit comments

Comments
 (0)