From db5d3131d16f57abd4f13d3f4b885d5f67bf6644 Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Thu, 6 Dec 2018 15:44:56 -0800 Subject: [PATCH] add fix for CUDA 10 --- aten/src/ATen/CMakeLists.txt | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/aten/src/ATen/CMakeLists.txt b/aten/src/ATen/CMakeLists.txt index aeb5b7866cf3dd..07b3c106be1199 100644 --- a/aten/src/ATen/CMakeLists.txt +++ b/aten/src/ATen/CMakeLists.txt @@ -206,6 +206,12 @@ IF(USE_CUDA AND NOT USE_ROCM) --generate-code arch=compute_50,code=sm_50 --generate-code arch=compute_60,code=sm_60 --generate-code arch=compute_70,code=sm_70) + elseif(${CUDA_VERSION_MAJOR} EQUAL "10") + SET(CUFFT_FAKELINK_OPTIONS + --generate-code arch=compute_35,code=sm_35 + --generate-code arch=compute_50,code=sm_50 + --generate-code arch=compute_60,code=sm_60 + --generate-code arch=compute_70,code=sm_70) else() MESSAGE(FATAL_ERROR "Unhandled major cuda version ${CUDA_VERSION_MAJOR}") endif()