Merge pull request #1317 from JackAKirk/usm-p2p-cuda-return

[CUDA] Return caught UR error directly.
oneapi-src · Apr 12, 2024 · 68e525a · 68e525a
2 parents 1473ed8 + 1d15931
commit 68e525a
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/source/adapters/cuda/usm_p2p.cpp b/source/adapters/cuda/usm_p2p.cpp
@@ -13,28 +13,24 @@
 
 UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PEnablePeerAccessExp(
  ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
-
- ur_result_t result = UR_RESULT_SUCCESS;
  try {
  ScopedContext active(commandDevice->getContext());
  UR_CHECK_ERROR(cuCtxEnablePeerAccess(peerDevice->getContext(), 0));
  } catch (ur_result_t err) {
- result = err;
+ return err;
  }
- return result;
+ return UR_RESULT_SUCCESS;
 }
 
 UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PDisablePeerAccessExp(
  ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
-
- ur_result_t result = UR_RESULT_SUCCESS;
  try {
  ScopedContext active(commandDevice->getContext());
  UR_CHECK_ERROR(cuCtxDisablePeerAccess(peerDevice->getContext()));
  } catch (ur_result_t err) {
- result = err;
+ return err;
  }
- return result;
+ return UR_RESULT_SUCCESS;
 }
 
 UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(