[CK_TILE] Adjust kBlockSize of reduce example for better perf (#1779)

* Observed a 2x perf improvement with kBlockSize = 256 * Using 512 threads may lead to redundant computations
ROCm · Jan 13, 2025 · 0b8f117 · 0b8f117
1 parent 3d50f57
commit 0b8f117
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/ck_tile/05_reduce/reduce.cpp b/example/ck_tile/05_reduce/reduce.cpp
@@ -52,7 +52,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
     // using WarpTile   = ck_tile::sequence<1, 512>;
     // using Vector = ck_tile::sequence<1, 8>;
 
-    constexpr ck_tile::index_t kBlockSize  = 512;
+    constexpr ck_tile::index_t kBlockSize  = 256;
     constexpr ck_tile::index_t kBlockPerCu = 1;
     ck_tile::index_t kGridSize             = (m / BlockTile::at(ck_tile::number<0>{}));
     std::cout << "grid size " << kGridSize << std::endl;