From de0af08a27db45e87b9f4d94efdc6bf5cd4061a6 Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Thu, 12 Aug 2021 09:43:13 -0700 Subject: [PATCH] Reset UCX-Py after rdmacm tests run --- dask_cuda/tests/test_dgx.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/dask_cuda/tests/test_dgx.py b/dask_cuda/tests/test_dgx.py index 9ec67b4b9..6c91666df 100644 --- a/dask_cuda/tests/test_dgx.py +++ b/dask_cuda/tests/test_dgx.py @@ -263,6 +263,12 @@ def test_ucx_infiniband_nvlink(params): ) p.start() p.join() + + # Starting a new cluster on the same pytest process after an rdmacm cluster + # has been used may cause UCX-Py to complain about being already initialized. + if params["enable_rdmacm"] is True: + ucp.reset() + assert not p.exitcode