What is the difference with GroupedGemm of Cutlass

[GuWei007]

https://github.com/NVIDIA/cutlass/blob/main/python/cutlass/op/gemm_grouped.py#L71

[tgale96]

Hi! This is a PyTorch wrapper around CUTLASS grouped GEMM kernels :)

[sleepwalker2017]

Hi! This is a PyTorch wrapper around CUTLASS grouped GEMM kernels :)Hi! This is a PyTorch wrapper around CUTLASS grouped GEMM kernels :)

Hi, the code seems to use the same configuration for all shapes, right?

How is the performance? Does it need tuning?

Thank you!