@@ -723,8 +723,6 @@ ur_result_t urBindlessImagesMapExternalArrayExp(
723
723
(hContext->ZeContext , hDevice->ZeDevice , ZeImage));
724
724
UR_CALL (createUrMemFromZeImage (hContext, ZeImage, /* OwnZeMemHandle*/ true ,
725
725
ZeImageDesc, phImageMem));
726
- externalMemoryData->urMemoryHandle =
727
- reinterpret_cast <ur_mem_handle_t >(*phImageMem);
728
726
return UR_RESULT_SUCCESS;
729
727
}
730
728
@@ -733,7 +731,7 @@ ur_result_t urBindlessImagesMapExternalLinearMemoryExp(
733
731
uint64_t size, ur_exp_external_mem_handle_t hExternalMem, void **phRetMem) {
734
732
UR_ASSERT (hContext && hDevice && hExternalMem,
735
733
UR_RESULT_ERROR_INVALID_NULL_HANDLE);
736
- UR_ASSERT (offset && size, UR_RESULT_ERROR_INVALID_BUFFER_SIZE);
734
+ UR_ASSERT (size, UR_RESULT_ERROR_INVALID_BUFFER_SIZE);
737
735
738
736
struct ur_ze_external_memory_data *externalMemoryData =
739
737
reinterpret_cast <ur_ze_external_memory_data *>(hExternalMem);
@@ -746,24 +744,22 @@ ur_result_t urBindlessImagesMapExternalLinearMemoryExp(
746
744
allocDesc.pNext = externalMemoryData->importExtensionDesc ;
747
745
void *mappedMemory;
748
746
749
- ze_result_t zeResult = zeMemAllocDevice (hContext->ZeContext , &allocDesc, size,
750
- 1 , hDevice->ZeDevice , &mappedMemory);
747
+ ze_result_t zeResult = ZE_CALL_NOCHECK (
748
+ zeMemAllocDevice, (hContext->getZeHandle (), &allocDesc, size, 0 ,
749
+ hDevice->ZeDevice , &mappedMemory));
751
750
if (zeResult != ZE_RESULT_SUCCESS) {
752
751
return UR_RESULT_ERROR_OUT_OF_RESOURCES;
753
752
}
754
753
755
754
zeResult = zeContextMakeMemoryResident (hContext->ZeContext , hDevice->ZeDevice ,
756
755
mappedMemory, size);
757
756
if (zeResult != ZE_RESULT_SUCCESS) {
758
- zeMemFree (hContext->ZeContext , mappedMemory);
757
+ ZE_CALL_NOCHECK ( zeMemFree, (hContext->getZeHandle () , mappedMemory) );
759
758
return UR_RESULT_ERROR_UNKNOWN;
760
759
}
761
760
*phRetMem = reinterpret_cast <void *>(
762
761
reinterpret_cast <uintptr_t >(mappedMemory) + offset);
763
762
764
- externalMemoryData->urMemoryHandle =
765
- reinterpret_cast <ur_mem_handle_t >(*phRetMem);
766
-
767
763
return UR_RESULT_SUCCESS;
768
764
}
769
765
@@ -787,8 +783,6 @@ ur_result_t urBindlessImagesReleaseExternalMemoryExp(
787
783
struct ur_ze_external_memory_data *externalMemoryData =
788
784
reinterpret_cast <ur_ze_external_memory_data *>(hExternalMem);
789
785
790
- UR_CALL (ur::level_zero::urMemRelease (externalMemoryData->urMemoryHandle ));
791
-
792
786
switch (externalMemoryData->type ) {
793
787
case UR_ZE_EXTERNAL_OPAQUE_FD:
794
788
delete (reinterpret_cast <ze_external_memory_import_fd_t *>(
0 commit comments