Skip to content

Commit

Permalink
add fix for CUDA 10
Browse files Browse the repository at this point in the history
  • Loading branch information
@soumith
soumith committed Dec 6, 2018
1 parent 524574a commit db5d313
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions aten/src/ATen/CMakeLists.txt
View file
Expand Up @@ -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()
Expand Down

2 comments on commit db5d313

@metaluga145
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

CUDA 10 supports Turing architecture with compute capability 7.5. Why it was not included in the release?

@soumith
Copy link
Member Author

@soumith soumith commented on db5d313 Dec 9, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

cc 7.5 is binary compatible with 7.0.
we didn't compile the cufft fakelink with 7.5, but the rest of the cuda bits actually do compile and include 7.5 architecture sass

Please sign in to comment.