@@ -195,7 +195,7 @@ static ur_result_t USMDeviceAllocImpl(void **ResultPtr,
195
195
auto Result = USMAllocationMakeResident (USMDeviceAllocationForceResidency,
196
196
Context, Device, *ResultPtr, Size);
197
197
if (Result == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY ||
198
- Result == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY ) {
198
+ Result == UR_RESULT_ERROR_OUT_OF_HOST_MEMORY ) {
199
199
return Result;
200
200
}
201
201
return UR_RESULT_SUCCESS;
@@ -232,7 +232,7 @@ static ur_result_t USMSharedAllocImpl(void **ResultPtr,
232
232
auto Result = USMAllocationMakeResident (USMSharedAllocationForceResidency,
233
233
Context, Device, *ResultPtr, Size);
234
234
if (Result == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY ||
235
- Result == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY ) {
235
+ Result == UR_RESULT_ERROR_OUT_OF_HOST_MEMORY ) {
236
236
return Result;
237
237
}
238
238
@@ -258,7 +258,7 @@ static ur_result_t USMHostAllocImpl(void **ResultPtr,
258
258
auto Result = USMAllocationMakeResident (USMHostAllocationForceResidency,
259
259
Context, nullptr , *ResultPtr, Size);
260
260
if (Result == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY ||
261
- Result == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY ) {
261
+ Result == UR_RESULT_ERROR_OUT_OF_HOST_MEMORY ) {
262
262
return Result;
263
263
}
264
264
return UR_RESULT_SUCCESS;
0 commit comments