pytorch/test/distributed/_tensor/__init__.py at f2f4dde2d37d2e2e6c76cdeeeea376cc7ffb8016 - saymrwulf/pytorch - Forgejo: Beyond coding. We forge.

saymrwulf/pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00

Wanchao Liang 527c5bdb45 [dtensor] PART 5: move DTensor basic tests to core distributed (#88178 )

This PR moves DTensor basic tests to torch.distributed, including
dtensor, device_mesh tests

part of https://github.com/pytorch/pytorch/issues/88838
Pull Request resolved: https://github.com/pytorch/pytorch/pull/88178
Approved by: https://github.com/fduwjj

2022-11-16 08:07:46 +00:00

1 line

17 B

Python

Raw Blame History

# shut up pylint