DeepSpeed
6996f393 - make ds_adam_step_plus_copy() works on AMD GPUs

Commit

2 years ago

make ds_adam_step_plus_copy() works on AMD GPUs

References

guanhua/rocm-cpu-adam

Author

GuanhuaWang

GuanhuaWang

Parents

Loading