[ROCm][CI] Fix cross-attention dispatch for encoder-decoder models (#38450)
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user