Skip to content

Commit 10f04d3

Browse files
committed
Avoid unnecessary copy when host ptr is not supplied.
1 parent 2e5b580 commit 10f04d3

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

sycl/plugins/level_zero/pi_level_zero.cpp

100644100755
Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1877,15 +1877,15 @@ pi_result piMemBufferCreate(pi_context Context, pi_mem_flags Flags, size_t Size,
18771877
ze_host_mem_alloc_desc_t ZeDesc = {};
18781878
ZeDesc.flags = 0;
18791879

1880-
ZE_CALL(zeMemAllocHost(Context->ZeContext, &ZeDesc, Size, 4096, &Ptr));
1880+
ZE_CALL(zeMemAllocHost(Context->ZeContext, &ZeDesc, Size, 1, &Ptr));
18811881

18821882
} else {
18831883
ze_device_mem_alloc_desc_t ZeDesc = {};
18841884
ZeDesc.flags = 0;
18851885
ZeDesc.ordinal = 0;
18861886

1887-
ZE_CALL(zeMemAllocDevice(Context->ZeContext, &ZeDesc, Size, 4096, ZeDevice,
1888-
&Ptr));
1887+
ZE_CALL(
1888+
zeMemAllocDevice(Context->ZeContext, &ZeDesc, Size, 1, ZeDevice, &Ptr));
18891889
}
18901890
if (HostPtr) {
18911891
if ((Flags & PI_MEM_FLAGS_HOST_PTR_USE) != 0 ||
@@ -4061,7 +4061,7 @@ piEnqueueMemBufferMap(pi_queue Queue, pi_mem Buffer, pi_bool BlockingMap,
40614061
ZeDesc.flags = 0;
40624062

40634063
ZE_CALL(
4064-
zeMemAllocHost(Queue->Context->ZeContext, &ZeDesc, Size, 4096, RetMap));
4064+
zeMemAllocHost(Queue->Context->ZeContext, &ZeDesc, Size, 1, RetMap));
40654065
}
40664066

40674067
ZE_CALL(zeCommandListAppendWaitOnEvents(ZeCommandList, NumEventsInWaitList,
@@ -4124,8 +4124,9 @@ pi_result piEnqueueMemUnmap(pi_queue Queue, pi_mem MemObj, void *MappedPtr,
41244124
if (MemObj->OnHost) {
41254125
// Wait on incoming events before doing the copy
41264126
piEventsWait(NumEventsInWaitList, EventWaitList);
4127-
memcpy(pi_cast<char *>(MemObj->getZeHandle()) + MapInfo.Offset, MappedPtr,
4128-
MapInfo.Size);
4127+
if (MemObj->MapHostPtr)
4128+
memcpy(pi_cast<char *>(MemObj->getZeHandle()) + MapInfo.Offset, MappedPtr,
4129+
MapInfo.Size);
41294130

41304131
// Signal this event
41314132
ZE_CALL(zeEventHostSignal(ZeEvent));

0 commit comments

Comments
 (0)