Merge pull request #31 from leofang/cccl

update cudax import path

Author: samaid

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

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__"})