Skip to content

[PyTorch] Pass canonical layouts to cuDNN grouped GEMM in fused grouped MLP - #3641

Open
hwanseoc wants to merge 5 commits into
NVIDIA:mainfrom
hwanseoc:hwanseoc/glu-canonical-layout
Open

hwanseoc wants to merge 5 commits into
NVIDIA:mainfrom
hwanseoc:hwanseoc/glu-canonical-layout

[PyTorch] Keep the typed cuDNN weight views on the grouped weight

76581d8
Select commit
Loading
Failed to load commit list.