trunk/38d3f6fd0d6e7ea5b68ee3854e1033ff08a5c92c: [FSDP2] Separate sharded and unsharded gradient dtypes (#194434)
- PyTorch 近 90 天出现 1780 次
- PyTorch 近 90 天第 1758 次发版
- 上一次:同一天稍早 · ciflow/torchtitan/199419: [DCP] Read checkpoint items in parallel in FileSystemReader
发生了什么
Fixes #170648 feature 0 : when mp_policy.reduce_dtype is None, it follows orig model.parameters() .grad_dtype. .grad_dtype were silently ignored before this PR if orig grad_dtype is set, mp_policy.reduce_dtype is resolved to orig grad_dtype (including explicit None) if not set, mp_policy.reduce_dtype is resolved to orig param dtype this is BC breaking : mp_policy.reduce_dtype=None no longer defaults to mp_policy.par…
摘要按规则整理自下方来源原文