Skip to content

GitLab

"...composable_kernel.git" did not exist on "a17b0414864c11c7167d2427f9a69ffce9171465"
Commit cee92951 authored by Jing Zhang's avatar Jing Zhang
Browse files

changing layout

parent 92c615b6
Hide whitespace changes
Inline Side-by-side
Showing with 43 additions and 6 deletions
example/15_grouped_gemm/grouped_gemm_transpose_xdl_fp16.cpp
View file @ cee92951
......@@ -106,13 +106,50 @@ int main(int argc, char* argv[])
int StrideA = K;
int StrideB = K;
int StrideM0 = S * NumHead * HeadDim;
int StrideM1 = HeadDim;
int StrideN0 = S * HeadDim;
int StrideN1 = 1;
if(i % 2 == 0)
{
gemm_descs.push_back(
{M, N, K, StrideA, StrideB, M0, M1, N0, N1, StrideM0, StrideM1, StrideN0, StrideN1});
int StrideM0 = S * NumHead * HeadDim;
int StrideM1 = 1;
int StrideN0 = S * HeadDim;
int StrideN1 = S;
gemm_descs.push_back({M,
N,
K,
StrideA,
StrideB,
M0,
M1,
N0,
N1,
StrideM0,
StrideM1,
StrideN0,
StrideN1});
}
else
{
int StrideM0 = S * NumHead * HeadDim;
int StrideM1 = HeadDim;
int StrideN0 = S * HeadDim;
int StrideN1 = 1;
gemm_descs.push_back({M,
N,
K,
StrideA,
StrideB,
M0,
M1,
N0,
N1,
StrideM0,
StrideM1,
StrideN0,
StrideN1});
}
}
auto f_host_tensor_descriptor =
......
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