Add decrementAndTest back in to fix logic in checks.

Author: martygrant

unified-runtime/source/adapters/level_zero/command_buffer.cpp

@@ -846,7 +846,7 @@ urCommandBufferRetainExp(ur_exp_command_buffer_handle_t CommandBuffer) {
 
 ur_result_t
 urCommandBufferReleaseExp(ur_exp_command_buffer_handle_t CommandBuffer) {
-  if (!CommandBuffer->decrementRefCount() == 0)
+  if (!CommandBuffer->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   UR_CALL(waitForOngoingExecution(CommandBuffer));

unified-runtime/source/adapters/level_zero/context.cpp

@@ -251,7 +251,7 @@ ur_device_handle_t ur_context_handle_t_::getRootDevice() const {
 // from the list of tracked contexts.
 ur_result_t ContextReleaseHelper(ur_context_handle_t Context) {
 
-  if (!Context->decrementRefCount() == 0)
+  if (!Context->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   if (IndirectAccessTrackingEnabled) {

unified-runtime/source/adapters/level_zero/device.cpp

@@ -1674,7 +1674,7 @@ ur_result_t urDeviceRetain(ur_device_handle_t Device) {
 ur_result_t urDeviceRelease(ur_device_handle_t Device) {
   // Root devices are destroyed during the piTearDown process.
   if (Device->isSubDevice()) {
-    if (Device->decrementRefCount() == 0) {
+    if (Device->decrementAndTest()) {
       delete Device;
     }
   }

unified-runtime/source/adapters/level_zero/event.cpp

@@ -1088,7 +1088,7 @@ ur_event_handle_t_::~ur_event_handle_t_() {
 
 ur_result_t urEventReleaseInternal(ur_event_handle_t Event,
                                    bool *isEventDeleted) {
-  if (!Event->decrementRefCount() == 0)
+  if (!Event->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   if (Event->OriginAllocEvent) {

unified-runtime/source/adapters/level_zero/kernel.cpp

@@ -946,7 +946,7 @@ ur_result_t urKernelRetain(
 ur_result_t urKernelRelease(
     /// [in] handle for the Kernel to release
     ur_kernel_handle_t Kernel) {
-  if (!Kernel->decrementRefCount() == 0)
+  if (!Kernel->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   auto KernelProgram = Kernel->Program;

unified-runtime/source/adapters/level_zero/memory.cpp

@@ -1642,7 +1642,7 @@ ur_result_t urMemRetain(
 ur_result_t urMemRelease(
     /// [in] handle of the memory object to release
     ur_mem_handle_t Mem) {
-  if (!Mem->decrementRefCount() == 0)
+  if (!Mem->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   if (Mem->isImage()) {

unified-runtime/source/adapters/level_zero/physical_mem.cpp

@@ -47,7 +47,7 @@ ur_result_t urPhysicalMemRetain(ur_physical_mem_handle_t hPhysicalMem) {
 }
 
 ur_result_t urPhysicalMemRelease(ur_physical_mem_handle_t hPhysicalMem) {
-  if (!hPhysicalMem->decrementRefCount() == 0)
+  if (!hPhysicalMem->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   if (checkL0LoaderTeardown()) {

unified-runtime/source/adapters/level_zero/program.cpp

@@ -565,7 +565,7 @@ ur_result_t urProgramRetain(
 ur_result_t urProgramRelease(
     /// [in] handle for the Program to release
     ur_program_handle_t Program) {
-  if (!Program->decrementRefCount() == 0)
+  if (!Program->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   delete Program;
@@ -1115,7 +1115,7 @@ void ur_program_handle_t_::ur_release_program_resources(bool deletion) {
   // must be destroyed before the Module can be destroyed.  So, be sure
   // to destroy build log before destroying the module.
   if (!deletion) {
-    if (!decrementRefCount() == 0) {
+    if (!decrementAndTest()) {
       return;
     }
   }

unified-runtime/source/adapters/level_zero/queue.cpp

@@ -612,7 +612,7 @@ ur_result_t urQueueRelease(
       // internal reference count. When the External Reference count == 0, then
       // cleanup of the queue begins and the final decrement of the internal
       // reference count is completed.
-      static_cast<void>(Queue->decrementRefCount() == 0);
+      static_cast<void>(Queue->decrementAndTest());
       return UR_RESULT_SUCCESS;
     }
 
@@ -1588,7 +1588,7 @@ void ur_queue_handle_t_::clearEndTimeRecordings() {
 }
 
 ur_result_t urQueueReleaseInternal(ur_queue_handle_t Queue) {
-  if (!Queue->decrementRefCount() == 0)
+  if (!Queue->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   for (auto &Cache : Queue->EventCaches) {

unified-runtime/source/adapters/level_zero/sampler.cpp

@@ -131,7 +131,7 @@ ur_result_t urSamplerRetain(
 ur_result_t urSamplerRelease(
     /// [in] handle of the sampler object to release
     ur_sampler_handle_t Sampler) {
-  if (!Sampler->decrementRefCount() == 0)
+  if (!Sampler->decrementAndTest())
     return UR_RESULT_SUCCESS;
 
   if (checkL0LoaderTeardown()) {