From 2b21582c7e8fd52624c354bb8f1572feba6b95e8 Mon Sep 17 00:00:00 2001 From: Michael Wang <13521008+isVoid@users.noreply.github.com> Date: Fri, 7 Aug 2026 15:46:14 -0700 Subject: [PATCH 1/2] Fix CUDA Runtime CTK test fixture on Windows ARM64 --- cuda_pathfinder/tests/test_ctk_root_discovery.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cuda_pathfinder/tests/test_ctk_root_discovery.py b/cuda_pathfinder/tests/test_ctk_root_discovery.py index f232afe7719..6f42f9f8006 100644 --- a/cuda_pathfinder/tests/test_ctk_root_discovery.py +++ b/cuda_pathfinder/tests/test_ctk_root_discovery.py @@ -81,9 +81,9 @@ def _create_nvvm_in_ctk(ctk_root): def _create_cudart_in_ctk(ctk_root): """Create a fake cudart lib in the platform-appropriate CTK subdirectory.""" if IS_WINDOWS: - lib_dir = ctk_root / "bin" + lib_dir = ctk_root / "bin" / windows_python_arch() lib_dir.mkdir(parents=True) - lib_file = lib_dir / "cudart64_12.dll" + lib_file = lib_dir / "cudart64_13.dll" else: lib_dir = ctk_root / "lib64" lib_dir.mkdir(parents=True) From 462c571d6f0b4c34782291af791809ab83e53fa2 Mon Sep 17 00:00:00 2001 From: Michael Wang <13521008+isVoid@users.noreply.github.com> Date: Fri, 7 Aug 2026 15:54:22 -0700 Subject: [PATCH 2/2] Make Windows CUDA Runtime fixtures explicit --- cuda_pathfinder/tests/test_ctk_root_discovery.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/cuda_pathfinder/tests/test_ctk_root_discovery.py b/cuda_pathfinder/tests/test_ctk_root_discovery.py index 6f42f9f8006..5a1c082e715 100644 --- a/cuda_pathfinder/tests/test_ctk_root_discovery.py +++ b/cuda_pathfinder/tests/test_ctk_root_discovery.py @@ -33,7 +33,7 @@ DYNAMIC_LIB_SUBPROCESS_MODULE, MODE_CANARY, ) -from cuda.pathfinder._utils.platform_aware import IS_WINDOWS +from cuda.pathfinder._utils.platform_aware import IS_WINDOWS, IS_WINDOWS_ARM64, IS_WINDOWS_X64 from cuda.pathfinder._utils.windows_arch import windows_python_arch _MODULE = "cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_lib" @@ -80,14 +80,16 @@ def _create_nvvm_in_ctk(ctk_root): def _create_cudart_in_ctk(ctk_root): """Create a fake cudart lib in the platform-appropriate CTK subdirectory.""" - if IS_WINDOWS: - lib_dir = ctk_root / "bin" / windows_python_arch() - lib_dir.mkdir(parents=True) + if IS_WINDOWS_X64: + lib_dir = ctk_root / "bin" / "x64" + lib_file = lib_dir / "cudart64_13.dll" + elif IS_WINDOWS_ARM64: + lib_dir = ctk_root / "bin" / "arm64" lib_file = lib_dir / "cudart64_13.dll" else: lib_dir = ctk_root / "lib64" - lib_dir.mkdir(parents=True) lib_file = lib_dir / "libcudart.so" + lib_dir.mkdir(parents=True) lib_file.write_bytes(b"fake") return lib_file