ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-16 19:09:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Grouped Gemm ckProfiler hotfix (#313)

Browse Source 
* add setWorkspace in profiler

* fix

[ROCm/composable_kernel commit: ab6c82c984]
This commit is contained in:
zjing14
2022-06-30 16:37:37 -05:00
committed by GitHub
parent bd01797525
commit 964f91fd15
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
profiler/include/profile_grouped_gemm_impl.hpp
View File

@@ -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();