@@ -1291,7 +1291,7 @@ cl_int CL_API_CALL clEnqueueSVMMemcpy_override(
1291
1291
}
1292
1292
1293
1293
if (isUSMPtr (context, dst_ptr) || isUSMPtr (context, src_ptr)) {
1294
- return clEnqueueMemcpyINTEL (
1294
+ cl_int ret = clEnqueueMemcpyINTEL (
1295
1295
command_queue,
1296
1296
blocking_copy,
1297
1297
dst_ptr,
@@ -1300,6 +1300,11 @@ cl_int CL_API_CALL clEnqueueSVMMemcpy_override(
1300
1300
num_events_in_wait_list,
1301
1301
event_wait_list,
1302
1302
event);
1303
+ if (ret == CL_SUCCESS && event) {
1304
+ auto & eventInfo = getLayerContext ().getEventInfo (*event);
1305
+ eventInfo.Type = CL_COMMAND_SVM_MEMCPY;
1306
+ }
1307
+ return ret;
1303
1308
}
1304
1309
1305
1310
return g_pNextDispatch->clEnqueueSVMMemcpy (
@@ -1342,7 +1347,7 @@ cl_int CL_API_CALL clEnqueueSVMMemFill_override(
1342
1347
}
1343
1348
1344
1349
if (isUSMPtr (context, svm_ptr)) {
1345
- return clEnqueueMemFillINTEL (
1350
+ cl_int ret = clEnqueueMemFillINTEL (
1346
1351
command_queue,
1347
1352
svm_ptr,
1348
1353
pattern,
@@ -1351,6 +1356,11 @@ cl_int CL_API_CALL clEnqueueSVMMemFill_override(
1351
1356
num_events_in_wait_list,
1352
1357
event_wait_list,
1353
1358
event);
1359
+ if (ret == CL_SUCCESS && event) {
1360
+ auto & eventInfo = getLayerContext ().getEventInfo (*event);
1361
+ eventInfo.Type = CL_COMMAND_SVM_MEMFILL;
1362
+ }
1363
+ return ret;
1354
1364
}
1355
1365
1356
1366
return g_pNextDispatch->clEnqueueSVMMemFill (
@@ -1375,11 +1385,16 @@ cl_int CL_API_CALL clEnqueueSVMMigrateMem_override(
1375
1385
cl_event* event)
1376
1386
{
1377
1387
// for now, just emit a marker
1378
- return g_pNextDispatch->clEnqueueMarkerWithWaitList (
1388
+ cl_int ret = g_pNextDispatch->clEnqueueMarkerWithWaitList (
1379
1389
command_queue,
1380
1390
num_events_in_wait_list,
1381
1391
event_wait_list,
1382
1392
event);
1393
+ if (ret == CL_SUCCESS && event) {
1394
+ auto & eventInfo = getLayerContext ().getEventInfo (*event);
1395
+ eventInfo.Type = CL_COMMAND_SVM_MIGRATE_MEM;
1396
+ }
1397
+ return ret;
1383
1398
}
1384
1399
1385
1400
cl_int CL_API_CALL clReleaseEvent_override (
0 commit comments