Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "e00df25aee329ef84450066019701c8d5075cfb3"
Unverified Commit c8e167a7 authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #76 from lgray/patch-1 

Ensure all GPU support is turned on
parents aa9ab1d2 05721150
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
CMakeLists.txt
View file @ c8e167a7
...@@ -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)
...@@ -72,3 +73,8 @@ if(WITH_CUDA) ...@@ -72,3 +73,8 @@ if(WITH_CUDA)
csrc/cuda/rw_cuda.h csrc/cuda/rw_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