pytorch
dcec2100 - [dtensor] add placement strategy and einsum strategy (#98227)

Commit
2 years ago
[dtensor] add placement strategy and einsum strategy (#98227) This adds placement strategy to the op schema and implement einsum strategy. It's the basic building piece for compile mode expansion and new op implementation Pull Request resolved: https://github.com/pytorch/pytorch/pull/98227 Approved by: https://github.com/XilunWu
Author
Committer
Parents
Loading