From c7d9c80b77320aec6559222bed7d47adcaefe4e3 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Wed, 12 Jun 2024 05:02:48 +0000 Subject: [PATCH] Default use kMaxSplits=64 in generate.py --- example/ck_tile/01_fmha/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/01_fmha/generate.py b/example/ck_tile/01_fmha/generate.py index 6d667ff1d8..a9734fe951 100644 --- a/example/ck_tile/01_fmha/generate.py +++ b/example/ck_tile/01_fmha/generate.py @@ -256,7 +256,7 @@ using fmha_trait_{F_idx} = ck_tile::TileFmhaFwdSplitKVCombineTraits<{F_spad}, {F_dvpad}, {F_lse}, {F_squant}, - 16, + 64, {F_occupancy}>; using fmha_pipeline_problem_{F_idx} = ck_tile::BlockFmhaSplitKVCombinePipelineProblem<