Really release the cached memory

cms-patatrack · Sep 25, 2018 · 97e51f7 · 97e51f7
1 parent 540b284
commit 97e51f7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/HeterogeneousCore/CUDAServices/src/CUDAService.cc b/HeterogeneousCore/CUDAServices/src/CUDAService.cc
@@ -288,6 +288,9 @@ CUDAService::CUDAService(edm::ParameterSet const& config, edm::ActivityRegistry&
 
 CUDAService::~CUDAService() {
   if (enabled_) {
+    // Explicitly destruct the allocator before the device resets below
+    allocator_.reset();
+
     for (int i = 0; i < numberOfDevices_; ++i) {
       cudaCheck(cudaSetDevice(i));
       cudaCheck(cudaDeviceSynchronize());