Skip to content

Commit 57b4b47

Browse files
authored
unwaive test for nvbug_5150466 (NVIDIA#3552)
Signed-off-by: Erin Ho <[email protected]>
1 parent 36bacf7 commit 57b4b47

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

tests/unittest/llmapi/test_llm_multi_gpu.py

-6
Original file line numberDiff line numberDiff line change
@@ -436,18 +436,12 @@ def __init__(
436436
@skip_single_gpu
437437
@pytest.mark.parametrize("pytorch_backend", [False, True])
438438
def test_llm_get_stats_tp2(pytorch_backend):
439-
if pytorch_backend:
440-
pytest.skip("https://nvbugs/5150466: Flaky hang")
441-
return
442439
llm_get_stats_test_harness(tp_size=2, pytorch_backend=pytorch_backend)
443440

444441

445442
@skip_single_gpu
446443
@pytest.mark.parametrize("pytorch_backend", [False, True])
447444
def test_llm_get_stats_async_tp2(pytorch_backend):
448-
if pytorch_backend:
449-
pytest.skip("https://nvbugs/5150466: Flaky hang")
450-
return
451445
llm_get_stats_async_test_harness(tp_size=2, pytorch_backend=pytorch_backend)
452446

453447

0 commit comments

Comments
 (0)