Merge pull request #2551 from przemektmalon/przemek/bindless-images-h…

…ost-usm Enable creation of bindless images backed by host USM
oneapi-src · Jan 27, 2025 · 0bb6789 · 0bb6789
2 parents e370a2b + eeff9f4
commit 0bb6789
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/source/adapters/cuda/image.cpp b/source/adapters/cuda/image.cpp
@@ -533,8 +533,10 @@ UR_APIEXPORT ur_result_t UR_APICALL urBindlessImagesSampledImageCreateExp(
         image_res_desc.resType = CU_RESOURCE_TYPE_MIPMAPPED_ARRAY;
         image_res_desc.res.mipmap.hMipmappedArray = (CUmipmappedArray)hImageMem;
       }
-    } else if (mem_type == CU_MEMORYTYPE_DEVICE) {
-      // We have a USM pointer
+    } else if (mem_type == CU_MEMORYTYPE_DEVICE ||
+               mem_type == CU_MEMORYTYPE_HOST) {
+      // We have a USM pointer.
+      // Images may be created from device or host USM.
       if (pImageDesc->type == UR_MEM_TYPE_IMAGE1D) {
         image_res_desc.resType = CU_RESOURCE_TYPE_LINEAR;
         image_res_desc.res.linear.devPtr = (CUdeviceptr)hImageMem;