Skip to content

GitLab

"examples/community/stable_diffusion_ipex.py" did not exist on "ca1a22296d02ab44547a33c1bb7e5a4015a2f843"
Commit ccaec8ec authored by Po Yen, Chen's avatar Po Yen, Chen
Browse files

Use kN0=32 for 2wave fmha pipeline

parent e6b69a63
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
example/ck_tile/01_fmha/codegen/ops/fmha_fwd.py
View file @ ccaec8ec
...@@ -395,10 +395,10 @@ def get_fmha_fwd_tile_dict_from_dtype(dtype : str, receipt) -> Optional[dict]: ...@@ -395,10 +395,10 @@ def get_fmha_fwd_tile_dict_from_dtype(dtype : str, receipt) -> Optional[dict]:
if receipt == 3: if receipt == 3:
if dtype == 'fp16' or dtype == 'bf16': if dtype == 'fp16' or dtype == 'bf16':
return { return {
'32' : FmhaFwdTileSize(128, 64, 16, 32, 32, 32, 2, 1, 1, 32, 32, 16, -1), '32' : FmhaFwdTileSize(128, 32, 32, 32, 32, 32, 2, 1, 1, 32, 32, 16, -1),
'64' : FmhaFwdTileSize(128, 64, 32, 64, 32, 64, 4, 1, 1, 32, 32, 16, -1), '64' : FmhaFwdTileSize(128, 32, 32, 64, 32, 64, 4, 1, 1, 32, 32, 16, -1),
'128' : FmhaFwdTileSize(128, 128, 32, 128, 32, 128, 4, 1, 1, 32, 32, 16, -1), '128' : FmhaFwdTileSize(128, 32, 32, 128, 32, 128, 4, 1, 1, 32, 32, 16, -1),
'256' : FmhaFwdTileSize(128, 128, 32, 256, 32, 256, 4, 1, 1, 32, 32, 16, -1), '256' : FmhaFwdTileSize(128, 32, 32, 256, 32, 256, 4, 1, 1, 32, 32, 16, -1),
} }
elif dtype == 'fp8' or dtype == 'bf8': elif dtype == 'fp8' or dtype == 'bf8':
return { return {
......
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