From 1e8c2d1cef889339f44e58795a80914375cf0947 Mon Sep 17 00:00:00 2001 From: Superjomn <328693+Superjomn@users.noreply.github.com> Date: Mon, 21 Apr 2025 06:44:09 +0000 Subject: [PATCH] unwaive multi-node test Signed-off-by: Superjomn <328693+Superjomn@users.noreply.github.com> --- tests/unittest/llmapi/test_llm_multi_gpu.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/unittest/llmapi/test_llm_multi_gpu.py b/tests/unittest/llmapi/test_llm_multi_gpu.py index fd7027f0b4b..e93b1c922ce 100644 --- a/tests/unittest/llmapi/test_llm_multi_gpu.py +++ b/tests/unittest/llmapi/test_llm_multi_gpu.py @@ -288,10 +288,8 @@ def run_command(command: str): raise e -@pytest.mark.skip(reason="https://nvbugspro.nvidia.com/bug/5223608: timeout") @skip_single_gpu def test_llm_multi_node(engine_from_checkpoint: tempfile.TemporaryDirectory): - # TODO[chunweiy]: reactivate this later nworkers = 2 test_case_file = os.path.join(os.path.dirname(__file__), "run_llm.py") os.path.join(os.path.dirname(__file__), "launch.py")