From 964f91fd156929b809aabb0e619af83a1f5fb315 Mon Sep 17 00:00:00 2001
From: zjing14 <zhangjing14@gmail.com>
Date: Thu, 30 Jun 2022 16:37:37 -0500
Subject: [PATCH] Grouped Gemm ckProfiler hotfix (#313)

* add setWorkspace in profiler

* fix

[ROCm/composable_kernel commit: ab6c82c984fe1a958e537d7c3f78ec8a3a9bcb2d]
---
 profiler/include/profile_grouped_gemm_impl.hpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/profiler/include/profile_grouped_gemm_impl.hpp b/profiler/include/profile_grouped_gemm_impl.hpp
index f3c0082452..92f45eccee 100644
--- 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();