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

viable/strict/1789770346: [inductor] Optimize full slice_scatter chains (#195631)

图:PyTorch Releases

发生了什么

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…

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

为什么在扩散

来源