pytorch
d3788370 - [spmd] add more decomp and fix a sharding bug (#100938)

Commit
1 year ago
[spmd] add more decomp and fix a sharding bug (#100938) This PR adds native_layernorm_backward op to the decomp table and fixes a sharding bug to not automatically do padding Pull Request resolved: https://github.com/pytorch/pytorch/pull/100938 Approved by: https://github.com/mrshenli
Author
Committer
Parents
Loading