From 765ffd6a188e5e960fb0ecfc6fcab0f1c5430ecf Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Tue, 10 Jun 2025 14:22:50 +0000 Subject: [PATCH 1/2] update cudax import path --- examples/device/cublasdx_fp64_emulation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/device/cublasdx_fp64_emulation.py b/examples/device/cublasdx_fp64_emulation.py index 81595ed..a5654f9 100644 --- a/examples/device/cublasdx_fp64_emulation.py +++ b/examples/device/cublasdx_fp64_emulation.py @@ -23,7 +23,7 @@ import numpy as np from numba import int32, int8, int16, float64, int64, types from numba.types import Tuple -import cuda.cooperative.experimental as cudax +import cuda.cccl.cooperative.experimental as cudax from common import mm_perf_GFlops, random_real from common_numba import time_numba From a81525ae031f3ad160fccff423ef23376d6f32a9 Mon Sep 17 00:00:00 2001 From: Yevhenii Havrylko Date: Tue, 10 Jun 2025 08:16:47 -0700 Subject: [PATCH 2/2] Update package check for cccl in device tests --- tests/example_tests/device_tests/test_device_samples.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/example_tests/device_tests/test_device_samples.py b/tests/example_tests/device_tests/test_device_samples.py index 375f74c..220435b 100644 --- a/tests/example_tests/device_tests/test_device_samples.py +++ b/tests/example_tests/device_tests/test_device_samples.py @@ -24,7 +24,7 @@ def test_sample(self, sample): # are using global memory alignment in the sample. pytest.skip("Skipping test for cublasdx_device_gemm_performance.py, requires libmathdx >= 0.2.1") if os.path.basename(sample) == "cublasdx_fp64_emulation.py": - spec = importlib.util.find_spec("cuda.cooperative") + spec = importlib.util.find_spec("cuda.cccl.cooperative") if spec is None: pytest.skip("Skipping test for cublasdx_fp64_emulation.py, requires cuda.cooperative module") run_sample(samples_path, sample, {"__name__": "__main__"})