From 2b34a256753dfe056d760058b0914d93cc1f03be Mon Sep 17 00:00:00 2001 From: Ye Wang Date: Fri, 6 Dec 2024 16:42:06 -0600 Subject: [PATCH] enable bias for te --- example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py b/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py index 11bcc0f570..196e55e987 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py @@ -1501,7 +1501,7 @@ def get_bwd_dq_dk_dv_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> continue elif receipt == 3: cond = dtype in ['fp16', 'bf16'] - cond &= bias in ['no', 'alibi'] + cond &= bias in ['no', 'bias', 'alibi'] cond &= dpad == dvpad cond &= deterministic == "f" if not cond: