saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/tensor/parallel
History
fduwjj 23b7035b3c [TP] Add an input resharding wrapper for TP and unit test for 2D + AC (#103334) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/103334
Approved by: https://github.com/kumpera
2023-06-23 04:05:01 +00:00
..
__init__.py
test_2d_parallel.py [TP] Add an input resharding wrapper for TP and unit test for 2D + AC (#103334) 2023-06-23 04:05:01 +00:00
test_parallelize_api.py [TP] Add an input resharding wrapper for TP and unit test for 2D + AC (#103334) 2023-06-23 04:05:01 +00:00
test_tp_examples.py [TP] Add an input resharding wrapper for TP and unit test for 2D + AC (#103334) 2023-06-23 04:05:01 +00:00
test_tp_style.py
test_view_sharding_dim_change.py