Merge pull request rusty1s#70 from lgray/patch-1

Ensure all GPU support is turned on

Author: rusty1s

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()