ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 14:59:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add KBatch support for gemm_ab_scale (#2740)

Browse Source 
* Add KBatch support for gemm_ab_scale

* Revert kernel parameters change

* Remove printing

* fix formatting

* fix check

* Use {} in if

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
This commit is contained in:
Sami Remes
2025-10-09 07:33:16 +01:00
committed by GitHub
parent e99356dabc
commit 9d4bfe3932
5 changed files with 34 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
profiler/include/profiler/profile_gemm_ab_scale_impl.hpp
View File

@@ -47,6 +47,7 @@ bool profile_gemm_ab_scale_impl(int do_verification,
int StrideA,
int StrideB,
int StrideE,
int KBatch,
int n_warmup,
int n_iter,
uint64_t rotating = 0)
@@ -238,6 +239,7 @@ bool profile_gemm_ab_scale_impl(int do_verification,
a_element_op,
b_element_op,
c_element_op);
op_ptr->SetKBatch(argument_ptr.get(), KBatch);
auto invoker_ptr = op_ptr->MakeInvokerPointer();