SWDEV-465509 - Save the handle type during import function.

Change-Id: If069abb6cd474a7b071617757041402b53575414
This commit is contained in:
kjayapra-amd committed 2024-08-16 10:55:36 -04:00
1 parent 1e4c60f286
commit 8ddb023512
1 file changed
+1
+1
View File
@@ -235,6 +235,7 @@ hipError_t hipMemImportFromShareableHandle(hipMemGenericAllocationHandle_t* hand
prop.type = hipMemAllocationTypePinned;
prop.location.type = hipMemLocationTypeDevice;
prop.location.id = hip::getCurrentDevice()->deviceId();
prop.requestedHandleType = shHandleType;
phys_mem_obj->getUserData().deviceId = hip::getCurrentDevice()->deviceId();
phys_mem_obj->getUserData().data = new hip::GenericAllocation(*phys_mem_obj, 0, prop);