ciflow/trunk/187952: Fix out= variant recompilations for torch.max/min/topk
- PyTorch 近 90 天出现 250 次
- PyTorch 近 90 天第 247 次发版
- 上一次:同一天稍早 · trunk/cbb5d2e8abc050fabcda61dcc95cb21d42fb4691: [torchtitan hash update] update the pinned torchtitan hash (#194670)
发生了什么
torch.max(x, dim, out=...), torch.min(..., out=...) and torch.topk(..., out=...) recompiled under torch.compile on every new input shape, while the functional variants specialize only once. With the default recompile_limit this also makes the function fall back to eager after 8 recompiles, losing compilation entirely. Root cause: the out overloads aten.max.dim_max, aten.min.dim_min and aten.topk.values had no Python…
摘要按规则整理自下方来源原文