change default parameter

622ce8f4 · letaoqin · 06e287e8 · 622ce8f4
Commit 622ce8f4 authored Oct 11, 2024 by letaoqin
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

example/66_gemm_bias_activation/gemm_bias_add_xdl_fp16.cpp example/66_gemm_bias_activation/gemm_bias_add_xdl_fp16.cpp +3 -3

No files found.
--- a/example/66_gemm_bias_activation/gemm_bias_add_xdl_fp16.cpp
+++ b/example/66_gemm_bias_activation/gemm_bias_add_xdl_fp16.cpp
@@ -140,9 +140,9 @@ int main(int argc, char* argv[])
    bool time_kernel     = true;

    // GEMM shape
-    ck::index_t M = 16;
-    ck::index_t N = 16;
-    ck::index_t K = 64;
+    ck::index_t M = 64;
+    ck::index_t N = 3072;
+    ck::index_t K = 768;

    if(argc == 1)
    {