From 616e2337941a9532025df5199ce75a3f4dc0a084 Mon Sep 17 00:00:00 2001 From: slippedJim Date: Thu, 17 Jul 2025 15:24:19 +0800 Subject: [PATCH] update (#2519) [ROCm/composable_kernel commit: 05b65d0c7c25374bb5180f62306a4703b9ab4a2d] --- example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py | 3 --- 1 file changed, 3 deletions(-) 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 89fbcff40c..1c46df0ab8 100644 --- a/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py +++ b/example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py @@ -536,7 +536,6 @@ def get_bwd_dq_dk_dv_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> cond = dtype in ['fp16', 'bf16'] cond &= mode == "batch" cond &= dropout in ['no', 'dropout_wg32', 'dropout_wg16'] - cond &= dpad == dvpad if not cond: continue # Aiter (mha_varlen_bwd) integration @@ -544,13 +543,11 @@ def get_bwd_dq_dk_dv_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> cond = dtype in ['fp16', 'bf16'] cond &= mode == "group" cond &= dropout in ['no', 'dropout_wg32', 'dropout_wg16'] - cond &= dpad == dvpad if not cond: continue # aiter::mha_bwd C++ api integration elif receipt == 600: cond = dtype in ['fp16', 'bf16'] - cond &= dpad == dvpad if not cond: continue api_pool.register_dq_dk_dv_traits(k.api_trait())