Skip to content

Commit 5b76f65

Browse files
authored
Add missing error handling (cuda_bindings/tests/test_nvjitlink.py) (#689)
1 parent 884fff3 commit 5b76f65

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

cuda_bindings/tests/test_nvjitlink.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,13 @@ def check_nvjitlink_usable():
6666
def get_dummy_ltoir():
6767
def CHECK_NVRTC(err):
6868
if err != nvrtc.nvrtcResult.NVRTC_SUCCESS:
69-
raise RuntimeError(f"Nvrtc Error: {err}")
69+
raise RuntimeError(repr(err))
7070

7171
empty_cplusplus_kernel = "__global__ void A() {}"
7272
err, program_handle = nvrtc.nvrtcCreateProgram(empty_cplusplus_kernel.encode(), b"", 0, [], [])
7373
CHECK_NVRTC(err)
74-
nvrtc.nvrtcCompileProgram(program_handle, 1, [b"-dlto"])
74+
err = nvrtc.nvrtcCompileProgram(program_handle, 1, [b"-dlto"])[0]
75+
CHECK_NVRTC(err)
7576
err, size = nvrtc.nvrtcGetLTOIRSize(program_handle)
7677
CHECK_NVRTC(err)
7778
empty_kernel_ltoir = b" " * size

0 commit comments

Comments
 (0)