You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Summary:
Deletes the custom op dispatch logic in `MXTensor` and switches to the
general one in `AOBaseTensor`.
To enable this, we move the mx ops to the same file as `MXTensor`. This
is to avoid the need for callsites to separately import `mx_ops.py`.
Note that there are a couple of custom mx ops which could instead use
the general implementation, leaving that for future PRs to keep this one
small.
Test Plan:
```
pytest test/prototype/mx_formats -s -x
```
Reviewers:
Subscribers:
Tasks:
Tags:
ghstack-source-id: 994257e
ghstack-comment-id: 3338749802
Pull Request resolved: #3080
0 commit comments