Skip to content

GitLab

Unverified Commit b16914be authored by vcarpani's avatar vcarpani Committed by GitHub
Browse files

Expose WITH_CUDA flag to CPP code. (#2585)

parent ea6b879e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
CMakeLists.txt
View file @ b16914be
...@@ -8,6 +8,7 @@ option(WITH_CUDA "Enable CUDA support" OFF) ...@@ -8,6 +8,7 @@ option(WITH_CUDA "Enable CUDA support" OFF)
if(WITH_CUDA) if(WITH_CUDA)
enable_language(CUDA) enable_language(CUDA)
add_definitions(-D__CUDA_NO_HALF_OPERATORS__) add_definitions(-D__CUDA_NO_HALF_OPERATORS__)
add_definitions(-DWITH_CUDA)
endif() endif()
find_package(Python3 COMPONENTS Development) find_package(Python3 COMPONENTS Development)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment