diff --git a/example/ck_tile/06_rmsnorm2d/example_rmsnorm2d_fwd.cpp b/example/ck_tile/06_rmsnorm2d/example_rmsnorm2d_fwd.cpp index a1a03727ab..bb2c949015 100644 --- a/example/ck_tile/06_rmsnorm2d/example_rmsnorm2d_fwd.cpp +++ b/example/ck_tile/06_rmsnorm2d/example_rmsnorm2d_fwd.cpp @@ -65,7 +65,7 @@ bool run(const ck_tile::ArgParser& arg_parser) constexpr bool kTwoPass = true; using BlockWarps = ck_tile::sequence<2, 2>; - using BlockTile = ck_tile::sequence<2, 256>; + using BlockTile = ck_tile::sequence<2, 128>; using WarpTile = ck_tile::sequence<1, 64>; using Vector = ck_tile::sequence<1, 1>;