Remove arch from adam compile options

45f030db · ngimel · GitHub · daea4188 · 45f030db
Unverified Commit 45f030db authored Oct 30, 2018 by ngimel Committed by GitHub Oct 30, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

setup.py setup.py +1 -2

No files found.
--- a/setup.py
+++ b/setup.py
@@ -37,8 +37,7 @@ if "--cuda_ext" in sys.argv:
                      sources=['apex/optimizers/csrc/fused_adam_cuda.cpp',
                               'apex/optimizers/csrc/fused_adam_cuda_kernel.cu'],
                      extra_compile_args={'cxx': ['-O3',],
-                                          'nvcc':['--gpu-architecture=sm_70', 
-                                                  '-O3', 
+                                          'nvcc':['-O3', 
                                                  '--use_fast_math']}))
    ext_modules.append(
        CUDAExtension(name='syncbn',