Formatting + fix K dimension for int8.

09419996 · Adam Osewski · 4d38b385 · 09419996
Commit 09419996 authored Aug 25, 2022 by Adam Osewski
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 8 deletions

example/15_grouped_gemm/run_grouped_gemm_example.inc example/15_grouped_gemm/run_grouped_gemm_example.inc +4 -8

No files found.
--- a/example/15_grouped_gemm/run_grouped_gemm_example.inc
+++ b/example/15_grouped_gemm/run_grouped_gemm_example.inc
@@ -235,11 +235,7 @@ bool run_grouped_gemm_example(int argc, char* argv[])
    {
        problem_size.Ms.push_back(256 + 256 * i);
        problem_size.Ns.push_back(128 + 128 * i);
-#ifdef BUILD_INT4_EXAMPLE
        problem_size.Ks.push_back(128 + 64 * i);
-#else
-        problem_size.Ks.push_back(64 + 64 * i);
-#endif

        problem_size.stride_As.push_back(problem_size.Ks[i]);
        problem_size.stride_Bs.push_back(problem_size.Ks[i]);