@@ -150,27 +150,15 @@ ur_result_t USMDeviceAllocImpl(void **ResultPtr, ur_context_handle_t,
150150 return UR_RESULT_SUCCESS;
151151}
152152
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 ,
155155 ur_usm_host_mem_flags_t ,
156156 ur_usm_device_mem_flags_t , size_t Size,
157157 uint32_t Alignment) {
158158 try {
159- ScopedContext Active (CommandDevice );
159+ ScopedContext Active (Device );
160160 UR_CHECK_ERROR (cuMemAllocManaged ((CUdeviceptr *)ResultPtr, Size,
161161 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- }
174162 } catch (ur_result_t Err) {
175163 return Err;
176164 }
0 commit comments