Skip to content

GitLab

Commit 6f5c8e3a authored by Andy Lugo's avatar Andy Lugo
Browse files

Remove repeated code

parent ddfc27ea
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 8 deletions
example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py
View file @ 6f5c8e3a
...@@ -494,14 +494,7 @@ def get_fwd_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> Tuple[Fm ...@@ -494,14 +494,7 @@ def get_fwd_blobs(kernel_filter : Optional[str], receipt, mask_impl) -> Tuple[Fm
cond &= pipeline.F_squant == 'f' cond &= pipeline.F_squant == 'f'
if not cond: if not cond:
continue continue
if receipt == 3: if receipt in (3, 4):
cond = dtype in ['fp16', 'bf16']
cond &= pipeline.F_vlayout == 'row'
cond &= pipeline.F_bias in ['no', 'alibi']
cond &= pipeline.F_squant == 'f'
if not cond:
continue
if receipt == 4:
cond = dtype in ['fp16', 'bf16'] cond = dtype in ['fp16', 'bf16']
cond &= pipeline.F_vlayout == 'row' cond &= pipeline.F_vlayout == 'row'
cond &= pipeline.F_bias in ['no', 'bias'] cond &= pipeline.F_bias in ['no', 'bias']
......
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