From eaca81945e73ce461f190f470e6e30d3cf085cc7 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 11 Jun 2024 14:42:32 +0000 Subject: [PATCH] Remove unnessary tile size for fp8 --- example/ck_tile/01_fmha/generate.py | 1 - 1 file changed, 1 deletion(-) diff --git a/example/ck_tile/01_fmha/generate.py b/example/ck_tile/01_fmha/generate.py index 1e0ab127d3..32ac200a26 100644 --- a/example/ck_tile/01_fmha/generate.py +++ b/example/ck_tile/01_fmha/generate.py @@ -897,7 +897,6 @@ def get_fmha_fwd_splitkv_combine_tile_dict_from_dtype(direction : str, dtype : s } elif dtype == 'fp8' or dtype == 'bf8': return { - '32' : FmhaFwdSplitKVCombineTileSize(128, 32, -1), '64' : FmhaFwdSplitKVCombineTileSize(128, 64, -1), '128' : FmhaFwdSplitKVCombineTileSize(128, 128, -1), '256' : FmhaFwdSplitKVCombineTileSize(128, 256, -1),