Open qthequartermasterman opened 1 month ago
This is how PyTorch is currently testing DTensors, courtesy of @Nan2018.
https://github.com/pytorch/pytorch/blob/main/torch/testing/_internal/common_distributed.py#L521
Looks like they've moved some tests around. I think this is the new location:
Is your feature request related to a problem? Please describe. A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
https://pytorch.org/docs/stable/distributed.tensor.parallel.html
There is a distributed tensor type (DTensor) that is not compatible with all torch operations. It would be nice to allow tensor_strategy to create Dtensors.
Describe the solution you'd like A clear and concise description of what you want to happen.
Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.
Additional context Add any other context or screenshots about the feature request here.