From 1a013411c3691535a514e99b88f1c1a993714f7c Mon Sep 17 00:00:00 2001 From: niboshi Date: Wed, 13 Nov 2019 16:50:02 +0000 Subject: [PATCH] Fix code grouping in CMakeLists.txt --- chainerx_cc/CMakeLists.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/chainerx_cc/CMakeLists.txt b/chainerx_cc/CMakeLists.txt index a378b5224464..a26d6721e9fe 100644 --- a/chainerx_cc/CMakeLists.txt +++ b/chainerx_cc/CMakeLists.txt @@ -149,12 +149,13 @@ endif() if(${CHAINERX_BUILD_CUDA}) find_package(CUDA REQUIRED QUIET) + add_definitions(-DCHAINERX_ENABLE_CUDA=1) + list(APPEND CUDA_NVCC_FLAGS "-std=c++${CMAKE_CXX_STANDARD}") # CUDA_cublas_device_LIBRARY is required for CUDA > 9.2 with cmake < to 3.12.2. # This is because cublas_device was deprecated in CUDA 9.2, but FindCUDA supported it in 3.12.2. set(CUDA_cublas_device_LIBRARY ${CUDA_LIBRARIES}) - add_definitions(-DCHAINERX_ENABLE_CUDA=1) # If CHAINERX_CUDNN_USE_CUPY is set, retrieve cuDNN files from CuPy installation. if(CHAINERX_CUDNN_USE_CUPY)