ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

added argument to examples

Browse Source
This commit is contained in:
Astha Rai
2025-06-23 18:42:20 +00:00
parent 3dc5dcb8a1
commit 694c2eaadb
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
example/01_gemm/common.hpp
View File

@@ -57,6 +57,7 @@ struct ProblemSizeStreamK_universal final
ck::index_t StrideB = -1;
ck::index_t StrideC = -1;
ck::index_t NumSKBlocks = -1; // number of stream-k blocks
ck::index_t Grid_size = -1; // defaults to max occupancy
ck::index_t Streamk_sel = 1; // defaults to 1-tile SK
};

3
example/01_gemm/run_gemm_example_streamk_v2.inc
View File

@@ -150,8 +150,7 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
StrideC,
a_element_op,
b_element_op,
c_element_op
/**problem_size.NumSKBlocks**/); // TODO: add NumSKBlocks arg into Problem?
c_element_op problem_size.NumSKBlocks); // TODO: add NumSKBlocks arg into Problem?
if(!gemm.IsSupportedArgument(argument))
{