Extract padN and RawStore

example/ck_tile/02_layernorm2d/example_layernorm2d_fwd.cpp

@@ -40,8 +40,8 @@ auto create_args(int argc, char* argv[])
         .insert("prec_sy",
                 "auto",
                 "output quant scale type, set auto will use fp32. used when fquant=1 or 2")
-        .insert("warmup", "5", "cold iter")
-        .insert("repeat", "20", "hot iter");
+        .insert("warmup", "10", "cold iter")
+        .insert("repeat", "40", "hot iter");
 
     bool result = arg_parser.parse(argc, argv);
     return std::make_tuple(result, arg_parser);
@@ -130,6 +130,8 @@ bool run(const ck_tile::ArgParser& arg_parser)
     x_residual_buf.ToDevice(x_residual_host.data());
 
     constexpr bool kTwoPass    = false;
+    constexpr bool kPadN       = true;
+    constexpr bool UseRawStore = true;
     constexpr auto kFuseAdd    = ck_tile::Layernorm2dFusedAddEnum::PRE_ADD_STORE;
     constexpr auto kFuseQuant  = ck_tile::Layernorm2dFusedQuantEnum::NO_SWEEP;
 
@@ -139,7 +141,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
     using Vector     = ck_tile::sequence<1, 8>;
 
     using Shape = ck_tile::Generic2dBlockShape<BlockTile, BlockWarps, WarpTile, Vector>;
-    using Trait = ck_tile::Layernorm2dFwdTraits<true, false, true, kTwoPass, kFuseAdd, kFuseQuant>;
+    using Trait = ck_tile::Layernorm2dFwdTraits<kPadN, false, true, kTwoPass, kFuseAdd, kFuseQuant>;
     using PipelineProblem = ck_tile::Layernorm2dFwdPipelineProblem<XDataType,
                                                                    GammaDataType,
                                                                    BetaDataType,
@@ -157,7 +159,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
     using Pipeline        = std::conditional_t<kTwoPass, TwoPassPipeline, OnePassPipeline>;
 
     using EpilogueProblem =
-        ck_tile::Default2DEpilogueProblem<ComputeDataType, YDataType, false, true, false>;
+        ck_tile::Default2DEpilogueProblem<ComputeDataType, YDataType, false, kPadN, UseRawStore>;
     using Epilogue = ck_tile::Default2DEpilogue<EpilogueProblem>;
     using Kernel   = ck_tile::Layernorm2dFwd<Pipeline, Epilogue>;