fmha config update

5e303778 · aska-0096 · 82fef9e9 · 5e303778
Commit 5e303778 authored Apr 07, 2023 by aska-0096
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

example/32_batched_gemm_scale_softmax_gemm/batched_gemm_scale_softmax_gemm_permute_wmma_fp16.cpp ...emm/batched_gemm_scale_softmax_gemm_permute_wmma_fp16.cpp +3 -3

No files found.
--- a/example/32_batched_gemm_scale_softmax_gemm/batched_gemm_scale_softmax_gemm_permute_wmma_fp16.cpp
+++ b/example/32_batched_gemm_scale_softmax_gemm/batched_gemm_scale_softmax_gemm_permute_wmma_fp16.cpp
@@ -96,12 +96,12 @@ using DeviceGemmInstance =
        256,
        //      Gemm 0
        128, // MPerBlock
-        64, // LPerBlock
-        32,  // KPerBlock
+        64,  // LPerBlock
+        64,  // KPerBlock
        8,   // K1
        //      Gemm 1
        64, // NPerBlock
-        32, // LTilePerBlock
+        64, // LTilePerBlock
        8,  // L1
        16, // MPerWMMA
        16, // LPerWMMA