File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed
Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff 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" )
104105def 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" )
122124def 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" )
149152def test_llm_kv_events_api ():
150153 llm = create_llm ()
151154 sampling_params = SamplingParams (max_tokens = 6 , temperature = 0.01 )
Original file line number Diff line number Diff 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
272273def test_llama_v2_7b_prompt_adapter_tp2 ():
You can’t perform that action at this time.
0 commit comments