- 09 5月, 2023 1 次提交
-
-
由 zhangkaihuo 提交于
-
- 08 5月, 2023 1 次提交
-
-
由 Zhang Ting 提交于
-
- 27 4月, 2023 1 次提交
-
-
由 Zhang Ting 提交于
* support OD level and skip dynamic loss scaling for bf16
-
- 24 4月, 2023 3 次提交
-
-
由 Yiqun Liu 提交于
[AMP] Allow to enable multi_precision through paddle.static.amp.decorate and add documents for some apis. (#53012) * Add document for some apis. test=docs_preview * Allow to set master_weight in paddle.static.amp.decorate. * Polish codes and add unittest. * Refine docs. * Remove the repetitive function.
-
由 Zhang Ting 提交于
-
由 Zhang Ting 提交于
* support promote dtype for static amp training * unify o1 and o2 * update for unittest * fix op_role * add use_promote arg * fix doc * add promote unittest * polish unittests * fix controflow and test
-
- 18 4月, 2023 2 次提交
-
-
由 Zhang Ting 提交于
-
由 Yiqun Liu 提交于
* Implement a common AmpTestBase. * Support overload of decorate. * Change the ignore list of flake and fix an error.
-
- 14 4月, 2023 1 次提交
-
-
由 Yiqun Liu 提交于
* Unify the static amp codes of fp16 and bf16. * Polish apis and add unittest. * Add operator stats collecting tools for program. * Add the check of number of bloat16 operators in unittest. * Add warning for operator not supported for amp. * Add testing of BF16 O1 and O2.
-
- 13 4月, 2023 1 次提交
-
-
由 liuruyan 提交于
-
- 12 4月, 2023 1 次提交
-
-
由 liuruyan 提交于
-
- 10 4月, 2023 1 次提交
-
-
由 Zhang Ting 提交于
* support set master_grad * move register_hook to auto_cast * update unittest * fix fp16 test * update for review comments
-
- 03 4月, 2023 1 次提交
-
-
由 engineer1109 提交于
-
- 30 3月, 2023 1 次提交
-
-
由 Yiqun Liu 提交于
* [AMP] Add python API for collecting operator stats. * Fix import and polish codes. * Add more unittest. * Add doc for the new APIs.
-