viable/strict/1789770346: [inductor] Optimize full slice_scatter chains (#195631)
- PyTorch 近 90 天出现 1165 次
- PyTorch 近 90 天第 1151 次发版
- 上一次:同一天稍早 · viable/strict/1789766664
发生了什么
The following summary was prepared with assistance from Codex and reviewed by the author. Rewrite slice_scatter chains that fill an entire tensor into peak memory efficient cat or copy_. Functionalizing chunking results in a chain of slice_scatter nodes, holding full-sized intermediate by the end of the chain. The main goal of this pass is to make the "chunking" region the most peak memory efficient, as this is the…
摘要按规则整理自下方来源原文