trunk/f3d603baf74daa6ac8fbf15feb37f0edb6ac9ad3: Avoid re-inferring DeepSeek recipes in the TMA heuristic (#197458)
- PyTorch: 1271 events in the last 90 days
- PyTorch: 1257th Release in the last 90 days
- Previous: earlier the same day · ciflow/trunk/197843: [INITIAL] Update
What happened
Human Note Simple fix for DeepSeek-style scaling. Agent note Prepared with AI assistance. At K=128, valid DeepSeek scale shapes [M, 1] and [1, 1] also satisfy the shape-only rowwise predicate. Re-inferring their recipes inside the already-selected main-loop heuristic therefore makes get_tile_size reject RowWise and compilation fails. Both supported main-loop recipes have 128-element K blocks. Use that invariant here…
Summary assembled by rule from the sources below