From 694c2eaadb0facfaf6ba8bdf8418d150b3d38d88 Mon Sep 17 00:00:00 2001 From: Astha Rai Date: Mon, 23 Jun 2025 18:42:20 +0000 Subject: [PATCH] added argument to examples --- example/01_gemm/common.hpp | 1 + example/01_gemm/run_gemm_example_streamk_v2.inc | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/example/01_gemm/common.hpp b/example/01_gemm/common.hpp index 9073ffcfc1..4104e83677 100644 --- a/example/01_gemm/common.hpp +++ b/example/01_gemm/common.hpp @@ -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 }; diff --git a/example/01_gemm/run_gemm_example_streamk_v2.inc b/example/01_gemm/run_gemm_example_streamk_v2.inc index 439dd4504f..f9f2f2cb89 100644 --- a/example/01_gemm/run_gemm_example_streamk_v2.inc +++ b/example/01_gemm/run_gemm_example_streamk_v2.inc @@ -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)) {