Skip to content

Commit

Permalink
Style
Browse files Browse the repository at this point in the history
  • Loading branch information
KyleFromNVIDIA committed Nov 8, 2024
1 parent 250517b commit 15be9f3
Showing 1 changed file with 14 additions and 4 deletions.
18 changes: 14 additions & 4 deletions cpp/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -85,24 +85,34 @@ else()
set(cuFile_FOUND 1)

# Check batch and stream API support (cuFile_BATCH_API_FOUND and cuFile_STREAM_API_FOUND)
try_compile(cuFile_BATCH_API_FOUND OUTPUT_VARIABLE batch_output SOURCE_FROM_CONTENT batch.cpp
try_compile(
cuFile_BATCH_API_FOUND OUTPUT_VARIABLE
batch_output
SOURCE_FROM_CONTENT
batch.cpp
[[#include <cufile.h>
int main() {
cuFileBatchIOSetUp(nullptr, 0);
return 0;
}
]])
]]
)
message(STATUS "batch output: ${batch_output}")
message(STATUS "Found cuFile Batch API: ${cuFile_BATCH_API_FOUND}")
try_compile(cuFile_STREAM_API_FOUND OUTPUT_VARIABLE stream_output SOURCE_FROM_CONTENT stream.cpp
try_compile(
cuFile_STREAM_API_FOUND OUTPUT_VARIABLE
stream_output
SOURCE_FROM_CONTENT
stream.cpp
[[#include <cufile.h>
int main() {
CUFileHandle_t fh;
CUstream stream;
cuFileReadAsync(fh, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr);
return 0;
}
]])
]]
)
message(STATUS "stream output: ${stream_output}")
message(STATUS "Found cuFile Stream API: ${cuFile_STREAM_API_FOUND}")
endif()
Expand Down

0 comments on commit 15be9f3

Please sign in to comment.