tree 0dde32ffc42e7b1e6c76b865028c658b8f48288e
parent b4ab8ac51526fa7bcd8bf865f00d1bfaafed0002
author Wanchao Liang <wanchaol@users.noreply.github.com> 1697342381 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1697385616 +0000

[dtensor][9/n] matrix ops to generate strategy (#110717)

This PR switches matrix ops to generate the sharding strategies, and
with the cost selection algorithm introduced in the previous PR we are
able to enable this and more ops to leverage strategy based sharding
prop

This also fixes a bunch of corner cases that existing propagation does
not cover, resulting in full coverage for baddbmm
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110717
Approved by: https://github.com/fduwjj
ghstack dependencies: #109145