Correctly implement reference counting for Native CPU

Author: PietroGhg

source/adapters/native_cpu/context.cpp

@@ -38,7 +38,9 @@ urContextRetain(ur_context_handle_t hContext) {
 
 UR_APIEXPORT ur_result_t UR_APICALL
 urContextRelease(ur_context_handle_t hContext) {
-  delete hContext;
+  hContext->decrementReferenceCount();
+  if(hContext->getReferenceCount() == 0)
+    delete hContext;
   return UR_RESULT_SUCCESS;
 }
 

source/adapters/native_cpu/context.hpp

@@ -13,8 +13,9 @@
 #include <ur_api.h>
 
 #include "device.hpp"
+#include "common.hpp"
 
-struct ur_context_handle_t_ {
+struct ur_context_handle_t_ : RefCounted {
   ur_context_handle_t_(ur_device_handle_t_ *phDevices) : _device{phDevices} {}
 
   ur_device_handle_t _device;

source/adapters/native_cpu/kernel.cpp

@@ -182,7 +182,9 @@ UR_APIEXPORT ur_result_t UR_APICALL urKernelRetain(ur_kernel_handle_t hKernel) {
 
 UR_APIEXPORT ur_result_t UR_APICALL
 urKernelRelease(ur_kernel_handle_t hKernel) {
-  delete hKernel;
+  hKernel->decrementReferenceCount();
+  if(hKernel->getReferenceCount() == 0)
+    delete hKernel;
 
   return UR_RESULT_SUCCESS;
 }

source/adapters/native_cpu/program.cpp

@@ -95,7 +95,9 @@ urProgramRetain(ur_program_handle_t hProgram) {
 
 UR_APIEXPORT ur_result_t UR_APICALL
 urProgramRelease(ur_program_handle_t hProgram) {
-  delete hProgram;
+  hProgram->decrementReferenceCount();
+  if(hProgram->getReferenceCount() == 0)
+    delete hProgram;
 
   return UR_RESULT_SUCCESS;
 }

source/adapters/native_cpu/queue.cpp

@@ -43,12 +43,16 @@ UR_APIEXPORT ur_result_t UR_APICALL urQueueCreate(
 
 UR_APIEXPORT ur_result_t UR_APICALL urQueueRetain(ur_queue_handle_t hQueue) {
   std::ignore = hQueue;
+  hQueue->incrementReferenceCount();
 
-  DIE_NO_IMPLEMENTATION;
+  return UR_RESULT_SUCCESS;
 }
 
 UR_APIEXPORT ur_result_t UR_APICALL urQueueRelease(ur_queue_handle_t hQueue) {
-  delete hQueue;
+  hQueue->decrementReferenceCount();
+  if(hQueue->getReferenceCount() == 0)
+    delete hQueue;
+
   return UR_RESULT_SUCCESS;
 }
 

source/adapters/native_cpu/queue.hpp

@@ -8,5 +8,6 @@
 //
 //===----------------------------------------------------------------------===//
 #pragma once
+#include "common.hpp"
 
-struct ur_queue_handle_t_ {};
+struct ur_queue_handle_t_ : RefCounted {};