Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "a4f72a314a85732ed67d5aa8d1088d207a7e0e61"
Commit ba029a15 authored by aska-0096's avatar aska-0096
Browse files

Change gridwise gemm from FIFO to KMN loop fashion

parent ef16ed47
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
View file @ ba029a15
...@@ -414,7 +414,7 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_wmma ...@@ -414,7 +414,7 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_wmma
constexpr auto KPack = math::integer_least_multiple(K1, WmmaK); constexpr auto KPack = math::integer_least_multiple(K1, WmmaK);
auto blockwise_gemm = auto blockwise_gemm =
BlockwiseGemmWMMA_k0mk1_k0nk1_m0m1m2n0n1n2m3_CShuffle_FIFO<BlockSize, BlockwiseGemmWMMA_k0mk1_k0nk1_m0m1m2n0n1n2m3_CShuffle<BlockSize,
FloatA, FloatA,
FloatB, FloatB,
FloatAcc, FloatAcc,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment