Wenwen Qu 655e9dae40 Feat(norm)/support fused precision (#319) ... * add fused precision support for norm * refactor code * refactor code * change the granularity of hook * fix bugs if self.model is ModuleList * add dtype condition for post hook * refactor code for split group * refactor code for pre/post hook * refactor code for split group * remove fp32 hook for norm * unit tests for fused precision * add doc for fused precision * add doc for En. version * reformat docs * Update mixed_precision.rst * Update mixed_precision.po * update mixed_precision.po		2023-09-26 20:39:55 +08:00
..
test_fused_precision	Feat(norm)/support fused precision (#319)	2023-09-26 20:39:55 +08:00
test_embedding.py	feat: add unitest for model (#300)	2023-09-14 13:18:34 +08:00
test_model_internlm.py	feat: add unitest for model (#300)	2023-09-14 13:18:34 +08:00
test_norm.py	feat: add unitest for model (#300)	2023-09-14 13:18:34 +08:00