Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
apex
Commits
3f3da214
Commit
3f3da214
authored
Dec 01, 2021
by
Hubert Lu
Browse files
Update run_rocm_distributed.sh
parent
405956c3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
tests/distributed/run_rocm_distributed.sh
tests/distributed/run_rocm_distributed.sh
+2
-0
No files found.
tests/distributed/run_rocm_distributed.sh
View file @
3f3da214
...
...
@@ -38,6 +38,8 @@ echo "Running syncbn python only tests"
python synced_batchnorm/python_single_gpu_unit_test.py
echo
"Running syncbn batchnorm1d tests"
python synced_batchnorm/test_batchnorm1d.py
#beware, you need a system with at least 4 gpus to test group_size<world_size (currently fail both on upstream and rocm fork)
#python -m torch.distributed.launch --nproc_per_node=4 test_groups.py --group_size=2
## Run the DDP Tests
echo
"running DDP tests"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment