Skip to content

Commit 172e340

Browse files
pass a few more linker tests
1 parent ab9ac6f commit 172e340

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

numba_cuda/numba/cuda/cudadrv/driver.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2806,7 +2806,6 @@ def add_cu(self, cu, name='<cudapy-cu>'):
28062806
numba_cuda_path = os.path.dirname(cudadrv_path)
28072807

28082808
nrt_path = os.path.join(numba_cuda_path, "runtime")
2809-
28102809
include_paths = cuda_include + [numba_cuda_path, nrt_path]
28112810
prog = Program(
28122811
cu.decode('utf-8'),

numba_cuda/numba/cuda/tests/cudadrv/test_linker.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from numba.cuda.testing import CUDATestCase, test_data_dir
66
from numba.cuda.cudadrv.driver import (CudaAPIError, Linker,
77
LinkerError)
8-
from numba.cuda.cudadrv.error import NvrtcError
8+
from cuda.core.experimental._utils import NVRTCError
99
from numba.cuda import require_context
1010
from numba.tests.support import ignore_internal_warnings
1111
from numba import cuda, void, float64, int64, int32, typeof, float32
@@ -187,14 +187,14 @@ def test_linking_cu_error(self):
187187

188188
link = str(test_data_dir / 'error.cu')
189189

190-
with self.assertRaises(NvrtcError) as e:
190+
with self.assertRaises(NVRTCError) as e:
191191
@cuda.jit('void(int32)', link=[link])
192192
def kernel(x):
193193
bar(x)
194194

195195
msg = e.exception.args[0]
196196
# Check the error message refers to the NVRTC compile
197-
self.assertIn('NVRTC Compilation failure', msg)
197+
self.assertIn('NVRTC_ERROR_COMPILATION', msg)
198198
# Check the expected error in the CUDA source is reported
199199
self.assertIn('identifier "SYNTAX" is undefined', msg)
200200
# Check the filename is reported correctly

0 commit comments

Comments
 (0)