From 34fea2935a443d506d5c4d4b54bc93310a30c0d7 Mon Sep 17 00:00:00 2001 From: rocking Date: Fri, 16 Aug 2024 20:30:49 +0000 Subject: [PATCH] Fix unexisted attribute --- example/ck_tile/01_fmha/codegen/ops/fmha_fwd_appendkv.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd_appendkv.py b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd_appendkv.py index 7608ccf16a..560ff481e9 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd_appendkv.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd_appendkv.py @@ -37,8 +37,8 @@ using fmha_pipeline_problem_{F_idx} = ck_tile::BlockFmhaFwdAppendKVPipelineProbl typename FmhaFwdTypeConfig::KDataType, typename FmhaFwdTypeConfig::VDataType, {F_bs}, - {F_bsk}, - {F_bd}, + {F_bsk}, + {F_bd}, {F_bdv}, {F_vlayout}, fmha_trait_{F_idx}>; @@ -328,8 +328,6 @@ def get_fwd_appendkv_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> if receipt == 2: cond = dtype in ['fp16', 'bf16'] cond &= pipeline.F_vlayout == 'row' - cond &= pipeline.F_bias in ['no', 'alibi'] - cond &= pipeline.F_squant == 'f' if not cond: continue api_pool.register_traits(k.api_trait())