Skip to content

Commit

Permalink
Define PTDS for the stream hook libs (#16182)
Browse files Browse the repository at this point in the history
We must define `CUDA_API_PER_THREAD_DEFAULT_STREAM` for the stream hook lib, since `cudaLaunchKernel` in CUDA 12.4+ is now a macro that expands to a different function when it's not defined.

Authors:
  - Paul Taylor (https://github.com/trxcllnt)

Approvers:
  - Robert Maynard (https://github.com/robertmaynard)

URL: #16182
  • Loading branch information
trxcllnt authored Jul 8, 2024
1 parent f6b355d commit d9a3728
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions cpp/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -925,6 +925,11 @@ if(CUDF_BUILD_STREAMS_TEST_UTIL)
add_library(
${_tgt} SHARED src/utilities/stacktrace.cpp tests/utilities/identify_stream_usage.cpp
)
if(CUDF_USE_PER_THREAD_DEFAULT_STREAM)
target_compile_definitions(
${_tgt} PUBLIC CUDA_API_PER_THREAD_DEFAULT_STREAM CUDF_USE_PER_THREAD_DEFAULT_STREAM
)
endif()

set_target_properties(
${_tgt}
Expand Down

0 comments on commit d9a3728

Please sign in to comment.