13
13
14
14
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PEnablePeerAccessExp (
15
15
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
16
+
17
+ ur_result_t result = UR_RESULT_SUCCESS;
16
18
try {
17
19
ScopedContext active (commandDevice->getContext ());
18
20
UR_CHECK_ERROR (cuCtxEnablePeerAccess (peerDevice->getContext (), 0 ));
19
21
} catch (ur_result_t err) {
20
- return err;
22
+ result = err;
21
23
}
22
- return UR_RESULT_SUCCESS ;
24
+ return result ;
23
25
}
24
26
25
27
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PDisablePeerAccessExp (
26
28
ur_device_handle_t commandDevice, ur_device_handle_t peerDevice) {
29
+
30
+ ur_result_t result = UR_RESULT_SUCCESS;
27
31
try {
28
32
ScopedContext active (commandDevice->getContext ());
29
33
UR_CHECK_ERROR (cuCtxDisablePeerAccess (peerDevice->getContext ()));
30
34
} catch (ur_result_t err) {
31
- return err;
35
+ result = err;
32
36
}
33
- return UR_RESULT_SUCCESS ;
37
+ return result ;
34
38
}
35
39
36
40
UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp (
@@ -41,25 +45,25 @@ UR_APIEXPORT ur_result_t UR_APICALL urUsmP2PPeerAccessGetInfoExp(
41
45
UrReturnHelper ReturnValue (propSize, pPropValue, pPropSizeRet);
42
46
43
47
int value;
44
- CUdevice_P2PAttribute cuAttr ;
48
+ CUdevice_P2PAttribute cu_attr ;
45
49
try {
46
50
ScopedContext active (commandDevice->getContext ());
47
51
switch (propName) {
48
52
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;
50
54
break ;
51
55
}
52
56
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;
54
58
break ;
55
59
}
56
60
default : {
57
61
return UR_RESULT_ERROR_INVALID_ENUMERATION;
58
62
}
59
63
}
60
64
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 ()));
63
67
} catch (ur_result_t err) {
64
68
return err;
65
69
}
0 commit comments