Skip to content

Commit 12cecfd

Browse files
authored
Update README.md instructions (#29)
1 parent 1d76f48 commit 12cecfd

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@ python3 -m tests.bench_all_to_all
3434

3535
## Multi-node Testing and Benchmarking
3636

37+
To run an all-to-all test across multiple nodes:
38+
3739
```bash
3840
export NODE_RANK= # 0, 1, ..., num_nodes-1
3941
export WORLD_SIZE= # num_nodes * 8
@@ -43,7 +45,11 @@ export MASTER_PORT=29500
4345
export NVSHMEM_IB_ENABLE_IBGDA=1
4446
```
4547

46-
After settings these environment variables, commands to run the tests and benchmarks are the same as the single-node case.
48+
An then run the tests normally:
49+
50+
```bash
51+
pytest -svx tests/test_all_to_all.py
52+
```
4753

4854
## Benchmark Results
4955

0 commit comments

Comments
 (0)