From a335dc4d6d3cd88c12ca9dd869d8c6621e969ccb Mon Sep 17 00:00:00 2001 From: Yao Matrix Date: Wed, 16 Apr 2025 17:04:57 +0800 Subject: [PATCH] enable `test_offloaded_cache_implementation` on XPU (#37514) Signed-off-by: YAO Matrix Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com> --- tests/generation/test_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/generation/test_utils.py b/tests/generation/test_utils.py index 5cda62c397..c738ca4645 100644 --- a/tests/generation/test_utils.py +++ b/tests/generation/test_utils.py @@ -1947,7 +1947,7 @@ class GenerationTesterMixin: ) @parameterized.expand([("offloaded",)]) # ("offloaded_static",) TODO: @raushan fixme in some models (eg T5) - @require_torch_gpu + @require_torch_accelerator @pytest.mark.generate def test_offloaded_cache_implementation(self, cache_implementation): """Tests we can generate by indicating `cache_implementation` for each possible cache class"""