Sep 16, 2025 · Jun 17, 2025 · Jun 17, 2025 · Jul 2, 2025 · Jul 2, 2025 · Jul 8, 2025
diff --git a/tests/unittest/llmapi/test_llm.py b/tests/unittest/llmapi/test_llm.py
 @pytest.mark.parametrize(
    "prompt_logprobs, logprobs, return_context_logits, return_generation_logits",
    [(2, None, True, False), (None, 2, False, False)])
 @pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5516849")
 def test_llm_return_logprobs(prompt_logprobs: Optional[int],
                             logprobs: Optional[int],
                             return_context_logits: bool,

 @pytest.mark.skip(reason="https://nvbugs/5516660")
 @force_ampere
 @pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5516849")
 def test_llm_return_logprobs_streaming():
    llm_return_logprobs_test_harness(2, 2, False, True, streaming=True)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -1882,7 +1882,6 @@ async def main():
		@pytest.mark.parametrize(
		"prompt_logprobs, logprobs, return_context_logits, return_generation_logits",
		[(2, None, True, False), (None, 2, False, False)])
		@pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5516849")
		def test_llm_return_logprobs(prompt_logprobs: Optional[int],
		logprobs: Optional[int],
		return_context_logits: bool,
Expand All		@@ -1894,7 +1893,6 @@ def test_llm_return_logprobs(prompt_logprobs: Optional[int],

		@pytest.mark.skip(reason="https://nvbugs/5516660")
		@force_ampere
		@pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5516849")
		def test_llm_return_logprobs_streaming():
		llm_return_logprobs_test_harness(2, 2, False, True, streaming=True)

Expand Down