/grouped_gemm

PyTorch bindings for CUTLASS grouped GEMM.

Primary LanguageCudaApache License 2.0Apache-2.0

Stargazers