Skip to content

GitLab

Commit ca2a0ebd authored by danyao12's avatar danyao12
Browse files

Merge branch 'ck_tile/fa_bwd_opt' of https://github.com/ROCm/composable_kernel... 

Merge branch 'ck_tile/fa_bwd_opt' of https://github.com/ROCm/composable_kernel into ck_tile/fa_bwd_opt
parents dcc3593f 85ec3c75
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
example/ck_tile/01_fmha/codegen/ops/fmha_bwd.py
View file @ ca2a0ebd
...@@ -490,6 +490,7 @@ def get_bwd_dq_dk_dv_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> ...@@ -490,6 +490,7 @@ def get_bwd_dq_dk_dv_blobs(kernel_filter : Optional[str], receipt, mask_impl) ->
if receipt == 2: if receipt == 2:
cond = dtype in ['fp16', 'bf16'] cond = dtype in ['fp16', 'bf16']
cond &= bias in ['no', 'alibi'] cond &= bias in ['no', 'alibi']
cond &= dropout in ['no', 'dropout_wg32', 'dropout_wg16']
if not cond: if not cond:
continue continue
api_pool.register_dq_dk_dv_traits(k.api_trait()) api_pool.register_dq_dk_dv_traits(k.api_trait())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment