Skip to content

Commit c972289

Browse files
committed
Arm backend: Replace mxfp_linear_op import with ao_ext
Importing ao_ext will register all registered MXFP ops in `backends.arm.ao_ext` (instead of importing them one by one). This means that MXFP Conv2d is registered now as well. Signed-off-by: Martin Lindström <Martin.Lindstroem@arm.com> Change-Id: Ibca361cfd178a573d99ce633efe010d5473c3e05
1 parent 7772b8a commit c972289

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

backends/arm/operator_support/tosa_supported_operators.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,10 @@
1414
import typing
1515
from typing import final, Optional, Sequence, Type
1616

17+
# Register Arm-specific torch.library ops and MXFP transforms at package
18+
# import time.
19+
import executorch.backends.arm.ao_ext # noqa: F401
20+
1721
import torch
1822
import torch.fx as fx
1923
from executorch.backends.arm._passes.arm_pass_utils import (
@@ -27,7 +31,6 @@
2731
FuseQuantizedActivationPass,
2832
)
2933
from executorch.backends.arm._passes.insert_table_ops import TableOps
30-
from executorch.backends.arm.ao_ext.ops import mxfp_linear_op # noqa: F401
3134
from executorch.backends.arm.common.annotation_meta import ArmAnnotationInfo
3235
from executorch.backends.arm.constants import DQ_OPS, MAX_RANK, Q_OPS
3336
from executorch.backends.arm.operator_support.control_flow_support import (

0 commit comments

Comments
 (0)