From b62557e712c8a6650ad1403d516006c10141a830 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=81kos=20Hadnagy?= Date: Tue, 22 Jul 2025 15:40:16 +0200 Subject: [PATCH] Add AMD expectations to Mistral3 tests (#39481) Add AMD expectations to mistral3 tests --- tests/models/mistral3/test_modeling_mistral3.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/models/mistral3/test_modeling_mistral3.py b/tests/models/mistral3/test_modeling_mistral3.py index 9f0e4ef6c5..4dfd539646 100644 --- a/tests/models/mistral3/test_modeling_mistral3.py +++ b/tests/models/mistral3/test_modeling_mistral3.py @@ -317,6 +317,7 @@ class Mistral3IntegrationTest(unittest.TestCase): { ("xpu", 3): "The image features two cats resting on a pink blanket. The cat on the left is a kitten", ("cuda", 8): 'The image features two cats lying on a pink surface, which appears to be a couch or a bed', + ("rocm", (9, 5)): "The image features two tabby cats lying on a pink surface, which appears to be a cushion or" } ) # fmt: skip expected_output = expected_outputs.get_expectation() @@ -363,6 +364,7 @@ class Mistral3IntegrationTest(unittest.TestCase): { ("xpu", 3): "Calm lake's mirror gleams,\nWhispering pines stand in silence,\nPath to peace begins.", ("cuda", 8): "Wooden path to calm,\nReflections whisper secrets,\nNature's peace unfolds.", + ("rocm", (9, 5)): "Calm waters reflect\nWooden path to distant shore\nSilence in the scene" } ) # fmt: skip expected_output = expected_outputs.get_expectation()