Skip to content

Commit 9999058

Browse files
authored
Update setup_linux.py
1 parent 6f69e46 commit 9999058

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lib/nms/setup_linux.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def build_extensions(self):
124124
# we're only going to use certain compiler args with nvcc and not with
125125
# gcc the implementation of this trick is in customize_compiler() below
126126
extra_compile_args={'gcc': ["-Wno-unused-function"],
127-
'nvcc': ['-arch=sm_35',
127+
'nvcc': ['-arch=compute_50',
128128
'--ptxas-options=-v',
129129
'-c',
130130
'--compiler-options',

0 commit comments

Comments
 (0)