From 1acf13fb1b7763c3f3f2f2bd1fff73f717e02938 Mon Sep 17 00:00:00 2001 From: yangg Date: Wed, 15 Jul 2026 10:15:48 -0700 Subject: [PATCH 1/2] NEW: add support on cutensorMp library loading --- .../cuda/pathfinder/_dynamic_libs/descriptor_catalog.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index de57ff23ad6..6220bc76e6d 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -375,6 +375,13 @@ class DescriptorSpec: site_packages_windows=("cutensor/bin",), dependencies=("cutensor", "cublasLt"), ), + DescriptorSpec( + name="cutensorMp", + packaged_with="other", + linux_sonames=("libcutensorMp.so.2",), + site_packages_linux=("cutensor/lib",), + dependencies=("cutensor", "cublasLt", "nccl"), + ), DescriptorSpec( name="nccl", packaged_with="other", From cf47da719b26bd8cb22d14896b92b8b51b952a56 Mon Sep 17 00:00:00 2001 From: "Ralf W. Grosse-Kunstleve" Date: Wed, 15 Jul 2026 14:23:06 -0700 Subject: [PATCH 2/2] 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. --- .../cuda/pathfinder/_dynamic_libs/descriptor_catalog.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py index 6220bc76e6d..67e4be96d07 100644 --- a/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py +++ b/cuda_pathfinder/cuda/pathfinder/_dynamic_libs/descriptor_catalog.py @@ -380,7 +380,7 @@ class DescriptorSpec: 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",