|
@@ -578,15 +578,15 @@ if __name__ == '__main__':
|
|
|
with cleanup_children():
|
|
|
loop = asyncio.new_event_loop()
|
|
|
loop.run_until_complete(test_key_manager())
|
|
|
- print(f"test_key_manager()")
|
|
|
+ print(f"PASSED test_key_manager()")
|
|
|
del loop
|
|
|
for n_clients in [0, 1, 2]:
|
|
|
for n_aux in [0, 1, 2]:
|
|
|
with cleanup_children():
|
|
|
_test_allreduce_once(n_clients, n_aux)
|
|
|
- print(f"_test_allreduce_once({n_clients}, {n_aux})")
|
|
|
+ print(f"PASSED _test_allreduce_once({n_clients}, {n_aux})")
|
|
|
for n_clients, n_aux in [(0, 4), (1, 3), (0, 3)]:
|
|
|
with cleanup_children():
|
|
|
_test_allreduce_once(n_clients, n_aux)
|
|
|
- print(f"_test_allreduce_once({n_clients}, {n_aux})")
|
|
|
+ print(f"PASSED _test_allreduce_once({n_clients}, {n_aux})")
|
|
|
print("DONE!")
|