trunk/ebc5cc437ec936657e95f35de3084d65a764ac6e: [MPS] Migrate `addc[mul|div]` to Metal (#196924)
- PyTorch 近 90 天出现 1543 次
- PyTorch 近 90 天第 1524 次发版
- 上一次:同一天稍早 · trunk/7da194f6c69ad90869cd35d0d850c8c59c36164a: [torchtitan hash update] update the pinned torchtitan hash (#198742)
发生了什么
Follows the precedent set by lerp and clamp: both ops now register addc{mul,div}_stub and run through the shared exec_ternary_kernel , which gains an alpha flavor for the value scalar so the family covers clamp , lerp , addcmul and addcdiv alike. Fixes the same cast error lerp had: the _cast_ ternary kernels were instantiated for the output dtype, so an out= wider than the common dtype widened the computation as wel…
摘要按规则整理自下方来源原文