Ma, Guokai 0f5406323c [CPU] FusedAdam and CPU training support (#3991) | 1 年之前 | |
---|---|---|
.. | ||
cpu_adam.cpp | 0f5406323c [CPU] FusedAdam and CPU training support (#3991) | 1 年之前 |
cpu_adam_impl.cpp | 0f5406323c [CPU] FusedAdam and CPU training support (#3991) | 1 年之前 |
fused_adam_frontend.cpp | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 年之前 |
multi_tensor_adam.cu | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 年之前 |
multi_tensor_apply.cuh | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 年之前 |