Repository navigation

feat(conversion): run aten._grouped_mm in TensorRT with a grouped GEMM kernel - #5021

Open

shoumikhin wants to merge 4 commits into

pytorch:mainfrom

shoumikhin:grouped-mm-kernel

Open

feat(conversion): run aten._grouped_mm in TensorRT with a grouped GEMM kernel#5021
shoumikhin wants to merge 4 commits into
pytorch:mainfrom
shoumikhin:grouped-mm-kernel

Commits on Nov 11, 2026
- fix(conversion): leave large aten._grouped_mm to PyTorch instead of the dense lowering
  shoumikhin
  committed
  01d51c4
Commits on Nov 12, 2026