Skip to content

Commit 1847ee9

Browse files
sophimaopcolberg
authored andcommitted
event: resolve memory leak in ndrange_kernel memory_migration struct
1 parent 2c69880 commit 1847ee9

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/acl_event.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -620,6 +620,13 @@ static void l_release_command_resources(acl_command_info_t &cmd) {
620620

621621
case CL_COMMAND_TASK:
622622
case CL_COMMAND_NDRANGE_KERNEL:
623+
if (cmd.info.ndrange_kernel.memory_migration.num_mem_objects != 0 &&
624+
cmd.info.ndrange_kernel.memory_migration.src_mem_list) {
625+
// src_mem should be user-provided buffers, users are responsible for
626+
// releasing them Just free the src memory list here
627+
acl_free(cmd.info.ndrange_kernel.memory_migration.src_mem_list);
628+
cmd.info.ndrange_kernel.memory_migration.src_mem_list = nullptr;
629+
}
623630
// Cleanup is handled via the completion callback.
624631
break;
625632

0 commit comments

Comments
 (0)