grouped_gemm_p2