Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
0c275ad5
Unverified
Commit
0c275ad5
authored
Aug 05, 2025
by
Nicolò Lucchesi
Committed by
GitHub
Aug 05, 2025
Browse files
[V0 Deprecation][TPU] Remove V1 flag check from tests (#22248)
Signed-off-by:
NickLucche
<
nlucches@redhat.com
>
parent
74333ae2
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
1 addition
and
21 deletions
+1
-21
tests/v1/tpu/test_mha_attn.py
tests/v1/tpu/test_mha_attn.py
+0
-7
tests/v1/tpu/test_multimodal.py
tests/v1/tpu/test_multimodal.py
+0
-7
tests/v1/tpu/test_sampler.py
tests/v1/tpu/test_sampler.py
+1
-7
No files found.
tests/v1/tpu/test_mha_attn.py
View file @
0c275ad5
...
@@ -12,17 +12,10 @@ import torch_xla
...
@@ -12,17 +12,10 @@ import torch_xla
import
torch_xla.core
import
torch_xla.core
import
torch_xla.core.xla_model
import
torch_xla.core.xla_model
from
vllm
import
envs
from
vllm.attention.layer
import
MultiHeadAttention
from
vllm.attention.layer
import
MultiHeadAttention
from
vllm.attention.selector
import
_cached_get_attn_backend
from
vllm.attention.selector
import
_cached_get_attn_backend
from
vllm.platforms
import
current_platform
from
vllm.platforms
import
current_platform
if
not
envs
.
VLLM_USE_V1
:
pytest
.
skip
(
"Skipping V1 tests. Rerun with `VLLM_USE_V1=1` to test."
,
allow_module_level
=
True
,
)
@
pytest
.
fixture
(
autouse
=
True
)
@
pytest
.
fixture
(
autouse
=
True
)
def
clear_cache
():
def
clear_cache
():
...
...
tests/v1/tpu/test_multimodal.py
View file @
0c275ad5
...
@@ -4,19 +4,12 @@
...
@@ -4,19 +4,12 @@
import
openai
import
openai
import
pytest
import
pytest
from
vllm
import
envs
from
vllm.multimodal.utils
import
encode_image_base64
,
fetch_image
from
vllm.multimodal.utils
import
encode_image_base64
,
fetch_image
from
vllm.platforms
import
current_platform
from
vllm.platforms
import
current_platform
from
...entrypoints.openai.test_vision
import
TEST_IMAGE_URLS
from
...entrypoints.openai.test_vision
import
TEST_IMAGE_URLS
from
...utils
import
RemoteOpenAIServer
from
...utils
import
RemoteOpenAIServer
if
not
envs
.
VLLM_USE_V1
:
pytest
.
skip
(
"Skipping V1 tests. Rerun with `VLLM_USE_V1=1` to test."
,
allow_module_level
=
True
,
)
@
pytest
.
fixture
(
scope
=
"session"
)
@
pytest
.
fixture
(
scope
=
"session"
)
def
base64_encoded_image
()
->
dict
[
str
,
str
]:
def
base64_encoded_image
()
->
dict
[
str
,
str
]:
...
...
tests/v1/tpu/test_sampler.py
View file @
0c275ad5
...
@@ -4,16 +4,10 @@ import random
...
@@ -4,16 +4,10 @@ import random
import
pytest
import
pytest
from
vllm
import
LLM
,
envs
from
vllm
import
LLM
from
vllm.platforms
import
current_platform
from
vllm.platforms
import
current_platform
from
vllm.sampling_params
import
SamplingParams
from
vllm.sampling_params
import
SamplingParams
if
not
envs
.
VLLM_USE_V1
:
pytest
.
skip
(
"Skipping V1 tests. Rerun with `VLLM_USE_V1=1` to test."
,
allow_module_level
=
True
,
)
@
pytest
.
mark
.
parametrize
(
"model_name"
,
[
"Qwen/Qwen2.5-1.5B-Instruct"
])
@
pytest
.
mark
.
parametrize
(
"model_name"
,
[
"Qwen/Qwen2.5-1.5B-Instruct"
])
@
pytest
.
mark
.
skipif
(
not
current_platform
.
is_tpu
(),
@
pytest
.
mark
.
skipif
(
not
current_platform
.
is_tpu
(),
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment