diff --git a/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.hpp b/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.hpp index 458e7ba643..7e482989d4 100644 --- a/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.hpp +++ b/example/ck_tile/18_flatmm/mixed_prec/a16w4_moe_flatmm.hpp @@ -13,7 +13,11 @@ // GEMM config with 16x16 warp tile struct A16W4_FlatmmConfig16 { - static constexpr ck_tile::index_t M_Tile = 128; +#if defined(__gfx950__) + static constexpr ck_tile::index_t M_Tile = 256; +#else + static constexpr ck_tile::index_t M_Tile = 64; +#endif static constexpr ck_tile::index_t N_Tile = 256; static constexpr ck_tile::index_t K_Tile = 256;