Skip to content

Commit 9e56be8

Browse files
committed
BUG: Handle error return in nvJitLinkGetErrorLogSize and nvJitLinkGetErrorLog.
1 parent edb1901 commit 9e56be8

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

cuda_core/cuda/core/_linker.pyx

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,11 +106,11 @@ cdef class Linker:
106106
cdef char* c_log_ptr
107107
if self._use_nvjitlink:
108108
c_h = as_cu(self._nvjitlink_handle)
109-
cynvjitlink.nvJitLinkGetErrorLogSize(c_h, &c_log_size)
109+
HANDLE_RETURN_NVJITLINK(cynvjitlink.nvJitLinkGetErrorLogSize(c_h, &c_log_size))
110110
log = bytearray(c_log_size)
111111
if c_log_size > 0:
112112
c_log_ptr = <char*>(<bytearray>log)
113-
cynvjitlink.nvJitLinkGetErrorLog(c_h, c_log_ptr)
113+
HANDLE_RETURN_NVJITLINK(cynvjitlink.nvJitLinkGetErrorLog(c_h, c_log_ptr))
114114
return log.decode("utf-8", errors="backslashreplace")
115115
else:
116116
return (<bytearray>self._drv_log_bufs[2]).decode(
@@ -132,11 +132,11 @@ cdef class Linker:
132132
cdef char* c_log_ptr
133133
if self._use_nvjitlink:
134134
c_h = as_cu(self._nvjitlink_handle)
135-
cynvjitlink.nvJitLinkGetInfoLogSize(c_h, &c_log_size)
135+
HANDLE_RETURN_NVJITLINK(cynvjitlink.nvJitLinkGetInfoLogSize(c_h, &c_log_size))
136136
log = bytearray(c_log_size)
137137
if c_log_size > 0:
138138
c_log_ptr = <char*>(<bytearray>log)
139-
cynvjitlink.nvJitLinkGetInfoLog(c_h, c_log_ptr)
139+
HANDLE_RETURN_NVJITLINK(cynvjitlink.nvJitLinkGetInfoLog(c_h, c_log_ptr))
140140
return log.decode("utf-8", errors="backslashreplace")
141141
else:
142142
return (<bytearray>self._drv_log_bufs[0]).decode(

0 commit comments

Comments
 (0)