Skip to content

Commit e3b0b33

Browse files
Arm backend: Add support for aten.erfinv.default (pytorch#17952)
Change-Id: I63613ecd98d1187ef5c4c0509fc1afc88f8ed47f Signed-off-by: Yufeng Shi <yufeng.shi@arm.com>
1 parent eabe49e commit e3b0b33

8 files changed

Lines changed: 506 additions & 0 deletions

File tree

backends/arm/_passes/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
from .decompose_div_tensor_mode import DecomposeDivTensorModePass # noqa
4646
from .decompose_elu_pass import DecomposeEluPass # noqa
4747
from .decompose_embedding_pass import DecomposeEmbeddingPass # noqa # noqa
48+
from .decompose_erfinv_pass import DecomposeErfinvPass # noqa
4849
from .decompose_expm1_pass import DecomposeExpm1Pass # noqa
4950
from .decompose_floor_divide_pass import DecomposeFloorDividePass # noqa
5051
from .decompose_gelu_pass import DecomposeGeluPass # noqa

backends/arm/_passes/arm_pass_manager.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@
5252
DecomposeDivTensorModePass,
5353
DecomposeEluPass,
5454
DecomposeEmbeddingPass,
55+
DecomposeErfinvPass,
5556
DecomposeExpm1Pass,
5657
DecomposeFloorDividePass,
5758
DecomposeGeluPass,
@@ -283,6 +284,7 @@ def _tosa_pipeline(
283284
DecomposeAsinhPass(),
284285
DecomposeCoshPass(),
285286
DecomposeAsinAndAcosPass(),
287+
DecomposeErfinvPass(),
286288
DecomposeSqrtPass(),
287289
DecomposeAtanPass(),
288290
DecomposeAtanhPass(),

0 commit comments

Comments
 (0)