Skip to content

Commit 9bd455f

Browse files
committed
Use camelCase always. Return caught UR error directly.
Signed-off-by: JackAKirk <[email protected]>
1 parent 0cdd69f commit 9bd455f

File tree

2 files changed

+16
-22
lines changed

2 files changed

+16
-22
lines changed

source/adapters/cuda/usm_p2p.cpp

Lines changed: 8 additions & 12 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(
@@ -45,16 +41,16 @@ UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
4541
UrReturnHelper ReturnValue(propSize, pPropValue, pPropSizeRet);
4642

4743
int value;
48-
CUdevice_P2PAttribute cu_attr;
44+
CUdevice_P2PAttribute cuAttr;
4945
try {
5046
ScopedContext active(commandDevice->getContext());
5147
switch (propName) {
5248
case UR_EXP_PEER_INFO_UR_PEER_ACCESS_SUPPORTED: {
53-
cu_attr = CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED;
49+
cuAttr = CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED;
5450
break;
5551
}
5652
case UR_EXP_PEER_INFO_UR_PEER_ATOMICS_SUPPORTED: {
57-
cu_attr = CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED;
53+
cuAttr = CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED;
5854
break;
5955
}
6056
default: {
@@ -63,7 +59,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
6359
}
6460

6561
UR_CHECK_ERROR(cuDeviceGetP2PAttribute(
66-
&value, cu_attr, commandDevice->get(), peerDevice->get()));
62+
&value, cuAttr, commandDevice->get(), peerDevice->get()));
6763
} catch (ur_result_t err) {
6864
return err;
6965
}

source/adapters/hip/usm_p2p.cpp

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

1414
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PEnablePeerAccessExp(
1515
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
16-
ur_result_t result = UR_RESULT_SUCCESS;
1716
try {
1817
ScopedContext active(commandDevice);
1918
UR_CHECK_ERROR(hipDeviceEnablePeerAccess(peerDevice->get(), 0));
2019
} catch (ur_result_t err) {
21-
result = err;
20+
return err;
2221
}
23-
return result;
22+
return UR_RESULT_SUCCESS;
2423
}
2524

2625
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PDisablePeerAccessExp(
2726
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
28-
ur_result_t result = UR_RESULT_SUCCESS;
2927
try {
3028
ScopedContext active(commandDevice);
3129
UR_CHECK_ERROR(hipDeviceDisablePeerAccess(peerDevice->get()));
3230
} catch (ur_result_t err) {
33-
result = err;
31+
return err;
3432
}
35-
return result;
33+
return UR_RESULT_SUCCESS;
3634
}
3735

3836
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
@@ -42,24 +40,24 @@ UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
4240
UrReturnHelper ReturnValue(propSize, pPropValue, pPropSizeRet);
4341

4442
int value;
45-
hipDeviceP2PAttr hip_attr;
43+
hipDeviceP2PAttr hipAttr;
4644
try {
4745
ScopedContext active(commandDevice);
4846
switch (propName) {
4947
case UR_EXP_PEER_INFO_UR_PEER_ACCESS_SUPPORTED: {
50-
hip_attr = hipDevP2PAttrAccessSupported;
48+
hipAttr = hipDevP2PAttrAccessSupported;
5149
break;
5250
}
5351
case UR_EXP_PEER_INFO_UR_PEER_ATOMICS_SUPPORTED: {
54-
hip_attr = hipDevP2PAttrNativeAtomicSupported;
52+
hipAttr = hipDevP2PAttrNativeAtomicSupported;
5553
break;
5654
}
5755
default: {
5856
return UR_RESULT_ERROR_INVALID_ENUMERATION;
5957
}
6058
}
6159
UR_CHECK_ERROR(hipDeviceGetP2PAttribute(
62-
&value, hip_attr, commandDevice->get(), peerDevice->get()));
60+
&value, hipAttr, commandDevice->get(), peerDevice->get()));
6361
} catch (ur_result_t err) {
6462
return err;
6563
}

0 commit comments

Comments
 (0)