We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cff6272 commit 28615e1Copy full SHA for 28615e1
setup.py
@@ -473,7 +473,7 @@ def setup_pytorch_extension() -> setuptools.Extension:
473
]
474
475
# Compiler flags
476
- cxx_flags = ["-O3"]
+ cxx_flags = ["-O3", "-fvisibility=hidden"]
477
nvcc_flags = [
478
"-O3",
479
"-gencode",
@@ -501,7 +501,6 @@ def setup_pytorch_extension() -> setuptools.Extension:
501
nvcc_flags.extend(["-gencode", "arch=compute_80,code=sm_80"])
502
if version >= (11, 8):
503
nvcc_flags.extend(["-gencode", "arch=compute_90,code=sm_90"])
504
- nvcc_flags.extend(["-gencode", "arch=compute_89,code=sm_89"])
505
nvcc_flags.extend(["-gencode", "arch=compute_100,code=sm_100"])
506
nvcc_flags.extend(["-gencode", "arch=compute_120,code=sm_120"])
507
0 commit comments