Skip to content

Commit

Permalink
shim: don't close the cufile driver at exit
Browse files Browse the repository at this point in the history
  • Loading branch information
madsbk committed Oct 25, 2024
1 parent bc5debb commit 3d02555
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion cpp/include/kvikio/shim/cufile.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -111,9 +111,11 @@ class cuFileAPI {

// cuFile is supposed to open and close the driver automatically but because of a bug in
// CUDA 11.8, it sometimes segfault. See <https://github.com/rapidsai/kvikio/issues/159>.
// Notice, we don't close the driver at program exit since we are not allowed to call
// cuFile after main:
// <https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#initialization>
driver_open();
}
~cuFileAPI() { driver_close(); }
#else
cuFileAPI() { throw std::runtime_error("KvikIO not compiled with cuFile.h"); }
#endif
Expand Down

0 comments on commit 3d02555

Please sign in to comment.