Skip to content

Commit a81525a

Browse files
committed
Update package check for cccl in device tests
1 parent 765ffd6 commit a81525a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/example_tests/device_tests/test_device_samples.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def test_sample(self, sample):
2424
# are using global memory alignment in the sample.
2525
pytest.skip("Skipping test for cublasdx_device_gemm_performance.py, requires libmathdx >= 0.2.1")
2626
if os.path.basename(sample) == "cublasdx_fp64_emulation.py":
27-
spec = importlib.util.find_spec("cuda.cooperative")
27+
spec = importlib.util.find_spec("cuda.cccl.cooperative")
2828
if spec is None:
2929
pytest.skip("Skipping test for cublasdx_fp64_emulation.py, requires cuda.cooperative module")
3030
run_sample(samples_path, sample, {"__name__": "__main__"})

0 commit comments

Comments
 (0)