use 7*13 group

e392ce24 · wangshaojie6 · 870a2482 · e392ce24
Commit e392ce24 authored Sep 14, 2022 by wangshaojie6
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

example/32_batched_gemm_scale_softmax_gemm/batched_gemm_lower_triangle_scale_softmax_gemm_permute_xdl_fp16.cpp ...mm_lower_triangle_scale_softmax_gemm_permute_xdl_fp16.cpp +2 -2

No files found.
--- a/example/32_batched_gemm_scale_softmax_gemm/batched_gemm_lower_triangle_scale_softmax_gemm_permute_xdl_fp16.cpp
+++ b/example/32_batched_gemm_scale_softmax_gemm/batched_gemm_lower_triangle_scale_softmax_gemm_permute_xdl_fp16.cpp
@@ -164,8 +164,8 @@ int main(int argc, char* argv[])
    // Output shape C[G0, M, G1, O]. Batch dim, outer dim, inner dim must match GEMM shape
    // C_g0_g1_m_o = reshape(C_g_m_o, [g0, g1, m, o])
    // C_g0_m_g1_o = permute(C_g0_g1_m_o, [0, 2, 1, 3])
-    ck::index_t G0 = 2;
-    ck::index_t G1 = 3;
+    ck::index_t G0 = 7;
+    ck::index_t G1 = 13;

    if(argc == 1)
    {