ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-18 17:48:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

CkProfiler Fix

Browse Source
This commit is contained in:
ozturkosu
2025-06-12 01:06:23 -04:00
parent 6e4ddf54ef
commit 50f448b291
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
profiler/include/profiler/profile_gemm_universal_streamk_impl.hpp
View File

@@ -181,6 +181,7 @@ bool profile_gemm_universal_streamk_impl(int do_verification,
auto grid_size_curr = grid_size_list[i];
index_t streamk_sel_curr = streamk_sel_list[j];
printf("streamk_sel_curr=%0d\n", streamk_sel_curr);
auto argument_ptr = op_ptr->MakeArgumentPointer(
static_cast<ADataType*>(a_device_buf.GetDeviceBuffer()),
static_cast<BDataType*>(b_device_buf.GetDeviceBuffer()),
@@ -191,11 +192,10 @@ bool profile_gemm_universal_streamk_impl(int do_verification,
StrideA,
StrideB,
StrideC,
streamk_sel_curr,
grid_size_curr,
a_element_op,
b_element_op,
c_element_op);
c_element_op,
streamk_sel_curr); // NumSKBlocks parameter
auto invoker_ptr = op_ptr->MakeInvokerPointer();