Skip to content

GitLab

Commit 903711c6 authored by Seth Howell's avatar Seth Howell
Browse files

setup.py: Clean up some extra prints. 


Signed-off-by: default avatarSeth Howell <sethh@nvidia.com>
parent b79ca2bb
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
setup.py
View file @ 903711c6
...@@ -34,7 +34,6 @@ if __name__ == '__main__': ...@@ -34,7 +34,6 @@ if __name__ == '__main__':
if int(os.getenv('DISABLE_SM90_FEATURES', 0)): if int(os.getenv('DISABLE_SM90_FEATURES', 0)):
# Prefer A100 # Prefer A100
print("Not using SM_90")
os.environ['TORCH_CUDA_ARCH_LIST'] = os.getenv('TORCH_CUDA_ARCH_LIST', '8.0') os.environ['TORCH_CUDA_ARCH_LIST'] = os.getenv('TORCH_CUDA_ARCH_LIST', '8.0')
# Disable some SM90 features: FP8, launch methods, and TMA # Disable some SM90 features: FP8, launch methods, and TMA
...@@ -53,7 +52,6 @@ if __name__ == '__main__': ...@@ -53,7 +52,6 @@ if __name__ == '__main__':
else: else:
# Prefer H800 series # Prefer H800 series
os.environ['TORCH_CUDA_ARCH_LIST'] = os.getenv('TORCH_CUDA_ARCH_LIST', '9.0') os.environ['TORCH_CUDA_ARCH_LIST'] = os.getenv('TORCH_CUDA_ARCH_LIST', '9.0')
print("Using SM_90")
# CUDA 12 flags # CUDA 12 flags
nvcc_flags.extend(['-rdc=true', '--ptxas-options=--register-usage-level=10']) nvcc_flags.extend(['-rdc=true', '--ptxas-options=--register-usage-level=10'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment