Skip to content

Commit 980030c

Browse files
authored
[Infra] - Waive failed case in post-merge (NVIDIA#5536)
Signed-off-by: qqiao <[email protected]>
1 parent 83a1f60 commit 980030c

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tests/unittest/llmapi/test_llm_kv_cache_events.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,7 @@ def test_kv_cache_event_data_serialization():
101101
serialized_event = KVCacheEventSerializer.serialize(events)
102102

103103

104+
@pytest.mark.skip(reason="https://nvbugs/5362412")
104105
def test_expected_kv_cache_events():
105106
llm = create_llm()
106107
sampling_params = SamplingParams(max_tokens=6, temperature=0.01)
@@ -119,6 +120,7 @@ def test_expected_kv_cache_events():
119120
assert event["data"]["type"] == "stored"
120121

121122

123+
@pytest.mark.skip(reason="https://nvbugs/5362412")
122124
def test_kv_cache_event_async_api():
123125
llm = create_llm()
124126
sampling_params = SamplingParams(max_tokens=6, temperature=0.01)
@@ -146,6 +148,7 @@ async def main():
146148
asyncio.run(main())
147149

148150

151+
@pytest.mark.skip(reason="https://nvbugs/5362412")
149152
def test_llm_kv_events_api():
150153
llm = create_llm()
151154
sampling_params = SamplingParams(max_tokens=6, temperature=0.01)

tests/unittest/llmapi/test_llm_multi_gpu.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,7 @@ def test_llama_7b_multi_lora_tp2():
267267
kv_cache_config=global_kv_cache_config)
268268

269269

270+
@pytest.mark.skip(reason="https://nvbugs/5362426")
270271
@pytest.mark.gpu2
271272
@pytest.mark.part3
272273
def test_llama_v2_7b_prompt_adapter_tp2():

0 commit comments

Comments
 (0)