Handle all cuGraphExecUpdate error codes, not just update failure

Check for CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE first to provide the
rich error message with the update result reason, then fall through
to HANDLE_RETURN for any other error code (CUDA_ERROR_INVALID_VALUE,
CUDA_ERROR_NOT_SUPPORTED, etc.) or success.

Made-with: Cursor

Author: Andy-Jost

cuda_core/cuda/core/_graph/_graph_builder.pyx

@@ -817,11 +817,11 @@ class Graph:
         cdef cydriver.CUresult err
         with nogil:
             err = cydriver.cuGraphExecUpdate(cu_exec, cu_graph, &result_info)
-        if err != cydriver.CUresult.CUDA_SUCCESS:
-            assert err == cydriver.CUresult.CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE
+        if err == cydriver.CUresult.CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE:
             reason = driver.CUgraphExecUpdateResult(result_info.result)
             msg = f"Graph update failed: {reason.__doc__.strip()} ({reason.name})"
             raise CUDAError(msg)
+        HANDLE_RETURN(err)
 
     def upload(self, stream: Stream):
         """Uploads the graph in a stream.