frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

add 2d tests to ci (#96711)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96711
Approved by: https://github.com/huydhn, https://github.com/fduwjj
This commit is contained in:
Wanchao Liang
2023-03-15 01:38:23 +00:00
committed by PyTorch MergeBot
parent 5612aa6acd
commit aa09f8891c
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.ci/pytorch/multigpu-test.sh
View File

@ -39,6 +39,11 @@ time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/
time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_matrix_ops
time python test/run_test.py --verbose -i distributed/_shard/sharded_tensor/ops/test_softmax
time python test/run_test.py --verbose -i distributed/_shard/sharded_optim/test_sharded_optim
# DTensor/TP tests
time python test/run_test.py --verbose -i distributed/tensor/parallel/test_2d_parallel
time python test/run_test.py --verbose -i distributed/tensor/parallel/test_tp_examples
# Other tests
time python test/run_test.py --verbose -i test_cuda_primary_ctx
time python test/run_test.py --verbose -i test_optim -- -k optimizers_with_varying_tensors