pytorch
599ae95d - [dtensor] use stack to manage mesh resources (#101202)

Commit
1 year ago
[dtensor] use stack to manage mesh resources (#101202) This PR changes the context manager behavior of device mesh, now we use a mesh env to track the current mesh and save the mesh to a stack so that we can allow nested context manager Pull Request resolved: https://github.com/pytorch/pytorch/pull/101202 Approved by: https://github.com/wz337
Author
Committer
Parents
Loading