[HybridParallel]Support fp16 in dygraph hybrid parallel #36420
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
zhiqiu
reviewed
Oct 14, 2021
| else: | ||
| ctx.is_fw_autocast = True | ||
| ctx.amp_mode = 'O1' | ||
| ctx.is_fw_autocast = False if tracer._amp_level == 0 else True |
There was a problem hiding this comment.
use tracer._amp_level==core.AmpLevel.O0 instead of 0
| ctx.is_fw_autocast = True | ||
| ctx.amp_mode = 'O1' | ||
| ctx.is_fw_autocast = False if tracer._amp_level == 0 else True | ||
| ctx.amp_level = 'O2' if tracer._amp_level == 2 else 'O1' |
|
|
||
| train_loss = self._broadcast_final_loss() | ||
|
|
||
| with paddle.amp.auto_cast(enable=False): |
There was a problem hiding this comment.
It is ok to put guard here, but I wonder if it is needed.
There was a problem hiding this comment.
Removing guard will cause diff of precision while broadcasting train_loss. So it is necessary to put guard here.
|
|
||
| # TODO(shenliang03) Since dp allreduce in the optimizer is | ||
| # after the gradscaler, check_finite needs to synchronize global | ||
| # information. In the future, we should use check_group to speed. | ||
| self._found_inf = paddle.cast(self._found_inf, dtype="int32") |
There was a problem hiding this comment.
I think we can consider make found_if int32 or fp32 originally to avoid these casts afterward.
fuyinno4
pushed a commit
that referenced
this pull request
Oct 26, 2021
…imizer (#36707) * fix bugs in HybridParallelClipGrad of hybrid_parallel_optimizer (#36237) * fix bugs in HybridParallelClipGrad of hybrid_parallel_optimizer * update * update * fix bugs in mp_layers、pp_layers and HybridParallelClipGrad (#36144) * fix calling bug of HybridParallelClipGrad * fix bugs of HybridParallelClipGrad * add unittest of pp with HybridParallelClipGrad * fix bugs in mp_layers.py * update * fix bugs in pp_layers.py * update * [HybridParallel]Rebuild code for pipeline (#36396) * add no_sync for parameters sync * add pipeline for moe * [HybridParallel]Support fp16 in dygraph hybrid parallel (#36420) * [HybridParallel]Support fp16 in dygraph hybrid parallel * update * update * update for recompute * add unittest of pp+fp16 * add unittest of recompute+fp16 * update * modify ut * modify ut of cond (#36475) * fix bugs of ClipGradByGlobalNorm in HybridParallel (#36555) * fix bugs of ClipGradByGlobalNorm * add unittests * add unittests * [HybridParallel]fix bug of check_inf in fleet_base.py (#36651) * fix bug of check_inf * fix allreduce * support ClipGradByGlobalNorm in sharding (#36012) * support ClipGradByGlobalNorm in sharding * support ClipGradByGlobalNorm in sharding * test=allcase * Update test_linalg_cond.py * Update hybrid_parallel_util.py * Update hybrid_parallel_util.py Co-authored-by: ShenLiang <[email protected]> Co-authored-by: zhaoyingli <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Bug fixes
PR changes
Others
Describe
[HybridParallel]Support fp16 in dygraph hybrid parallel
单卡FP32与3D混合并行+recompute+FP16的loss曲线精度对比
