Skip to content

GitLab

Unverified Commit 8e967c63 authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #17 from lgray/patch-1 

Ensure all GPU support is turned on
parents c88c95f2 e405535f
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
CMakeLists.txt
View file @ 8e967c63
...@@ -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)
...@@ -64,3 +65,8 @@ if(WITH_CUDA) ...@@ -64,3 +65,8 @@ if(WITH_CUDA)
csrc/cuda/weighting_cuda.h csrc/cuda/weighting_cuda.h
DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/${PROJECT_NAME}/cuda) DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/${PROJECT_NAME}/cuda)
endif() endif()
if(WITH_CUDA)
set_property(TARGET torch_cuda PROPERTY INTERFACE_COMPILE_OPTIONS "")
set_property(TARGET torch_cpu PROPERTY INTERFACE_COMPILE_OPTIONS "")
endif()
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