Commit 519146b7 authored by Jing Zhang's avatar Jing Zhang
Browse files

fixed ci

parent 08e0781e
...@@ -79,7 +79,7 @@ bool TestGroupedGemm(DeviceGroupedGemmPtr_& groupedGemmPtr) ...@@ -79,7 +79,7 @@ bool TestGroupedGemm(DeviceGroupedGemmPtr_& groupedGemmPtr)
{ {
int M = 256 + 256 * i; int M = 256 + 256 * i;
int N = 128 + 128 * i; int N = 128 + 128 * i;
int K = 64 + 64 * i; int K = 128 + 64 * i;
int AStride = std::is_same<ck::tensor_layout::gemm::RowMajor, ALayout>::value ? K : M; int AStride = std::is_same<ck::tensor_layout::gemm::RowMajor, ALayout>::value ? K : M;
int BStride = std::is_same<ck::tensor_layout::gemm::RowMajor, BLayout>::value ? N : K; int BStride = std::is_same<ck::tensor_layout::gemm::RowMajor, BLayout>::value ? N : K;
......
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