From feecba09afcd3c9a37c6b03b868dedf4cb851eb1 Mon Sep 17 00:00:00 2001 From: rasmith Date: Fri, 5 Dec 2025 02:42:25 -0600 Subject: [PATCH] [CI/Build][AMD] Use float16 in test_reset_prefix_cache_e2e to avoid accuracy issues (#29997) Signed-off-by: Randall Smith Co-authored-by: Randall Smith --- tests/v1/core/test_reset_prefix_cache_e2e.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/v1/core/test_reset_prefix_cache_e2e.py b/tests/v1/core/test_reset_prefix_cache_e2e.py index 083fc3f34f545..b80789945d2fc 100644 --- a/tests/v1/core/test_reset_prefix_cache_e2e.py +++ b/tests/v1/core/test_reset_prefix_cache_e2e.py @@ -21,6 +21,7 @@ def test_reset_prefix_cache_e2e(monkeypatch): max_num_batched_tokens=32, max_model_len=2048, compilation_config={"mode": 0}, + dtype="float16", ) engine = LLMEngine.from_engine_args(engine_args) sampling_params = SamplingParams(