viable/strict/1791287563: [inductor-perf] allow fusion into template epilogue inputs. (#198809)
- PyTorch 近 90 天出现 1965 次
- PyTorch 近 90 天第 1939 次发版
- 上一次:同一天稍早 · viable/strict/1791284024
发生了什么
Template prologue fusion lets a pointwise producer be fused into the code where a template loads an input (load_input()). Inputs that a template consumes only in its epilogue, like the addmm/baddbmm bias, could not take a fused producer. A computed bias was always written to memory by a separate kernel and read back in the template epilogue: tmp = bias * 2.0 - 1.0 addmm(tmp, a, b) This PR lets producers of those inp…
摘要按规则整理自下方来源原文