[offload] Fix missing reference decrement introduced by merge resolution

Added line which has been dropped from the 'deinitRuntime()' during merge-conflict resolution. Change-Id: Iee2c8b2fe63d8cd36cdb9befca2e8c93384087d9
ROCm · Apr 30, 2024 · 79ca523 · 79ca523
1 parent 73aa06a
commit 79ca523
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/offload/src/OffloadRTL.cpp b/offload/src/OffloadRTL.cpp
@@ -54,6 +54,7 @@ void deinitRuntime() {
     PM = nullptr;
   }
 
+  RefCount--;
 }
 
 // HACK: These depricated device stubs still needs host versions for fallback

diff --git a/offload/src/interface.cpp b/offload/src/interface.cpp
@@ -478,17 +478,17 @@ EXTERN void __tgt_push_mapper_component(void *RtMapperHandle, void *Base,
 }
 
 EXTERN void __tgt_set_info_flag(uint32_t NewInfoLevel) {
-  OMPT_IF_BUILT(ReturnAddressSetterRAII RA(__builtin_return_address(0)));
   assert(PM && "Runtime not initialized");
+  OMPT_IF_BUILT(ReturnAddressSetterRAII RA(__builtin_return_address(0)));
   std::atomic<uint32_t> &InfoLevel = getInfoLevelInternal();
   InfoLevel.store(NewInfoLevel);
   for (auto &R : PM->pluginAdaptors())
     R.set_info_flag(NewInfoLevel);
 }
 
 EXTERN int __tgt_print_device_info(int64_t DeviceId) {
-  OMPT_IF_BUILT(ReturnAddressSetterRAII RA(__builtin_return_address(0)));
   assert(PM && "Runtime not initialized");
+  OMPT_IF_BUILT(ReturnAddressSetterRAII RA(__builtin_return_address(0)));
   auto DeviceOrErr = PM->getDevice(DeviceId);
   if (!DeviceOrErr)
     FATAL_MESSAGE(DeviceId, "%s", toString(DeviceOrErr.takeError()).c_str());