Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[Build] Add a reasonable default for CMAKE_CUDA_COMPILER in *nix #17293

Merged
merged 5 commits into from
Feb 25, 2020
Merged

[Build] Add a reasonable default for CMAKE_CUDA_COMPILER in *nix #17293

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
3e9f244
[Build] Add a reasonable default for CMAKE_CUDA_COMPILER in *nix
larroy Jan 13, 2020
ea2734e
CR
larroy Jan 16, 2020
1a4a2e6
CR
larroy Feb 4, 2020
d92f769
Update as per CR comments
larroy Feb 24, 2020
cce5732
include(CheckLanguage)
leezu Feb 25, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -87,6 +87,12 @@ message(STATUS "CMake version '${CMAKE_VERSION}' using generator '${CMAKE_GENERA
project(mxnet C CXX)
if(USE_CUDA)
cmake_minimum_required(VERSION 3.13.2) # CUDA 10 (Turing) detection available starting 3.13.2
include(CheckLanguage)
check_language(CUDA)
if (NOT CMAKE_CUDA_COMPILER_LOADED AND UNIX AND EXISTS "/usr/local/cuda/bin/nvcc")
set(CMAKE_CUDA_COMPILER "/usr/local/cuda/bin/nvcc")
larroy marked this conversation as resolved.
Show resolved Hide resolved
message(WARNING "CMAKE_CUDA_COMPILER guessed: ${CMAKE_CUDA_COMPILER}")
endif()
enable_language(CUDA)
set(CMAKE_CUDA_STANDARD 11)
include(CheckCXXCompilerFlag)
Expand Down Expand Up @@ -592,7 +598,6 @@ if(USE_CUDA)

string(REPLACE ";" " " CUDA_ARCH_FLAGS_SPACES "${CUDA_ARCH_FLAGS}")


find_package(CUDAToolkit REQUIRED cublas cufft cusolver curand
OPTIONAL_COMPONENTS nvToolsExt nvrtc)

Expand Down