From 1c64c2d8924a0066dec6f8ef47d08462510ba59c Mon Sep 17 00:00:00 2001 From: Stefano Sinigardi Date: Wed, 29 Apr 2020 17:33:03 +0200 Subject: [PATCH] enable cc70,cc72,cc75 for half_precision, add message for omp missing on macOS --- CMakeLists.txt | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 7674b6bd..fbf15a1d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -54,7 +54,6 @@ if(CMAKE_CUDA_COMPILER AND ENABLE_CUDA) set(CUDA_ARCHITECTURES "Auto" CACHE STRING "\"Auto\" detects local machine GPU compute arch at runtime, \"Common\" and \"All\" cover common and entire subsets of architectures, \"Names\" is a list of architectures to enable by name, \"Numbers\" is a list of compute capabilities (version number) to enable") set_property(CACHE CUDA_ARCHITECTURES PROPERTY STRINGS "Auto" "Common" "All" "Kepler Maxwell Kepler+Tegra Maxwell+Tegra Pascal" "3.0 7.5") enable_language(CUDA) - find_package(CUDAToolkit REQUIRED) if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS "9.0") message(STATUS "Unsupported CUDA version, please upgrade to CUDA 9+. Disabling CUDA support") set(ENABLE_CUDA "FALSE" CACHE BOOL "Enable CUDA support" FORCE) @@ -62,7 +61,7 @@ if(CMAKE_CUDA_COMPILER AND ENABLE_CUDA) find_package(CUDA REQUIRED) cuda_select_nvcc_arch_flags(CUDA_ARCH_FLAGS ${CUDA_ARCHITECTURES}) message(STATUS "Building with CUDA flags: " "${CUDA_ARCH_FLAGS}") - if (NOT "arch=compute_70,code=sm_70" IN_LIST CUDA_ARCH_FLAGS) + if (NOT "arch=compute_70,code=sm_70" IN_LIST CUDA_ARCH_FLAGS AND NOT "arch=compute_72,code=sm_72" IN_LIST CUDA_ARCH_FLAGS AND NOT "arch=compute_75,code=sm_75" IN_LIST CUDA_ARCH_FLAGS) set(ENABLE_CUDNN_HALF "FALSE" CACHE BOOL "Enable CUDNN Half precision" FORCE) message(STATUS "Your setup does not supports half precision (it requires CC >= 7.5)") endif() @@ -125,6 +124,10 @@ endif() find_package(Stb REQUIRED) find_package(OpenMP) +if(APPLE AND NOT OPENMP_FOUND) + message(STATUS " -> To enable OpenMP on macOS, please install libomp from Homebrew") +endif() + set(ADDITIONAL_CXX_FLAGS "-Wall -Wno-unused-result -Wno-unknown-pragmas -Wfatal-errors -Wno-deprecated-declarations -Wno-write-strings") set(ADDITIONAL_C_FLAGS "-Wall -Wno-unused-result -Wno-unknown-pragmas -Wfatal-errors -Wno-deprecated-declarations -Wno-write-strings")