Skip to content

Commit 4217a2d

Browse files
committed
Revert unrelated cuda changes.
These changes will be made in separate PRs. Signed-off-by: JackAKirk <jack.kirk@codeplay.com>
1 parent 12de78a commit 4217a2d

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

source/adapters/cuda/usm_p2p.cpp

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,24 +13,28 @@
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;
1618
try {
1719
ScopedContext active(commandDevice->getContext());
1820
UR_CHECK_ERROR(cuCtxEnablePeerAccess(peerDevice->getContext(), 0));
1921
} catch (ur_result_t err) {
20-
return err;
22+
result = err;
2123
}
22-
return UR_RESULT_SUCCESS;
24+
return result;
2325
}
2426

2527
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PDisablePeerAccessExp(
2628
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
29+
30+
ur_result_t result = UR_RESULT_SUCCESS;
2731
try {
2832
ScopedContext active(commandDevice->getContext());
2933
UR_CHECK_ERROR(cuCtxDisablePeerAccess(peerDevice->getContext()));
3034
} catch (ur_result_t err) {
31-
return err;
35+
result = err;
3236
}
33-
return UR_RESULT_SUCCESS;
37+
return result;
3438
}
3539

3640
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
@@ -41,25 +45,25 @@ UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
4145
UrReturnHelper ReturnValue(propSize, pPropValue, pPropSizeRet);
4246

4347
int value;
44-
CUdevice_P2PAttribute cuAttr;
48+
CUdevice_P2PAttribute cu_attr;
4549
try {
4650
ScopedContext active(commandDevice->getContext());
4751
switch (propName) {
4852
case UR_EXP_PEER_INFO_UR_PEER_ACCESS_SUPPORTED: {
49-
cuAttr = CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED;
53+
cu_attr = CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED;
5054
break;
5155
}
5256
case UR_EXP_PEER_INFO_UR_PEER_ATOMICS_SUPPORTED: {
53-
cuAttr = CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED;
57+
cu_attr = CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED;
5458
break;
5559
}
5660
default: {
5761
return UR_RESULT_ERROR_INVALID_ENUMERATION;
5862
}
5963
}
6064

61-
UR_CHECK_ERROR(cuDeviceGetP2PAttribute(&value, cuAttr, commandDevice->get(),
62-
peerDevice->get()));
65+
UR_CHECK_ERROR(cuDeviceGetP2PAttribute(
66+
&value, cu_attr, commandDevice->get(), peerDevice->get()));
6367
} catch (ur_result_t err) {
6468
return err;
6569
}

0 commit comments

Comments
 (0)