Skip to content

GitLab

Commit 92d58a8b authored by Jing Zhang's avatar Jing Zhang
Browse files

fixed

parent 2a0592a9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
include/ck/tensor_operation/gpu/block/blockwise_gemm_wmma.hpp
View file @ 92d58a8b
...@@ -326,7 +326,7 @@ struct BlockwiseGemmWMMA ...@@ -326,7 +326,7 @@ struct BlockwiseGemmWMMA
// read A // read A
a_thread_copy_.Run( a_thread_copy_.Run(
a_block_desc_k0_m0_m1_m2_k1, a_block_desc_k0_m0_m1_m2_k1,
make_tuple(Number<k * KPack / A_KRow>{}, m0, I0, I0, I0, I0), make_tuple(Number<k * KPack / A_K1>{}, m0, I0, I0, I0, I0),
a_block_buf, a_block_buf,
a_thread_desc_, a_thread_desc_,
make_tuple(I0, m0, I0, I0, I0, I0), make_tuple(I0, m0, I0, I0, I0, I0),
...@@ -336,7 +336,7 @@ struct BlockwiseGemmWMMA ...@@ -336,7 +336,7 @@ struct BlockwiseGemmWMMA
// read B // read B
b_thread_copy_.Run( b_thread_copy_.Run(
b_block_desc_k0_n0_n1_n2_k1, b_block_desc_k0_n0_n1_n2_k1,
make_tuple(Number<k * KPack / B_KRow>{}, n0, I0, I0, I0, I0), make_tuple(Number<k * KPack / B_K1>{}, n0, I0, I0, I0, I0),
b_block_buf, b_block_buf,
b_thread_desc_, b_thread_desc_,
make_tuple(I0, n0, I0, I0, I0, I0), make_tuple(I0, n0, I0, I0, I0, I0),
...@@ -560,7 +560,6 @@ struct BlockwiseGemmWMMA ...@@ -560,7 +560,6 @@ struct BlockwiseGemmWMMA
}); });
} }
} }
#endif #endif
protected: protected:
......
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