From 7dbe5e7d37f2d6e078a429c61ce009a2549f8aae Mon Sep 17 00:00:00 2001
From: Philip Maybank <pmaybank@amd.com>
Date: Tue, 29 Jul 2025 11:55:27 +0100
Subject: [PATCH] replace hard-coded WaveSize with call to get_warp_size()

---
 .../02_gemm/block_gemm_pipeline_agmem_bgmem_creg.hpp            | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/example/ck_tile/99_toy_example/02_gemm/block_gemm_pipeline_agmem_bgmem_creg.hpp b/example/ck_tile/99_toy_example/02_gemm/block_gemm_pipeline_agmem_bgmem_creg.hpp
index bcd611fc6e..ed623c3142 100644
--- a/example/ck_tile/99_toy_example/02_gemm/block_gemm_pipeline_agmem_bgmem_creg.hpp
+++ b/example/ck_tile/99_toy_example/02_gemm/block_gemm_pipeline_agmem_bgmem_creg.hpp
@@ -58,7 +58,7 @@ struct BlockGemmPipelineAGmemBGmemCReg
         constexpr index_t NPerXDL = BlockGemm::WarpGemm::kN;
         constexpr index_t KPerXDL = BlockGemm::WarpGemm::WarpGemmAttribute::Impl::kK;
 
-        constexpr index_t WaveSize = 32;
+        constexpr index_t WaveSize = get_warp_size();
         constexpr index_t WaveNumM = BlockGemm::MWarp;
         constexpr index_t WaveNumN = BlockGemm::NWarp;