Skip to content

Commit 28615e1

Browse files
author
Steve Chan
committed
slimmer lib, remove unused SM 89
1 parent cff6272 commit 28615e1

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

setup.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,7 @@ def setup_pytorch_extension() -> setuptools.Extension:
473473
]
474474

475475
# Compiler flags
476-
cxx_flags = ["-O3"]
476+
cxx_flags = ["-O3", "-fvisibility=hidden"]
477477
nvcc_flags = [
478478
"-O3",
479479
"-gencode",
@@ -501,7 +501,6 @@ def setup_pytorch_extension() -> setuptools.Extension:
501501
nvcc_flags.extend(["-gencode", "arch=compute_80,code=sm_80"])
502502
if version >= (11, 8):
503503
nvcc_flags.extend(["-gencode", "arch=compute_90,code=sm_90"])
504-
nvcc_flags.extend(["-gencode", "arch=compute_89,code=sm_89"])
505504
nvcc_flags.extend(["-gencode", "arch=compute_100,code=sm_100"])
506505
nvcc_flags.extend(["-gencode", "arch=compute_120,code=sm_120"])
507506

0 commit comments

Comments
 (0)