From 2d4480a1235908c74caf9cd17c96e65519afa498 Mon Sep 17 00:00:00 2001 From: rocking Date: Sat, 26 Oct 2024 10:23:20 +0000 Subject: [PATCH] Refine tile size --- example/ck_tile/06_rmsnorm2d/example_rmsnorm2d_fwd.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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>;