Merge branch 'develop' into wavelet_model

e59daa22 · Adam Osewski · GitHub · 7e1acffc · 236bd148 · e59daa22
Unverified Commit e59daa22 authored Nov 30, 2022 by Adam Osewski Committed by GitHub Nov 30, 2022
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 7 deletions

test/gemm_split_k/gemm_split_k.cpp test/gemm_split_k/gemm_split_k.cpp +5 -7

No files found.
--- a/test/gemm_split_k/gemm_split_k.cpp
+++ b/test/gemm_split_k/gemm_split_k.cpp
@@ -226,9 +226,8 @@ int main(int argc, char* argv[])
    std::vector<gemmArgs> test_cases;
    if(argc == 1)
    {
-        test_cases = {{GemmMatrixLayout::MK_KN_MN, 3, 3, 3, 3, 3, 3, 1}};
+        test_cases = {{GemmMatrixLayout::MK_KN_MN, 1024, 1024, 1024, 1024, 1024, 1024, 2},
-        // JD: Populate with more and meaningful
+                      {GemmMatrixLayout::MK_KN_MN, 1024, 1024, 1024, 1024, 1024, 1024, 8}};
-        return 0;
    }
    else if(argc == 9)
    {
@@ -253,11 +252,10 @@ int main(int argc, char* argv[])
        printf("arg2 to 7: M, N, K, StrideA, StrideB, StrideC KBatch\n");
        return -1;
    }
+    bool error = false;
    for(const auto& kinder : test_cases)
    {
-        const auto res = test_gemm(kinder);
+        error |= test_gemm(kinder);
-        if(!res)
-            return -1;
    }
-    return 0;
+    return error ? 1 : 0;
 }