change init and mblock

This commit is contained in:
coderfeli
2025-03-19 07:58:01 +00:00
parent 30ec15bd1e
commit 79275d4be4
2 changed files with 4 additions and 4 deletions

View File

@@ -20,10 +20,10 @@ float flatmm_calc(const ck_tile::FlatmmHostArgs& args, const ck_tile::stream_con
constexpr bool kPadN = false;
constexpr bool kPadK = false;
constexpr int kBlockPerCu = 1;
constexpr int kBlockPerCu = 2;
// This part comes from the Codegen
constexpr ck_tile::index_t M_Tile = 32;
constexpr ck_tile::index_t M_Tile = 128;
constexpr ck_tile::index_t N_Tile = 128;
constexpr ck_tile::index_t K_Tile = 64;