saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/distributed/tensor/README.md
Wanchao Liang 96f4abba17 [dtensor] move all tests to distribute/tensor folder (#144166) 
as titled, mainly moving files

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144166
Approved by: https://github.com/Skylion007
2025-01-08 00:32:33 +00:00

242 B
Raw Permalink Blame History

Run distributed tensor tests:

from root, run (either CPU or GPU)

pytest test/distributed/tensor/test_dtensor.py

run specific test cases and print stdout/stderr:

pytest test/distributed/tensor/test_dtensor.py -s -k test_from_local