Skip to content

[Common][PyTorch] Add strided batched GEMM in BF16/MXFP8 - #3160

Open
yaox12 wants to merge 10 commits into
NVIDIA:mainfrom
yaox12:xiny/batch_gemm
Open

[Common][PyTorch] Add strided batched GEMM in BF16/MXFP8#3160
yaox12 wants to merge 10 commits into
NVIDIA:mainfrom
yaox12:xiny/batch_gemm

Pass 3D operands from BatchedLinear to strided batched GEMM

bb59ce8
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Oct 21, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO