KVIKIO_CUFILE_DUMMY_BATCH_TYPEDEFS

rapidsai · Aug 28, 2023 · f4f71ab · f4f71ab
1 parent c979aef
commit f4f71ab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
@@ -71,6 +71,9 @@ else()
     set(cuFile_BATCH_AND_STREAM_API_FOUND TRUE)
   endif()
   message(STATUS "Found cuFile's Batch and Stream API: ${cuFile_BATCH_AND_STREAM_API_FOUND}")
+
+  # When the batch and stream API isn't found, we might need some dummy typedefs
+  string(FIND "${CUFILE_H_STR}" "CUfileOpcode" CUfileOpcode_location)
 endif()
 
 # library targets
@@ -90,6 +93,9 @@ if(cuFile_FOUND)
   if(cuFile_BATCH_AND_STREAM_API_FOUND)
     target_compile_definitions(kvikio INTERFACE KVIKIO_CUFILE_BATCH_AND_STREAM_API_FOUND)
   endif()
+  if(CUfileOpcode_location EQUAL "-1")
+    target_compile_definitions(kvikio INTERFACE KVIKIO_CUFILE_DUMMY_BATCH_TYPEDEFS)
+  endif()
 endif()
 target_link_libraries(kvikio INTERFACE ${CMAKE_DL_LIBS})
 target_compile_features(kvikio INTERFACE cxx_std_17)

diff --git a/cpp/include/kvikio/shim/cufile_h_wrapper.hpp b/cpp/include/kvikio/shim/cufile_h_wrapper.hpp
@@ -32,9 +32,9 @@ using CUfileDriverControlFlags_t = enum CUfileDriverControlFlags {
 using CUfileHandle_t = void*;
 #endif
 
-// If the Batch API isn't defined, we define some of the data types here.
+// When the batch and stream API isn't found, we might need some dummy typedefs.
 // Notice, this doesn't need to be ABI compatible with the cufile definitions.
-#ifndef KVIKIO_CUFILE_BATCH_AND_STREAM_API_FOUND
+#ifdef KVIKIO_CUFILE_DUMMY_BATCH_TYPEDEFS
 typedef enum CUfileOpcode { CUFILE_READ = 0, CUFILE_WRITE } CUfileOpcode_t;
 
 typedef enum CUFILEStatus_enum {