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 63442eef0b..23dbd5e419 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 @@ -288,7 +288,7 @@ def get_fwd_appendkv_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> if dtype in ['fp16', 'bf16']: for rope in ROPE_MAP.keys(): # FIXME: it will be very complicated if we consider all the padding cases, - # so I just use 't' on all the dimensions + # so I just use 't' for the padding flags pipelines.append(FmhaFwdAppendKVPipeline('row', 't', 't', 't', 't', rope)) pipelines.append(FmhaFwdAppendKVPipeline('col', 't', 't', 't', 't', rope)) elif dtype in ['fp8', 'bf8']: