[SYCL][Graph] Adds retrain/release for the commandbuffer dummy handle…

…r in the unitest PI mock (#310)
reble · Aug 28, 2023 · a6a5df6 · a6a5df6
1 parent 12499c2
commit a6a5df6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sycl/unittests/helpers/PiMockPlugin.hpp b/sycl/unittests/helpers/PiMockPlugin.hpp
@@ -1278,11 +1278,13 @@ mock_piextCommandBufferCreate(pi_context context, pi_device device,
 
 inline pi_result
 mock_piextCommandBufferRetain(pi_ext_command_buffer command_buffer) {
+  retainDummyHandle(command_buffer);
   return PI_SUCCESS;
 }
 
 inline pi_result
 mock_piextCommandBufferRelease(pi_ext_command_buffer command_buffer) {
+  releaseDummyHandle(command_buffer);
   return PI_SUCCESS;
 }