Skip to content

GitLab

Commit e392ce24 authored by wangshaojie6's avatar wangshaojie6
Browse files

use 7*13 group

parent 870a2482
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
example/32_batched_gemm_scale_softmax_gemm/batched_gemm_lower_triangle_scale_softmax_gemm_permute_xdl_fp16.cpp
View file @ e392ce24
...@@ -164,8 +164,8 @@ int main(int argc, char* argv[]) ...@@ -164,8 +164,8 @@ int main(int argc, char* argv[])
// Output shape C[G0, M, G1, O]. Batch dim, outer dim, inner dim must match GEMM shape // Output shape C[G0, M, G1, O]. Batch dim, outer dim, inner dim must match GEMM shape
// C_g0_g1_m_o = reshape(C_g_m_o, [g0, g1, m, o]) // C_g0_g1_m_o = reshape(C_g_m_o, [g0, g1, m, o])
// C_g0_m_g1_o = permute(C_g0_g1_m_o, [0, 2, 1, 3]) // C_g0_m_g1_o = permute(C_g0_g1_m_o, [0, 2, 1, 3])
ck::index_t G0 = 2; ck::index_t G0 = 7;
ck::index_t G1 = 3; ck::index_t G1 = 13;
if(argc == 1) if(argc == 1)
{ {
......
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