Skip to content

GitLab

Unverified Commit 65490af6 authored by Chaitanya Sri Krishna Lolla's avatar Chaitanya Sri Krishna Lolla Committed by GitHub
Browse files

enable multi tensor apply fusedadagrad (#9)

parent f7fa414f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
setup.py
View file @ 65490af6
......@@ -152,6 +152,7 @@ if "--cuda_ext" in sys.argv:
'csrc/hip/multi_tensor_lamb_stage_1.hip',
'csrc/hip/multi_tensor_lamb_stage_2.hip',
'csrc/hip/multi_tensor_adam.hip',
'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,
......
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