Skip to content

Commit 738ab61

Browse files
authored
[nvbugs/5404000] fix: waive request_perf_metrics_draft test on pre-Hopper GPUs (#6339)
Signed-off-by: Aurelien Chartier <[email protected]>
1 parent bca1415 commit 738ab61

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/unittest/bindings/test_executor_bindings.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424

2525
from utils.cpp_paths import *
2626
from utils.llm_data import llm_models_root
27+
from utils.util import skip_pre_hopper
2728

2829

2930
@pytest.fixture
@@ -2183,6 +2184,8 @@ def test_request_perf_metrics_kv_cache(model_path):
21832184
assert kv_cache_metrics.kv_cache_hit_rate == 1.0
21842185

21852186

2187+
# Skip test for pre-Hopper: https://nvbugs/5404000
2188+
@skip_pre_hopper
21862189
@pytest.mark.parametrize("exclude_input_from_output", [False, True])
21872190
def test_request_perf_metrics_draft(model_path_draft_tokens_external,
21882191
exclude_input_from_output: bool):

0 commit comments

Comments
 (0)