From 6dd3ecbc1b15f0468ea0efc514c64bf9b7356b3c Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Fri, 17 Oct 2025 08:26:51 +0000 Subject: [PATCH 1/4] Initial plan From 2377bc8f8dcefb60cfb1dfbee0a669aa2011048d Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Fri, 17 Oct 2025 08:33:36 +0000 Subject: [PATCH 2/4] Enable cuTENSOR preload for CUDA 13.x packages Co-authored-by: kmaehashi <939877+kmaehashi@users.noreply.github.com> --- dist_config.py | 6 +++--- release-tests/pkg_wheel/test_preload.py | 3 --- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/dist_config.py b/dist_config.py index 93c7e1b..9e2e589 100644 --- a/dist_config.py +++ b/dist_config.py @@ -180,7 +180,7 @@ class _WheelLinuxConfig(TypedDict): 'image': 'cupy/cupy-release-tools:cuda-runfile-13.0.0-centos7', 'libs': [], 'includes': [], - 'preloads': ['nccl'], # TODO(kmaehashi): Re-add cuTENSOR + 'preloads': ['cutensor', 'nccl'], 'verify_image': 'nvidia/cuda:{system}', 'verify_systems': [ # Test on all supported CUDA version variants. @@ -198,7 +198,7 @@ class _WheelLinuxConfig(TypedDict): 'image': 'cupy/cupy-release-tools:cuda-runfile-13.0.0-el8', 'libs': [], 'includes': [], - 'preloads': ['nccl'], + 'preloads': ['cutensor', 'nccl'], 'preloads_cuda_version': '13.x', 'builder_dockerfile': 'Dockerfile.el8', 'verify_image': 'nvidia/cuda:{system}', @@ -274,7 +274,7 @@ class _WheelWindowsConfig(TypedDict): 'name': 'cupy-cuda13x', 'kind': 'cuda', 'libs': [], - 'preloads': [], # TODO(kmaehashi): Re-add cuTENSOR + 'preloads': ['cutensor'], 'cudart_lib': 'cudart64_13', # binary compatible between CUDA 13.x 'check_version': lambda x: 13000 <= x < 13010, # CUDA 13.0 } diff --git a/release-tests/pkg_wheel/test_preload.py b/release-tests/pkg_wheel/test_preload.py index 2d746ff..618e84f 100644 --- a/release-tests/pkg_wheel/test_preload.py +++ b/release-tests/pkg_wheel/test_preload.py @@ -34,9 +34,6 @@ def test_cutensor(self): if cupy.cuda.runtime.runtimeGetVersion() < 10010: # cuTENSOR is only available for CUDA 10.1+. return - if 13000 <= cupy.cuda.runtime.runtimeGetVersion(): - # TODO(kmaehashi): cuTENSOR is not yet available in CUDA 13+. - return preload_version = self._get_config()['cutensor']['version'] major, minor, patchlevel = (int(x) for x in preload_version.split('.')) expected_version = major * 10000 + minor * 100 + patchlevel From e14155a905a7f0a8b11e3d1f351cd46b77aff2ae Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Fri, 17 Oct 2025 08:36:21 +0000 Subject: [PATCH 3/4] Correct aarch64 config to follow existing pattern Co-authored-by: kmaehashi <939877+kmaehashi@users.noreply.github.com> --- dist_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dist_config.py b/dist_config.py index 9e2e589..8423537 100644 --- a/dist_config.py +++ b/dist_config.py @@ -198,7 +198,7 @@ class _WheelLinuxConfig(TypedDict): 'image': 'cupy/cupy-release-tools:cuda-runfile-13.0.0-el8', 'libs': [], 'includes': [], - 'preloads': ['cutensor', 'nccl'], + 'preloads': ['nccl'], 'preloads_cuda_version': '13.x', 'builder_dockerfile': 'Dockerfile.el8', 'verify_image': 'nvidia/cuda:{system}', From 2742d2a27e42fc8a0db968ee71ed6a75b01c2eb4 Mon Sep 17 00:00:00 2001 From: Kenichi Maehashi Date: Tue, 25 Nov 2025 13:52:52 +0000 Subject: [PATCH 4/4] update for cuTENSOR 2.3 for Windows new directory structure --- builder/setup_cuda_opt_lib.py | 1 + 1 file changed, 1 insertion(+) diff --git a/builder/setup_cuda_opt_lib.py b/builder/setup_cuda_opt_lib.py index 2555d0c..c9a8e6d 100755 --- a/builder/setup_cuda_opt_lib.py +++ b/builder/setup_cuda_opt_lib.py @@ -98,6 +98,7 @@ def main() -> None: elif sys.platform == 'win32': _install_library( 'cutensor', src_dir, dst_dir, { + 'bin': 'bin', 'lib': 'bin', 'include': 'include', })