Xilun Wu add0b450bd [DTensor][BE] improve DTensor ops correctness check utils (#158112) ... **Summary** Implemented the test pattern described in https://github.com/pytorch/pytorch/pull/157991#discussion_r2196363170 as a util method in `DTensorTestBase`. The difference to `DTensorTestBase._test_op` is: 1. allowing users to specify the `Partial` placement. 2. supporting tree-like output structure. **Test** so far only adopt `DTensorTestBase._test_op_on_dtensor` in `DistTensorOpsTest.test_split_on_partial`. `pytest test/distributed/tensor/test_tensor_ops.py -s -k test_split_on_partial` Pull Request resolved: https://github.com/pytorch/pytorch/pull/158112 Approved by: https://github.com/Skylion007, https://github.com/zpcore ghstack dependencies: #158051		2025-07-15 04:50:34 +00:00
..
_internal	[DTensor][BE] improve DTensor ops correctness check utils (#158112)	2025-07-15 04:50:34 +00:00
__init__.py
_comparison.py	[Torch] Fix error message formatting in fp8 comparison logic (#153647)	2025-05-27 02:51:05 +00:00
_creation.py	PEP585 update - torch/testing (#145200)	2025-01-20 22:42:42 +00:00
_utils.py	Fix code descriptions in the test package. (#148145)	2025-03-04 19:14:41 +00:00