🐛 Fix dlpack tensor evaporation

yhs0602 · Jan 1, 2025 · 8eef6af · 8eef6af
1 parent 04fb1d6
commit 8eef6af
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/cpp/ipc_cuda.cpp b/src/cpp/ipc_cuda.cpp
@@ -5,6 +5,9 @@
 #include <string>
 
 static void deleteDLManagedTensor(DLManagedTensor *self) {
+    return;
+    /*
+    Allocate once, deallocate once. (On exit)
     cudaError_t err = cudaIpcCloseMemHandle(self->dl_tensor.data);
     if (err != cudaSuccess) {
         fprintf(stderr, "Failed to close CUDA IPC handle: %s\n", cudaGetErrorString(err));
@@ -14,6 +17,7 @@ static void deleteDLManagedTensor(DLManagedTensor *self) {
     fflush(stderr);
     free(self->dl_tensor.shape);
     free(self);
+    */
 }
 
 DLManagedTensor *