Skip to content

Create tests/distributed/test_mnnvl_alltoall.py #35241

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
puririshi98 wants to merge 25 commits into
base: main
Choose a base branch
from
+1,098 −0
Open

Create tests/distributed/test_mnnvl_alltoall.py #35241

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
a93eb3d
Create tests/distributed/test_mnnvl_alltoall.py
puririshi98 Feb 24, 2026
81518b4
Merge branch 'main' into patch-1
puririshi98 Feb 25, 2026
6d9f403
Merge branch 'main' into patch-1
puririshi98 Feb 26, 2026
a0bb03a
fix lint
puririshi98 Feb 27, 2026
0c1d891
Merge branch 'main' into patch-1
puririshi98 Feb 27, 2026
3e549ee
Merge branch 'main' into patch-1
puririshi98 Feb 27, 2026
96e3808
precommit
puririshi98 Feb 27, 2026
6f25a5c
Merge branch 'main' into patch-1
puririshi98 Feb 27, 2026
e067566
Merge branch 'main' into patch-1
puririshi98 Feb 27, 2026
9b5c72f
Merge branch 'main' into patch-1
puririshi98 Feb 27, 2026
fe5774a
Merge branch 'main' into patch-1
puririshi98 Feb 28, 2026
7d7ba9a
Merge branch 'main' into patch-1
puririshi98 Feb 28, 2026
9108f52
Merge branch 'main' into patch-1
puririshi98 Mar 3, 2026
ee1b67e
Merge branch 'main' into patch-1
puririshi98 Mar 5, 2026
c9cb5f8
Update distributed.yaml
puririshi98 Mar 6, 2026
9e4ea3e
addressing review to move to h100 ci
puririshi98 Mar 6, 2026
82303e5
Merge branch 'main' into patch-1
puririshi98 Mar 9, 2026
ea9f002
Update test_mnnvl_alltoall.py
puririshi98 Mar 9, 2026
e95863c
Merge branch 'main' into patch-1
puririshi98 Mar 9, 2026
21cf347
Fix test_mnnvl_alltoall
puririshi98 Mar 9, 2026
222db47
Merge branch 'main' into patch-1
puririshi98 Mar 10, 2026
605905b
Update test_mnnvl_alltoall.py
puririshi98 Mar 10, 2026
0c603c6
Merge branch 'main' into patch-1
puririshi98 Mar 10, 2026
a67efdf
Merge branch 'main' into patch-1
puririshi98 Mar 10, 2026
4361115
Merge branch 'main' into patch-1
puririshi98 Mar 11, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .buildkite/test_areas/distributed.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -121,11 +121,14 @@ steps:
- vllm/v1/engine/llm_engine.py
- vllm/v1/executor/uniproc_executor.py
- vllm/v1/worker/gpu_worker.py
- tests/distributed/test_mnnvl_alltoall.py

commands:
# https://github.com/NVIDIA/nccl/issues/1838
- export NCCL_CUMEM_HOST_ENABLE=0
# test with torchrun tp=2 and dp=4 with ep
- torchrun --nproc-per-node=8 ../examples/offline_inference/torchrun_dp_example.py --tp-size=2 --pp-size=1 --dp-size=4 --enable-ep
- pytest -v -s tests/distributed/test_mnnvl_alltoall.py

- label: Distributed Tests (4 GPUs)(A100)
device: a100
Expand Down
Loading
Loading