From 95eb2951a3b2764b2c640c2bf39038a813abd59b Mon Sep 17 00:00:00 2001 From: brandon-b-miller Date: Fri, 22 Aug 2025 07:20:42 -0700 Subject: [PATCH 1/3] add dep --- pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/pyproject.toml b/pyproject.toml index b7c66dd32..c5798dd20 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,6 +32,7 @@ cu12 = [ "nvidia-cuda-nvcc-cu12", # for libNVVM "nvidia-cuda-runtime-cu12", "nvidia-cuda-nvrtc-cu12", + "nvidia-cuda-cccl-cu12", ] test = [ "psutil", From daaa4f2a5d2db4d4bee8b584a72471ca74f19daf Mon Sep 17 00:00:00 2001 From: brandon-b-miller Date: Fri, 22 Aug 2025 08:09:29 -0700 Subject: [PATCH 2/3] CI fixes --- .../numba/cuda/tests/cudapy/test_inspect.py | 23 +++++++++++++++++++ .../generate_raw_ltoir.py | 2 +- 2 files changed, 24 insertions(+), 1 deletion(-) diff --git a/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py b/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py index 5a038a11c..efd96a2f5 100644 --- a/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py +++ b/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py @@ -10,12 +10,27 @@ ) +def skip_on_cuda_version_issues(self): + # FIXME: This should be unskipped once the cause of certain nvdisasm + # versions failing to dump SASS with certain driver / nvJitLink + # versions is understood + self.skipTest("Relocation information required for analysis not preserved") + + @skip_on_cudasim("Simulator does not generate code to be inspected") class TestInspect(CUDATestCase): @property def cc(self): return cuda.current_context().device.compute_capability + def skip_on_cuda_version_issues(self): + # FIXME: This should be unskipped once the cause of certain nvdisasm + # versions failing to dump SASS with certain driver / nvJitLink + # versions is understood + self.skipTest( + "Relocation information required for analysis not preserved" + ) + def test_monotyped(self): sig = (float32, int32) @@ -110,6 +125,8 @@ def _test_inspect_sass(self, kernel, name, sass): @skip_without_nvdisasm("nvdisasm needed for inspect_sass()") def test_inspect_sass_eager(self): + self.skip_on_cuda_version_issues() + sig = (float32[::1], int32[::1]) @cuda.jit(sig, lineinfo=True) @@ -122,6 +139,8 @@ def add(x, y): @skip_without_nvdisasm("nvdisasm needed for inspect_sass()") def test_inspect_sass_lazy(self): + self.skip_on_cuda_version_issues() + @cuda.jit(lineinfo=True) def add(x, y): i = cuda.grid(1) @@ -139,6 +158,8 @@ def add(x, y): "Missing nvdisasm exception only generated when it is not present" ) def test_inspect_sass_nvdisasm_missing(self): + self.skip_on_cuda_version_issues() + @cuda.jit((float32[::1],)) def f(x): x[0] = 0 @@ -150,6 +171,8 @@ def f(x): @skip_without_nvdisasm("nvdisasm needed for inspect_sass_cfg()") def test_inspect_sass_cfg(self): + self.skip_on_cuda_version_issues() + sig = (float32[::1], int32[::1]) @cuda.jit(sig) diff --git a/numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py b/numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py index 18959945f..c260b53e1 100644 --- a/numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py +++ b/numba_cuda/numba/cuda/tests/test_binary_generation/generate_raw_ltoir.py @@ -6,7 +6,7 @@ import subprocess import sys -from cuda import nvrtc +from cuda.bindings import nvrtc from numba.cuda.memory_management.nrt import get_include # Magic number found at the start of an LTO-IR file From ab60841c1cdb833cabc7451c2f23eb8218b739b5 Mon Sep 17 00:00:00 2001 From: brandon-b-miller Date: Fri, 22 Aug 2025 08:17:13 -0700 Subject: [PATCH 3/3] clean --- numba_cuda/numba/cuda/tests/cudapy/test_inspect.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py b/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py index efd96a2f5..30bd31bc4 100644 --- a/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py +++ b/numba_cuda/numba/cuda/tests/cudapy/test_inspect.py @@ -10,13 +10,6 @@ ) -def skip_on_cuda_version_issues(self): - # FIXME: This should be unskipped once the cause of certain nvdisasm - # versions failing to dump SASS with certain driver / nvJitLink - # versions is understood - self.skipTest("Relocation information required for analysis not preserved") - - @skip_on_cudasim("Simulator does not generate code to be inspected") class TestInspect(CUDATestCase): @property