Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "7719005848cd02cf7d58dbc4bd69fce795e0523e"
Commit c2bd0148 authored by Adam Osewski's avatar Adam Osewski
Browse files

Turn on logging in grouped gemm and gemm splitk profiler

parent 1fb6a1e0
Show 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, ...@@ -165,7 +165,7 @@ bool profile_gemm_splitk_impl(int do_verification,
std::string op_name = op_ptr->GetTypeString(); std::string op_name = op_ptr->GetTypeString();
float ave_time = 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; 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, ...@@ -198,7 +198,6 @@ bool profile_grouped_gemm_impl(int do_verification,
if(gemm_ptr->IsSupportedArgument(argument_ptr.get())) if(gemm_ptr->IsSupportedArgument(argument_ptr.get()))
{ {
if(kbatch > 1) if(kbatch > 1)
{ {
using DeviceOpSplitK = using DeviceOpSplitK =
...@@ -222,7 +221,7 @@ bool profile_grouped_gemm_impl(int do_verification, ...@@ -222,7 +221,7 @@ bool profile_grouped_gemm_impl(int do_verification,
} }
float ave_time = 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) 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