← 返回事件
持续讨论AI发版2.14

ciflow/rocm-mi300/201029: Keep a zero exact through a low-precision rounding cast

图:PyTorch Releases

发生了什么

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.

摘要按规则整理自下方来源原文

为什么在扩散

来源