Merge pull request #1061 from ldorau/Check_return_value_of_set_context

ldorau · web-flow · commit e31c85680e9c · 2025-01-27T13:14:39.000+01:00
Check return value of set_context()
diff --git a/src/provider/provider_cuda.c b/src/provider/provider_cuda.c
@@ -594,7 +594,10 @@ static umf_result_t cu_memory_provider_open_ipc_handle(void *provider,
         LOG_ERR("cuIpcOpenMemHandle() failed.");
     }
 
-    set_context(restore_ctx, &restore_ctx);
+    umf_result = set_context(restore_ctx, &restore_ctx);
+    if (umf_result != UMF_RESULT_SUCCESS) {
+        LOG_ERR("Failed to restore CUDA context, ret = %d", umf_result);
+    }
 
     return cu2umf_result(cu_result);
 }

Original file line number	Diff line number	Diff line change
`@@ -594,7 +594,10 @@ static umf_result_t cu_memory_provider_open_ipc_handle(void *provider,`
`594`	`594`	`LOG_ERR("cuIpcOpenMemHandle() failed.");`
`595`	`595`	`}`
`596`	`596`
`597`		`- set_context(restore_ctx, &restore_ctx);`
	`597`	`+ umf_result = set_context(restore_ctx, &restore_ctx);`
	`598`	`+ if (umf_result != UMF_RESULT_SUCCESS) {`
	`599`	`+ LOG_ERR("Failed to restore CUDA context, ret = %d", umf_result);`
	`600`	`+ }`
`598`	`601`
`599`	`602`	`return cu2umf_result(cu_result);`
`600`	`603`	`}`