From dc2d4f39d1705e89dd77020ef45a7cad25fd361c Mon Sep 17 00:00:00 2001
From: yadaish <yadai@amd.com>
Date: Tue, 25 Nov 2025 06:06:46 +0000
Subject: [PATCH] enable int for moe ffn

---
 .../18_flatmm/mixed_prec/a16w4_moe_flatmm.cpp    | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.cpp b/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.cpp
index 709c772f6a..ba9f3ea106 100644
--- a/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.cpp
+++ b/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.cpp
@@ -484,6 +484,22 @@ int run_a16w4_moe_flatmm_example(int argc, char* argv[])
                                                                ck_tile::MoeFlatmmKind::kFFN_gemm2>(
                     argc, argv, Row{}, Col{}, Row{});
             }
+            else if(mixed_prec == "fp16xint4")
+            {
+                return run_a16w4_moe_gemm_example_with_layouts<ck_tile::half_t,
+                                                               ck_tile::pk_fp4_t,
+                                                               FlatmmConfig,
+                                                               ck_tile::MoeFlatmmKind::kFFN_gemm2>(
+                    argc, argv, Row{}, Col{}, Row{});
+            }
+            else if(mixed_prec == "bf16xint4")
+            {
+                return run_a16w4_moe_gemm_example_with_layouts<ck_tile::bfloat16_t,
+                                                               ck_tile::pk_fp4_t,
+                                                               FlatmmConfig,
+                                                               ck_tile::MoeFlatmmKind::kFFN_gemm2>(
+                    argc, argv, Row{}, Col{}, Row{});
+            }
             else
             {
                 throw std::runtime_error("Unsupported precision type for gemm2!");