Dzmitry Huba
a51f877287
Enable local tensor mode for another set of DTensor tests ( #166105 )
...
Enable local tensor mode DTensor tests for the optimizers, op strategy, matrix ops,
math ops, init ops, experimental ops, embedding ops, dynamic, convolution ops, main api.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/166105
Approved by: https://github.com/ezyang
2025-10-27 23:58:24 +00:00
Liao, Wei
e19e02c84c
port distributed tensor test files for Intel GPU ( #161604 )
...
In this pr, we port test/distributed/tensor test filesfor Intel GPU
We could enable Intel GPU with following methods and try the best to keep the original code styles:
Use torch.accelerator for general gpu
Skip the case if running on xpu which has known issues
Pull Request resolved: https://github.com/pytorch/pytorch/pull/161604
Approved by: https://github.com/guangyey , https://github.com/d4l3k
2025-09-04 07:49:25 +00:00
Electron4444
832ab990c9
Use init_device_mesh API for select tests where possible ( #158675 )
...
This addresses reviews made for:
#158538
#108749
It interchanged all the specific DevideMesh constructor calls with the API provided by the test cases, to improve abstraction
Pull Request resolved: https://github.com/pytorch/pytorch/pull/158675
Approved by: https://github.com/wconstab
2025-07-22 19:28:42 +00:00
Xilun Wu
cbb03e6971
[BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files ( #153225 )
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/153225
Approved by: https://github.com/kwen2501 , https://github.com/fegin
2025-05-09 20:40:54 +00:00
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
PyTorch MergeBot
6c54963f75
Revert "[dtensor] move all tests to distribute/tensor folder ( #144166 )"
...
This reverts commit 2e1ea8598fhttps://github.com/pytorch/pytorch/pull/144166 on behalf of https://github.com/huydhn due to Sorry for reverting your change, but inductor/test_compiled_autograd needs to be updated ([comment](https://github.com/pytorch/pytorch/pull/144166#issuecomment-2575969871 ))
2025-01-07 18:31:36 +00:00
Wanchao Liang
2e1ea8598f
[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-07 06:45:14 +00:00
