Skip to content

GitLab

Unverified Commit 8dd19e3b authored by Chaitanya Sri Krishna Lolla's avatar Chaitanya Sri Krishna Lolla Committed by GitHub
Browse files

skipping bfloat16 mgpu tests (#32)

parent 9c80f6d3
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 10 deletions
tests/distributed/run_rocm_distributed.sh
View file @ 8dd19e3b
......@@ -18,16 +18,16 @@ rm rank1master.pth
# Test with opt_level="O5"
echo "running opt_level O5"
python3.6 -m torch.distributed.launch --nproc_per_node=2 amp_master_params/amp_master_params.py --opt_level "O5"
python3.6 amp_master_params/compare.py
# delete the model files
echo "O5 test completed. Deleting model files"
rm rank0model.pth
rm rank1model.pth
rm rank0master.pth
rm rank1master.pth
#echo "running opt_level O5"
#python3.6 -m torch.distributed.launch --nproc_per_node=2 amp_master_params/amp_master_params.py --opt_level "O5"
#python3.6 amp_master_params/compare.py
#
## delete the model files
#echo "O5 test completed. Deleting model files"
#rm rank0model.pth
#rm rank1model.pth
#rm rank0master.pth
#rm rank1master.pth
## Run the Sync BN Tests.
echo "Running syncbn tests"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment