Liran Bachar 69af361167 CPUAdam fp16 and bf16 support (#5409) | 5 月之前 | |
---|---|---|
.. | ||
cpu_adam.cpp | 69af361167 CPUAdam fp16 and bf16 support (#5409) | 5 月之前 |
cpu_adam_impl.cpp | 69af361167 CPUAdam fp16 and bf16 support (#5409) | 5 月之前 |
fused_adam_frontend.cpp | b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) | 1 年之前 |
multi_tensor_adam.cu | 9b6ef9e1f0 64bit indexing fused adam (#5187) | 6 月之前 |
multi_tensor_apply.cuh | 9b6ef9e1f0 64bit indexing fused adam (#5187) | 6 月之前 |