ensure all GPU support is turned on.

5444e856 · Lindsey Gray · GitHub · 7936d3b0 · 5444e856
Unverified Commit 5444e856 authored Jul 13, 2020 by Lindsey Gray Committed by GitHub Jul 13, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +3 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -8,6 +8,9 @@ option(WITH_CUDA "Enable CUDA support" OFF)
 if(WITH_CUDA)
  enable_language(CUDA)
  add_definitions(-D__CUDA_NO_HALF_OPERATORS__)
+  add_definitions(-DWITH_CUDA)
+  set_property(TARGET torch_cuda PROPERTY INTERFACE_COMPILE_OPTIONS "")
+  set_property(TARGET torch_cpu PROPERTY INTERFACE_COMPILE_OPTIONS "")
 endif()

 find_package(Python3 COMPONENTS Development)