change resetRefCount to have a param and a default value so l0 adapter can start ref count from 0

Author: martygrant

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

@@ -301,6 +301,8 @@ ur_adapter_handle_t_::ur_adapter_handle_t_()
   ZeInitResult = ZE_RESULT_ERROR_UNINITIALIZED;
   ZesResult = ZE_RESULT_ERROR_UNINITIALIZED;
 
+  resetRefCount(0);
+
 #ifdef UR_STATIC_LEVEL_ZERO
   // Given static linking of the L0 Loader, we must delay the loader's
   // destruction of its context until after the UR Adapter is destroyed.
@@ -667,15 +669,15 @@ ur_result_t urAdapterGet(
     uint32_t *NumAdapters) {
   if (NumEntries > 0 && Adapters) {
     if (GlobalAdapter) {
-      if (GlobalAdapter->incrementRefCount() == 1) {
+      if (GlobalAdapter->incrementRefCount() == 0) {
         adapterStateInit();
       }
     } else {
       // If the GetAdapter is called after the Library began or was torndown,
       // then temporarily create a new Adapter handle and register a new
       // cleanup.
       GlobalAdapter = new ur_adapter_handle_t_();
-      if (GlobalAdapter->incrementRefCount() == 1) {
+      if (GlobalAdapter->incrementRefCount() == 0) {
         adapterStateInit();
       }
       std::atexit(globalAdapterOnDemandCleanup);
@@ -693,7 +695,7 @@ ur_result_t urAdapterGet(
 ur_result_t urAdapterRelease(ur_adapter_handle_t) {
   // Check first if the Adapter pointer is valid
   if (GlobalAdapter) {
-    if (GlobalAdapter->decrementRefCount() == 0) {
+    if (GlobalAdapter->decrementAndTest()) {
       auto result = adapterStateTeardown();
 #ifdef UR_STATIC_LEVEL_ZERO
       // Given static linking of the L0 Loader, we must delay the loader's

unified-runtime/source/ur/ur.hpp

@@ -237,7 +237,7 @@ template <typename getddi> struct handle_base {
   uint32_t incrementRefCount() { return ++Count; }
   uint32_t decrementRefCount() { return --Count; }
   bool decrementAndTest() { return --Count == 0; }
-  void resetRefCount() { Count = 1; }
+  void resetRefCount(uint32_t value = 1) { Count = value; }
 
 private:
   std::atomic_uint32_t Count{1};