GPU: Fix deinitialization order

Was wrong for a long time, but not spotted since return value was not checked before
AliceO2Group · Mar 14, 2024 · b316ab5 · b316ab5
1 parent f50987d
commit b316ab5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/GPU/GPUTracking/Base/cuda/GPUReconstructionCUDA.cu b/GPU/GPUTracking/Base/cuda/GPUReconstructionCUDA.cu
@@ -53,9 +53,6 @@ GPUReconstructionCUDABackend::GPUReconstructionCUDABackend(const GPUSettingsDevi
 GPUReconstructionCUDABackend::~GPUReconstructionCUDABackend()
 {
   if (mMaster == nullptr) {
-    for (unsigned int i = 0; i < mInternals->kernelModules.size(); i++) {
-      GPUFailedMsg(cuModuleUnload(*mInternals->kernelModules[i]));
-    }
     delete mInternals;
   }
 }
@@ -451,6 +448,10 @@ int GPUReconstructionCUDA::ExitDevice_Runtime()
     }
 
     GPUFailedMsgI(cudaFreeHost(mHostMemoryBase));
+    for (unsigned int i = 0; i < mInternals->kernelModules.size(); i++) {
+      GPUFailedMsg(cuModuleUnload(*mInternals->kernelModules[i]));
+    }
+
     GPUFailedMsgI(cudaDeviceReset());
     GPUInfo("CUDA Uninitialized");
   }