[AI Codemod][PerfAICT] Reuse batch-dim views in bmm_out_or_baddbmm_ to cut select dispatch overhead (#198570)
Differential Revision: D118252827 Pull Request resolved: https://github.com/pytorch/pytorch/pull/198570 Approved by: https://github.com/Skylion007
Y
Yedidya Feldblum committed
85d04a64ce27b93e528c0ae28c0f5313ef68e7a7
Parent: 38cca96
Committed by PyTorch MergeBot <pytorchmergebot@users.noreply.github.com>
on 10/1/2026, 1:03:17 AM