Skip to content

Commit d2f2977

Browse files
wanchaolwanchaol
committed
Update on "[dtensor][6/N] change to a better/safer op registration"
This PR changes the op registration to a better mechanism, now we require the directly overload registration instead of the op key str, this have several benefits: 1. We ensure that the op registration registers the correct op, which means it would be faild if the op registration become wrong (this PR already fixing several op registration errors as we use direct OpOverload registration 2. If the overload name get changed/deleted, we immediately know it at the source code compilation level, which is safer 3. This also keep it consistents with the op registration mechanism with other tensor subclasses within PyTorch Differential Revision: [D42876250](https://our.internmc.facebook.com/intern/diff/D42876250) [ghstack-poisoned]
2 parents eb70e35 + 25a87d3 commit d2f2977

File tree

0 file changed

+0
-0
lines changed

    0 file changed

    +0
    -0
    lines changed

    0 commit comments

    Comments
     (0)