ciflow/torchtitan/201029: Keep a zero exact through a low-precision rounding cast
- PyTorch 近 90 天出现 2487 次
- PyTorch 近 90 天第 2454 次发版
- 上一次:同一天稍早 · ciflow/torchtitan/201319: Update
发生了什么
Rounding a constant zero to float16 or bfloat16 gives exactly zero, so the prologue zero-mask analysis can keep it, as it did before #197132 . A prologue whose only low-precision op is a rounding cast then keeps its masked zeros and gets no extra tl.where, matching 2.14's generated code. Other values still become unknown. The end-to-end test now checks the generated code for the mask, the same way test_upcast does.
摘要按规则整理自下方来源原文