Add KBatch info to profiler output summary log.

e9fd26c8 · Adam Osewski · 25205ea5 · e9fd26c8
Commit e9fd26c8 authored May 12, 2023 by Adam Osewski
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

profiler/include/profiler/profile_gemm_splitk_impl.hpp profiler/include/profiler/profile_gemm_splitk_impl.hpp +3 -3

No files found.
--- a/profiler/include/profiler/profile_gemm_splitk_impl.hpp
+++ b/profiler/include/profiler/profile_gemm_splitk_impl.hpp
@@ -246,9 +246,9 @@ bool profile_gemm_splitk_impl(int do_verification,
    }

    std::cout << " M = " << M << " N = " << N << " K = " << K << " StrideA = " << StrideA
-              << " StrideB = " << StrideB << " StrideC = " << StrideC << " : " << best_ave_time
-              << " ms, " << best_tflops << " TFlops, " << best_gb_per_sec << " GB/s, "
-              << best_op_name << std::endl;
+              << " StrideB = " << StrideB << " StrideC = " << StrideC << " KBatch = " << KBatch
+              << " : " << best_ave_time << " ms, " << best_tflops << " TFlops, " << best_gb_per_sec
+              << " GB/s, " << best_op_name << std::endl;

    return pass;
 }