Fix compile args, adding version_dependent_macros. (#12)

267e696d · Jeff Daily · GitHub · b2da92fc · 267e696d
Unverified Commit 267e696d authored May 20, 2020 by Jeff Daily Committed by GitHub May 20, 2020
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 5 deletions

setup.py setup.py +4 -5

No files found.
--- a/setup.py
+++ b/setup.py
@@ -155,8 +155,7 @@ if "--cuda_ext" in sys.argv:
                                       'csrc/hip/multi_tensor_adagrad.hip',
                                       'csrc/hip/multi_tensor_novograd.hip',
                                       'csrc/hip/multi_tensor_lamb.hip'],
-                              extra_compile_args={'cxx' : ['-O3'] + version_dependent_macros,
-                                                  'nvcc': []}))
+                              extra_compile_args=['-O3'] + version_dependent_macros))

        if not is_rocm_pytorch:
            ext_modules.append(