onnxruntime
a3f17c8b - update lamb and GatherGrad kernel for ROCm EP (#7184)

Commit

4 years ago

update lamb and GatherGrad kernel for ROCm EP (#7184) With ROCm4.1, the CUDA implementation of Lamb and GatherGrad can be utilized for ROCm EP.

References

Author

weixingzhang

Parents