gfx90a update

ROCm · Feb 19, 2025 · 30abf0b · 30abf0b
1 parent f17b6ff
commit 30abf0b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py b/tests/models/qwen2_5_vl/test_modeling_qwen2_5_vl.py
@@ -237,7 +237,7 @@ class Qwen2_5_VLModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.Test
     def test_generate_continue_from_past_key_values(self):
         super().test_generate_continue_from_past_key_values()
 
-    @skipIfRocm(arch=['gfx942','gfx1201','gfx1200'])
+    @skipIfRocm(arch=['gfx942','gfx1201','gfx1200','gfx90a'])
     def test_generate_with_static_cache(self):
         super().test_generate_with_static_cache()
 

diff --git a/tests/models/whisper/test_modeling_whisper.py b/tests/models/whisper/test_modeling_whisper.py
@@ -43,6 +43,7 @@
     require_torchaudio,
     slow,
     torch_device,
+    skipIfRocm,
 )
 from transformers.utils import cached_property, is_flax_available, is_torch_available, is_torchaudio_available
 from transformers.utils.import_utils import is_datasets_available
@@ -395,6 +396,10 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
     # `0.5` is for `test_disk_offload` (which also works for `test_model_parallelism`)
     model_split_percents = [0.5, 0.8, 0.9]
 
+    @skipIfRocm(arch='gfx90a', os_name='ubuntu', os_version='24.04')
+    def test_longform_generate_multi_batch_cond_prev(self):
+        super().test_longform_generate_multi_batch_cond_prev()
+
     # TODO: Fix the failed tests
     def is_pipeline_test_to_skip(
         self,