Skip to content

Commit a01db21

Browse files
Changed name of pass used in test.
Change-Id: If6bf02bcf5aec4da3d2c92bc51f8efd1337647a6
1 parent 07f19af commit a01db21

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

backends/arm/test/passes/test_decompose_softmaxes_pass.py renamed to backends/arm/test/passes/test_decompose_softmax_pass.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,7 @@
66
from typing import Tuple
77

88
import torch
9-
from executorch.backends.arm._passes.decompose_softmaxes_pass import (
10-
DecomposeSoftmaxesPass,
11-
)
9+
from executorch.backends.arm._passes.decompose_softmax_pass import DecomposeSoftmaxPass
1210

1311
from executorch.backends.arm.test.tester.test_pipeline import PassPipeline
1412

@@ -71,7 +69,7 @@ def test_softmax_basic_tosa_MI():
7169
"executorch_exir_dialects_edge__ops_aten_sum_dim_IntList": 1,
7270
},
7371
ops_not_after_pass=["executorch_exir_dialects_edge__ops_aten__softmax_default"],
74-
pass_list=[DecomposeSoftmaxesPass],
72+
pass_list=[DecomposeSoftmaxPass],
7573
)
7674
pipeline.run()
7775

@@ -100,6 +98,6 @@ def test_softmax_log_tosa_MI():
10098
ops_not_after_pass=[
10199
"executorch_exir_dialects_edge__ops_aten__log_softmax_default"
102100
],
103-
pass_list=[DecomposeSoftmaxesPass],
101+
pass_list=[DecomposeSoftmaxPass],
104102
)
105103
pipeline.run()

0 commit comments

Comments
 (0)