Return tuple on failed malloc

vzhurba01 · vzhurba01 · commit f00d7bfb4da9 · 2025-04-03T14:17:33.000-07:00
diff --git a/cuda_bindings/cuda/bindings/driver.pyx.in b/cuda_bindings/cuda/bindings/driver.pyx.in
@@ -27841,7 +27841,7 @@ def cuDeviceRegisterAsyncNotification(device, callbackFunc, userData):
     cdef cuAsyncCallbackData *cbData = NULL
     cbData = <cuAsyncCallbackData *>malloc(sizeof(cbData[0]))
     if cbData == NULL:
-        return CUresult.CUDA_ERROR_OUT_OF_MEMORY
+        return (CUresult.CUDA_ERROR_OUT_OF_MEMORY, None)
     cbData.callback = cycallbackFunc
     cbData.userData = cyuserData_ptr
 
@@ -35589,7 +35589,7 @@ def cuStreamAddCallback(hStream, callback, userData, unsigned int flags):
     cdef cuStreamCallbackData *cbData = NULL
     cbData = <cuStreamCallbackData *>malloc(sizeof(cbData[0]))
     if cbData == NULL:
-        return CUresult.CUDA_ERROR_OUT_OF_MEMORY
+        return (CUresult.CUDA_ERROR_OUT_OF_MEMORY,)
     cbData.callback = cycallback
     cbData.userData = cyuserData_ptr
 
@@ -39422,7 +39422,7 @@ def cuLaunchHostFunc(hStream, fn, userData):
     cdef cuHostCallbackData *cbData = NULL
     cbData = <cuHostCallbackData *>malloc(sizeof(cbData[0]))
     if cbData == NULL:
-        return CUresult.CUDA_ERROR_OUT_OF_MEMORY
+        return (CUresult.CUDA_ERROR_OUT_OF_MEMORY,)
     cbData.callback = cyfn
     cbData.userData = cyuserData_ptr
 
diff --git a/cuda_bindings/cuda/bindings/runtime.pyx.in b/cuda_bindings/cuda/bindings/runtime.pyx.in
@@ -18501,7 +18501,7 @@ def cudaDeviceRegisterAsyncNotification(int device, callbackFunc, userData):
     cdef cudaAsyncCallbackData *cbData = NULL
     cbData = <cudaAsyncCallbackData *>malloc(sizeof(cbData[0]))
     if cbData == NULL:
-        return cudaError_t.cudaErrorMemoryAllocation
+        return (cudaError_t.cudaErrorMemoryAllocation, None)
     cbData.callback = cycallbackFunc
     cbData.userData = cyuserData_ptr
 
@@ -20692,7 +20692,7 @@ def cudaStreamAddCallback(stream, callback, userData, unsigned int flags):
     cdef cudaStreamCallbackData *cbData = NULL
     cbData = <cudaStreamCallbackData *>malloc(sizeof(cbData[0]))
     if cbData == NULL:
-        return cudaError_t.cudaErrorMemoryAllocation
+        return (cudaError_t.cudaErrorMemoryAllocation,)
     cbData.callback = cycallback
     cbData.userData = cyuserData_ptr
 
@@ -23072,7 +23072,7 @@ def cudaLaunchHostFunc(stream, fn, userData):
     cdef cudaStreamHostCallbackData *cbData = NULL
     cbData = <cudaStreamHostCallbackData *>malloc(sizeof(cbData[0]))
     if cbData == NULL:
-        return cudaError_t.cudaErrorMemoryAllocation
+        return (cudaError_t.cudaErrorMemoryAllocation,)
     cbData.callback = cyfn
     cbData.userData = cyuserData_ptr
 
