use 4096 for dl gemm

6b870bc0 · wangshaojie6 · 2bab820b · 6b870bc0
Commit 6b870bc0 authored Jun 10, 2022 by wangshaojie6
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

example/01_gemm/gemm_dl_fp16.cpp example/01_gemm/gemm_dl_fp16.cpp +1 -1

No files found.
--- a/example/01_gemm/gemm_dl_fp16.cpp
+++ b/example/01_gemm/gemm_dl_fp16.cpp
@@ -61,7 +61,7 @@ int main(int argc, char* argv[])
    bool time_kernel     = false;
    // GEMM shape
-    ck::index_t M = 3840;
+    ck::index_t M = 4096;
    ck::index_t N = 4096;
    ck::index_t K = 4096;