grouped_gemm
