Skip to content

Commit 3ca846e

Browse files
committed
Update setup.py for the nvcc compilation options for half float.
Remove compute_35 and compute_50 because 1. they do not support half float; 2. google colab's GPU has been updated from K80 (compute_35) to T4 (compute_75).
1 parent 9c0d5db commit 3ca846e

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -267,9 +267,9 @@ def prepare_extension_options():
267267
extra_compile_args = {'gcc': get_cpp_flags()}
268268

269269
if with_cuda:
270-
cuda9_gencode = (' -gencode arch=compute_35,code=sm_35'
271-
' -gencode arch=compute_50,code=sm_50'
272-
' -gencode arch=compute_60,code=sm_60'
270+
# compute_35 and compute_50 are removed because 1. they do not support half float;
271+
# 2. google colab's GPU has been updated from K80 (compute_35) to T4 (compute_75).
272+
cuda9_gencode = (' -gencode arch=compute_60,code=sm_60'
273273
' -gencode arch=compute_70,code=sm_70')
274274
cuda10_gencode = ' -gencode arch=compute_75,code=sm_75'
275275
cuda11_gencode = ' -gencode arch=compute_80,code=sm_80'

0 commit comments

Comments
 (0)