[Fix] Fixed FlexPinnedBuffer destructor

NVIDIA-Merlin · May 25, 2023 · 7a22173 · 7a22173
1 parent f1aa393
commit 7a22173
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/merlin/flexible_buffer.cuh b/include/merlin/flexible_buffer.cuh
@@ -36,7 +36,7 @@ class FlexPinnedBuffer {
   }
   ~FlexPinnedBuffer() {
     try {
-      if (!ptr_) CUDA_CHECK(cudaFreeHost(ptr_));
+      if (ptr_) CUDA_CHECK(cudaFreeHost(ptr_));
     } catch (const nv::merlin::CudaException& e) {
       cerr << "[HierarchicalKV] Failed to free FlexPinnedBuffer!" << endl;
     }