Wei Fu 9b6ef9e1f0 64bit indexing fused adam (#5187) | 6 months ago | |
---|---|---|
.. | ||
cpu_adam.cpp | 0f5406323c [CPU] FusedAdam and CPU training support (#3991) | 1 year ago |
cpu_adam_impl.cpp | c1ba6a104f [CANN] Support cpu offload optimizer for Ascend NPU (#4568) | 11 months ago |
fused_adam_frontend.cpp | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 year ago |
multi_tensor_adam.cu | 9b6ef9e1f0 64bit indexing fused adam (#5187) | 6 months ago |
multi_tensor_apply.cuh | 9b6ef9e1f0 64bit indexing fused adam (#5187) | 6 months ago |