diff options
| author | Marcin Junczys-Dowmunt <marcinjd@microsoft.com> | 2018-12-07 05:00:10 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <marcinjd@microsoft.com> | 2018-12-07 05:00:10 +0300 |
| commit | d7a746051c0b9cf383c1cead7410549d81db077f (patch) | |
| tree | c61a716eb061ede8b5a26120d725e39acb483032 | |
| parent | b66799babd52c35bbd128644347f78162dc2bbc3 (diff) | |
disable ptxas warning
| -rw-r--r-- | CMakeLists.txt | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 68a35cfc..51449338 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -109,17 +109,17 @@ if(CUDA_FOUND) LIST(APPEND CUDA_NVCC_FLAGS -DUSE_NCCL; ) - # @TODO: Figure out how to set this to suppress PTXAS warning. - #set(CUDA8_GENCODE, "-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_61,code=sm_61") + # disables compilation for sm_30 to avoid ptxas warning... that's general Kepler support. But K80s are supported for instance by sm_35 + set(GENCODE "-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_61,code=sm_61") # sets output to build, passes cuda location from FindCUDA, sets c++ compiler to the same one CMake uses. add_custom_command(OUTPUT ${NCCL_STATIC} COMMAND make src.build - BUILDDIR="${CMAKE_CURRENT_BINARY_DIR}/nccl" - CUDA_HOME="${CUDA_TOOLKIT_ROOT_DIR}" - #CUDA8_GENCODE="$CUDA8_GENCODE" - CXX="${CMAKE_CXX_COMPILER}" - WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/src/3rd_party/nccl") + BUILDDIR=${CMAKE_CURRENT_BINARY_DIR}/nccl + CUDA_HOME=${CUDA_TOOLKIT_ROOT_DIR} + CUDA8_GENCODE=${GENCODE} + CXX=${CMAKE_CXX_COMPILER} + WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/src/3rd_party/nccl) add_custom_target(nccl_target DEPENDS ${NCCL_STATIC}) add_library(nccl STATIC IMPORTED) set_target_properties(nccl PROPERTIES IMPORTED_LOCATION ${NCCL_STATIC}) |