@@ -150,27 +150,15 @@ ur_result_t USMDeviceAllocImpl(void **ResultPtr, ur_context_handle_t,
150
150
return UR_RESULT_SUCCESS;
151
151
}
152
152
153
- ur_result_t USMSharedAllocImpl (void **ResultPtr, ur_context_handle_t Context ,
154
- ur_device_handle_t CommandDevice ,
153
+ ur_result_t USMSharedAllocImpl (void **ResultPtr, ur_context_handle_t ,
154
+ ur_device_handle_t Device ,
155
155
ur_usm_host_mem_flags_t ,
156
156
ur_usm_device_mem_flags_t , size_t Size,
157
157
uint32_t Alignment) {
158
158
try {
159
- ScopedContext Active (CommandDevice );
159
+ ScopedContext Active (Device );
160
160
UR_CHECK_ERROR (cuMemAllocManaged ((CUdeviceptr *)ResultPtr, Size,
161
161
CU_MEM_ATTACH_GLOBAL));
162
- if (getAttribute (CommandDevice,
163
- CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS)) {
164
- UR_CHECK_ERROR (cuMemAdvise ((CUdeviceptr)*ResultPtr, Size,
165
- CU_MEM_ADVISE_SET_ACCESSED_BY,
166
- CommandDevice->get ()));
167
- }
168
- for (const auto &Dev : Context->getDevices ()) {
169
- if (getAttribute (Dev, CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS)) {
170
- UR_CHECK_ERROR (cuMemAdvise ((CUdeviceptr)*ResultPtr, Size,
171
- CU_MEM_ADVISE_SET_ACCESSED_BY, Dev->get ()));
172
- }
173
- }
174
162
} catch (ur_result_t Err) {
175
163
return Err;
176
164
}
0 commit comments