[ROCm][CI][Bugfix] Multi-Modal Model Support Fixes and Attention Backend Improvements (#30270)

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
This commit is contained in:
Andreas Karatzas
2025-12-18 20:17:27 -06:00
committed by GitHub
parent 6a09612b2e
commit 7b43db210c
7 changed files with 109 additions and 47 deletions

View File

@@ -39,7 +39,7 @@ models = [MODEL_NAME]
def granite_speech_attention_config():
"""Return attention config for Granite Speech tests on ROCm."""
if current_platform.is_rocm():
return {"backend": "TRITON_ATTN"}
return {"backend": "ROCM_AITER_FA"}
return None