Grouped Gemm ckProfiler hotfix (#313)

* add setWorkspace in profiler * fix

Grouped Gemm ckProfiler hotfix (#313)
* add setWorkspace in profiler * fix
ab6c82c9 · zjing14 · GitHub · 93c99f3d · ab6c82c9
Unverified Commit ab6c82c9 authored Jun 30, 2022 by zjing14 Committed by GitHub Jun 30, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

profiler/include/profile_grouped_gemm_impl.hpp profiler/include/profile_grouped_gemm_impl.hpp +4 -0

No files found.
--- a/profiler/include/profile_grouped_gemm_impl.hpp
+++ b/profiler/include/profile_grouped_gemm_impl.hpp
@@ -232,6 +232,10 @@ void profile_grouped_gemm_impl(int do_verification,

        auto invoker_ptr = gemm_ptr->MakeInvokerPointer();

+        DeviceMem gemm_desc_workspace(gemm_ptr->GetWorkSpaceSize(argument_ptr.get()));
+
+        gemm_ptr->SetWorkSpacePointer(argument_ptr.get(), gemm_desc_workspace.GetDeviceBuffer());
+
        if(gemm_ptr->IsSupportedArgument(argument_ptr.get()))
        {
            std::string gemm_name = gemm_ptr->GetTypeString();