diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml index f3fcfe1c5..1223a4121 100644 --- a/.github/workflows/run-tests.yml +++ b/.github/workflows/run-tests.yml @@ -39,7 +39,7 @@ jobs: run: | cd tests export HIVEMIND_MEMORY_SHARING_STRATEGY=file_descriptor - pytest --durations=0 --durations-min=1.0 -n auto -v + pytest --durations=0 --durations-min=1.0 -n 4 -v build_and_test_p2pd: runs-on: ubuntu-latest timeout-minutes: 10 diff --git a/tests/test_allreduce_fault_tolerance.py b/tests/test_allreduce_fault_tolerance.py index c2dcd5594..e122e9b01 100644 --- a/tests/test_allreduce_fault_tolerance.py +++ b/tests/test_allreduce_fault_tolerance.py @@ -153,10 +153,10 @@ def _make_tensors(): _make_tensors(), hivemind.DHT(initial_peers=dht.get_visible_maddrs(), start=True), prefix="test", - request_timeout=1.0, - min_matchmaking_time=1.0, - next_chunk_timeout=0.5, - allreduce_timeout=10, + request_timeout=1.5, + min_matchmaking_time=2.0, + next_chunk_timeout=1.0, + allreduce_timeout=15, part_size_bytes=2**16, client_mode=(i == 1), start=True,