Program does not exit when using torch.distributed.tensor.distribute_module
.
#139060
Labels
module: dtensor
distributed tensor tag
oncall: distributed
Add this issue/PR to distributed oncall triage queue
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
🐛 Describe the bug
The following program (mainly based on the example provided in PyTorch DTensor) does not exit automatically after outputting the shape of
y
:The command used to run this program is:
The full output is:
It seems that the program should exit after the final output.
Versions
cc @H-Huang @awgu @kwen2501 @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @c-p-i-o @tianyu-l @XilunWu
The text was updated successfully, but these errors were encountered: