Skip to content

Commit 062199a

Browse files
authored
Merge pull request #2551 from przemektmalon/przemek/bindless-images-host-usm
Enable creation of bindless images backed by host USM
2 parents 5d20358 + e0b411c commit 062199a

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

source/adapters/cuda/image.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -533,8 +533,10 @@ UR_APIEXPORT ur_result_t UR_APICALL urBindlessImagesSampledImageCreateExp(
533533
image_res_desc.resType = CU_RESOURCE_TYPE_MIPMAPPED_ARRAY;
534534
image_res_desc.res.mipmap.hMipmappedArray = (CUmipmappedArray)hImageMem;
535535
}
536-
} else if (mem_type == CU_MEMORYTYPE_DEVICE) {
537-
// We have a USM pointer
536+
} else if (mem_type == CU_MEMORYTYPE_DEVICE ||
537+
mem_type == CU_MEMORYTYPE_HOST) {
538+
// We have a USM pointer.
539+
// Images may be created from device or host USM.
538540
if (pImageDesc->type == UR_MEM_TYPE_IMAGE1D) {
539541
image_res_desc.resType = CU_RESOURCE_TYPE_LINEAR;
540542
image_res_desc.res.linear.devPtr = (CUdeviceptr)hImageMem;

0 commit comments

Comments
 (0)