From 1ca839925a4e9d490d5ff076d2f6eb2af82bd5c4 Mon Sep 17 00:00:00 2001
From: Philip Maybank <pmaybank@amd.com>
Date: Tue, 29 Jul 2025 11:51:28 +0100
Subject: [PATCH] solve bug with missing endif

---
 .../gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp
index 2ff99395d6..dec2f6e392 100644
--- a/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp
+++ b/include/ck_tile/ops/gemm/pipeline/gemm_pipeline_agmem_bgmem_creg_v1_default_policy.hpp
@@ -401,6 +401,15 @@ struct GemmPipelineAGmemBGmemCRegV1DefaultPolicy
                                                        WarpTile::at(I1),
                                                        WarpTile::at(I2),
                                                        Problem::TransposeC>;
+#endif                                                       
+        using WarpGemm        = WarpGemmWmmaDispatcher<typename Problem::ComputeDataType,
+                                                       typename Problem::ComputeDataType,
+                                                       AccDataType,
+                                                       WarpTile::at(I0),
+                                                       WarpTile::at(I1),
+                                                       WarpTile::at(I2),
+                                                       Problem::TransposeC>;
+
         using BlockGemmPolicy = BlockGemmASmemBSmemCRegV1CustomPolicy<typename Problem::ADataType,
                                                                       typename Problem::BDataType,
                                                                       typename Problem::CDataType,