Skip to content

Commit

Permalink
Adjusted CUDA Arches (#41754)
Browse files Browse the repository at this point in the history
  • Loading branch information
jim19930609 authored Apr 14, 2022
1 parent 921a6fb commit 1c15af3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cmake/cuda.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ elseif(NEW_RELEASE_ALL)
add_definitions(-DNEW_RELEASE_ALL)
set(paddle_known_gpu_archs "35 50 52 60 61 70 75 80 86")
set(paddle_known_gpu_archs10 "35 50 52 60 61 70 75")
set(paddle_known_gpu_archs11 "35 50 60 61 70 75 80")
set(paddle_known_gpu_archs11 "50 60 61 70 75 80")
elseif(NEW_RELEASE_PYPI)
message("Using New Release Strategy - Cubin Packge")
add_definitions(-DNEW_RELEASE_PYPI)
Expand Down

0 comments on commit 1c15af3

Please sign in to comment.