[ROCm][CI] Fix cross-attention dispatch for encoder-decoder models (#38450)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
This commit is contained in:
Andreas Karatzas
2026-03-29 00:08:03 -05:00
committed by GitHub
parent 5b8c30d62b
commit 43cc5138e5
6 changed files with 90 additions and 19 deletions

View File

@@ -446,7 +446,7 @@ def parse_attention_types(node: ast.ClassDef) -> str:
if not types:
return "Decoder"
return "All" if len(types) >= 3 else ", ".join(sorted(types))
return "All" if types >= set(type_map.values()) else ", ".join(sorted(types))
def parse_impl_bool_attr(