Skip to content

Commit 68e525a

Browse files
authored
Merge pull request #1317 from JackAKirk/usm-p2p-cuda-return
[CUDA] Return caught UR error directly.
2 parents 1473ed8 + 1d15931 commit 68e525a

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

source/adapters/cuda/usm_p2p.cpp

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,28 +13,24 @@
1313

1414
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PEnablePeerAccessExp(
1515
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
16-
17-
ur_result_t result = UR_RESULT_SUCCESS;
1816
try {
1917
ScopedContext active(commandDevice->getContext());
2018
UR_CHECK_ERROR(cuCtxEnablePeerAccess(peerDevice->getContext(), 0));
2119
} catch (ur_result_t err) {
22-
result = err;
20+
return err;
2321
}
24-
return result;
22+
return UR_RESULT_SUCCESS;
2523
}
2624

2725
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PDisablePeerAccessExp(
2826
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
29-
30-
ur_result_t result = UR_RESULT_SUCCESS;
3127
try {
3228
ScopedContext active(commandDevice->getContext());
3329
UR_CHECK_ERROR(cuCtxDisablePeerAccess(peerDevice->getContext()));
3430
} catch (ur_result_t err) {
35-
result = err;
31+
return err;
3632
}
37-
return result;
33+
return UR_RESULT_SUCCESS;
3834
}
3935

4036
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(

0 commit comments

Comments
 (0)