@@ -32,7 +32,7 @@ ur_command_list_manager::~ur_command_list_manager() {
32
32
}
33
33
34
34
ur_result_t ur_command_list_manager::appendGenericFillUnlocked (
35
- ur_mem_handle_t dst, size_t offset, size_t patternSize,
35
+ ur_mem_buffer_t * dst, size_t offset, size_t patternSize,
36
36
const void *pPattern, size_t size, uint32_t numEventsInWaitList,
37
37
const ur_event_handle_t *phEventWaitList, ur_event_handle_t *phEvent,
38
38
ur_command_t commandType) {
@@ -42,7 +42,7 @@ ur_result_t ur_command_list_manager::appendGenericFillUnlocked(
42
42
auto waitListView = getWaitListView (phEventWaitList, numEventsInWaitList);
43
43
44
44
auto pDst = ur_cast<char *>(dst->getDevicePtr (
45
- device, ur_mem_handle_t_ ::device_access_mode_t ::read_only, offset, size,
45
+ device, ur_mem_buffer_t ::device_access_mode_t ::read_only, offset, size,
46
46
[&](void *src, void *dst, size_t size) {
47
47
ZE2UR_CALL_THROWS (zeCommandListAppendMemoryCopy,
48
48
(zeCommandList.get (), dst, src, size, nullptr ,
@@ -254,11 +254,12 @@ ur_result_t ur_command_list_manager::appendUSMMemcpy(
254
254
}
255
255
256
256
ur_result_t ur_command_list_manager::appendMemBufferFill (
257
- ur_mem_handle_t hBuffer , const void *pPattern, size_t patternSize,
257
+ ur_mem_handle_t hMem , const void *pPattern, size_t patternSize,
258
258
size_t offset, size_t size, uint32_t numEventsInWaitList,
259
259
const ur_event_handle_t *phEventWaitList, ur_event_handle_t *phEvent) {
260
260
TRACK_SCOPE_LATENCY (" ur_command_list_manager::appendMemBufferFill" );
261
261
262
+ auto hBuffer = hMem->getBuffer ();
262
263
UR_ASSERT (offset + size <= hBuffer->getSize (), UR_RESULT_ERROR_INVALID_SIZE);
263
264
264
265
std::scoped_lock<ur_shared_mutex, ur_shared_mutex> lock (this ->Mutex ,
@@ -277,7 +278,7 @@ ur_result_t ur_command_list_manager::appendUSMFill(
277
278
278
279
std::scoped_lock<ur_shared_mutex> lock (this ->Mutex );
279
280
280
- ur_usm_handle_t_ dstHandle (context, size, pMem);
281
+ ur_usm_handle_t dstHandle (context, size, pMem);
281
282
return appendGenericFillUnlocked (&dstHandle, 0 , patternSize, pPattern, size,
282
283
numEventsInWaitList, phEventWaitList,
283
284
phEvent, UR_COMMAND_USM_FILL);
0 commit comments