From 86e03b8a30f9eb5fb7093917c7a6615f73a5ae81 Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Sun, 9 Dec 2018 15:52:25 -0800 Subject: [PATCH] add fix for CUDA 10 (#14971) Summary: Linux binaries-only fix for CUDA10 Pull Request resolved: https://github.com/pytorch/pytorch/pull/14971 Differential Revision: D13395932 Pulled By: soumith fbshipit-source-id: a72d6ab6b98c6c936e6391d55d2e4e45b9f1e6dd --- 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 aeb5b78..07b3c10 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() -- 2.7.4