Skip to content

Commit 3947a7f

Browse files
committed
mxtensor: switch to AOBaseTensor dispatch
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: 9dd0007 ghstack-comment-id: 3338749802 Pull Request resolved: #3080
1 parent 0544eb3 commit 3947a7f

File tree

3 files changed

+304
-363
lines changed

3 files changed

+304
-363
lines changed

torchao/prototype/mx_formats/mx_ops.py

Lines changed: 0 additions & 344 deletions
This file was deleted.

0 commit comments

Comments
 (0)