fixed ci

519146b7 · Jing Zhang · 08e0781e · 519146b7
Commit 519146b7 authored Mar 22, 2022 by Jing Zhang
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/grouped_gemm/grouped_gemm_fp16.cpp test/grouped_gemm/grouped_gemm_fp16.cpp +1 -1

No files found.
--- a/test/grouped_gemm/grouped_gemm_fp16.cpp
+++ b/test/grouped_gemm/grouped_gemm_fp16.cpp
@@ -79,7 +79,7 @@ bool TestGroupedGemm(DeviceGroupedGemmPtr_& groupedGemmPtr)
    {
        int M = 256 + 256 * 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 BStride = std::is_same<ck::tensor_layout::gemm::RowMajor, BLayout>::value ? N : K;