Change gridwise gemm from FIFO to KMN loop fashion

ba029a15 · aska-0096 · ef16ed47 · ba029a15
Commit ba029a15 authored Mar 01, 2023 by aska-0096
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
@@ -414,7 +414,7 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_wmma
        constexpr auto KPack = math::integer_least_multiple(K1, WmmaK);

        auto blockwise_gemm =
-            BlockwiseGemmWMMA_k0mk1_k0nk1_m0m1m2n0n1n2m3_CShuffle_FIFO<BlockSize,
+            BlockwiseGemmWMMA_k0mk1_k0nk1_m0m1m2n0n1n2m3_CShuffle<BlockSize,
                                                         FloatA,
                                                         FloatB,
                                                         FloatAcc,