Skip to content

Commit 3a61e41

Browse files
committed
update
Signed-off-by: zhewenli <zhewenli@meta.com>
1 parent a695390 commit 3a61e41

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

tests/entrypoints/openai/test_run_batch.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,13 @@
4444

4545

4646
def test_empty_file():
47+
from vllm.platforms import current_platform
48+
49+
if current_platform.is_rocm():
50+
pytest.skip(
51+
"intfloat/multilingual-e5-small is encoder-only, not supported on ROCm"
52+
)
53+
4754
with (
4855
tempfile.NamedTemporaryFile("w") as input_file,
4956
tempfile.NamedTemporaryFile("r") as output_file,
@@ -128,6 +135,13 @@ def test_completions_invalid_input():
128135

129136

130137
def test_embeddings():
138+
from vllm.platforms import current_platform
139+
140+
if current_platform.is_rocm():
141+
pytest.skip(
142+
"intfloat/multilingual-e5-small is encoder-only, not supported on ROCm"
143+
)
144+
131145
with (
132146
tempfile.NamedTemporaryFile("w") as input_file,
133147
tempfile.NamedTemporaryFile("r") as output_file,
@@ -159,6 +173,11 @@ def test_embeddings():
159173

160174
@pytest.mark.parametrize("input_batch", [INPUT_SCORE_BATCH, INPUT_RERANK_BATCH])
161175
def test_score(input_batch):
176+
from vllm.platforms import current_platform
177+
178+
if current_platform.is_rocm():
179+
pytest.skip("BAAI/bge-reranker-v2-m3 is encoder-only, not supported on ROCm")
180+
162181
with (
163182
tempfile.NamedTemporaryFile("w") as input_file,
164183
tempfile.NamedTemporaryFile("r") as output_file,

0 commit comments

Comments
 (0)