[ROCm][CI] Unblock `AMD: Language Models Test (Extended Pooling)` (#48513)
Signed-off-by: Micah Williamson <micah.williamson@amd.com>
Micah Williamson committed
Jul 13, 2026 at 13:38 UTC
18c4067a5443ac04186e0936ace16268829ebe81
1 file changed
+4
tests/models/language/pooling/test_token_classification.py
+4
@@ -128,6 +128,10 @@ PRIVACY_FILTER_PROMPTS = [
128
]
129
130
131
+@pytest.mark.skipif(
132
+ current_platform.is_rocm(),
133
+ reason="Workspace growth allocation issue on ROCm. See https://github.com/vllm-project/vllm/issues/48510",
134
+)
135
@pytest.mark.parametrize("model", ["openai/privacy-filter"])
136
@pytest.mark.parametrize("dtype", ["bfloat16"])
137
@torch.inference_mode