Skip to content

GitLab

Commit c2bd0148 authored by Adam Osewski's avatar Adam Osewski
Browse files

Turn on logging in grouped gemm and gemm splitk profiler

parent 1fb6a1e0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
profiler/include/profiler/profile_gemm_splitk_impl.hpp
View file @ c2bd0148
......@@ -165,7 +165,7 @@ bool profile_gemm_splitk_impl(int do_verification,
std::string op_name = op_ptr->GetTypeString();
float ave_time =
invoker_ptr->Run(argument_ptr.get(), StreamConfig{nullptr, time_kernel});
invoker_ptr->Run(argument_ptr.get(), StreamConfig{nullptr, time_kernel, 1});
std::size_t flop = std::size_t(2) * M * N * K;
......
profiler/include/profiler/profile_grouped_gemm_impl.hpp
View file @ c2bd0148
......@@ -198,7 +198,6 @@ bool profile_grouped_gemm_impl(int do_verification,
if(gemm_ptr->IsSupportedArgument(argument_ptr.get()))
{
if(kbatch > 1)
{
using DeviceOpSplitK =
......@@ -222,7 +221,7 @@ bool profile_grouped_gemm_impl(int do_verification,
}
float ave_time =
invoker_ptr->Run(argument_ptr.get(), StreamConfig{nullptr, time_kernel});
invoker_ptr->Run(argument_ptr.get(), StreamConfig{nullptr, time_kernel, 1});
if(time_kernel)
{
......
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