diff --git a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py index 5697af1ba8..139a7f6f1a 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py @@ -702,7 +702,6 @@ def get_fwd_blobs(kernel_filter : Optional[str], receipt, optdim_list, mask_impl cond = dtype in ['fp8'] cond &= pipeline.F_vlayout == 'row' cond &= hdim == 128 - cond &= pipeline.F_squant == 't' if not cond: continue