Merge pull request #70 from lgray/patch-1

Ensure all GPU support is turned on

Merge pull request #70 from lgray/patch-1
Ensure all GPU support is turned on
8c7560b0 · Matthias Fey · GitHub · 7936d3b0 · 57c0c342 · 8c7560b0
Unverified Commit 8c7560b0 authored Jul 14, 2020 by Matthias Fey Committed by GitHub Jul 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +6 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -8,6 +8,7 @@ option(WITH_CUDA "Enable CUDA support" OFF)
 if(WITH_CUDA)
  enable_language(CUDA)
  add_definitions(-D__CUDA_NO_HALF_OPERATORS__)
+  add_definitions(-DWITH_CUDA)
 endif()

 find_package(Python3 COMPONENTS Development)
@@ -73,3 +74,8 @@ if(WITH_CUDA)
    csrc/cuda/spspmm_cuda.h
    DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/${PROJECT_NAME}/cuda)
 endif()
+
+if(WITH_CUDA)
+  set_property(TARGET torch_cuda PROPERTY INTERFACE_COMPILE_OPTIONS "")
+  set_property(TARGET torch_cpu PROPERTY INTERFACE_COMPILE_OPTIONS "")
+endif()