From ac12c5dc311c36eab880076f927a1b4bf6f6be1a Mon Sep 17 00:00:00 2001 From: Parag Kulkarni Date: Mon, 23 Oct 2023 13:23:14 +0530 Subject: [PATCH] test_utility_n_multi_rank_with_launcher: Add logic --- tests/unit/legate/driver/test_command.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/unit/legate/driver/test_command.py b/tests/unit/legate/driver/test_command.py index d278ee1c0..be74c755d 100644 --- a/tests/unit/legate/driver/test_command.py +++ b/tests/unit/legate/driver/test_command.py @@ -1228,7 +1228,10 @@ def test_utility_n_multi_rank_with_launcher( result = m.cmd_bgwork(config, system, launcher) - assert result == ("-ll:bgwork", value) + if "ucx" in install_info.networks: + assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1") + else: + assert result == ("-ll:bgwork", value) @pytest.mark.parametrize("launch", ("mpirun", "jsrun", "srun")) @pytest.mark.parametrize("value", ("2", "3", "10"))