Skip to content

GitLab

Unverified Commit 267e696d authored by Jeff Daily's avatar Jeff Daily Committed by GitHub
Browse files

Fix compile args, adding version_dependent_macros. (#12)

parent b2da92fc
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 5 deletions
setup.py
View file @ 267e696d
...@@ -155,8 +155,7 @@ if "--cuda_ext" in sys.argv: ...@@ -155,8 +155,7 @@ if "--cuda_ext" in sys.argv:
'csrc/hip/multi_tensor_adagrad.hip', 'csrc/hip/multi_tensor_adagrad.hip',
'csrc/hip/multi_tensor_novograd.hip', 'csrc/hip/multi_tensor_novograd.hip',
'csrc/hip/multi_tensor_lamb.hip'], 'csrc/hip/multi_tensor_lamb.hip'],
extra_compile_args={'cxx' : ['-O3'] + version_dependent_macros, extra_compile_args=['-O3'] + version_dependent_macros))
'nvcc': []}))
if not is_rocm_pytorch: if not is_rocm_pytorch:
ext_modules.append( ext_modules.append(
......
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