Skip to content

GitLab

Unverified Commit 58cde5c0 authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI] Skip trtllm kvfp8 dequant tests on ROCm (#37330) 


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent 761e0aa7
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
tests/kernels/attention/test_trtllm_kvfp8_dequant.py
View file @ 58cde5c0
......@@ -12,6 +12,12 @@ import torch
from vllm.platforms import current_platform
if current_platform.is_rocm():
pytest.skip(
"trtllm kvfp8 dequant is not supported on ROCm.",
allow_module_level=True,
)
FP8_DTYPE = current_platform.fp8_dtype()
NUM_BLOCKS = 128
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment