File tree 1 file changed +3
-5
lines changed
1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change 6
6
from typing import Tuple
7
7
8
8
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
12
10
13
11
from executorch .backends .arm .test .tester .test_pipeline import PassPipeline
14
12
@@ -71,7 +69,7 @@ def test_softmax_basic_tosa_MI():
71
69
"executorch_exir_dialects_edge__ops_aten_sum_dim_IntList" : 1 ,
72
70
},
73
71
ops_not_after_pass = ["executorch_exir_dialects_edge__ops_aten__softmax_default" ],
74
- pass_list = [DecomposeSoftmaxesPass ],
72
+ pass_list = [DecomposeSoftmaxPass ],
75
73
)
76
74
pipeline .run ()
77
75
@@ -100,6 +98,6 @@ def test_softmax_log_tosa_MI():
100
98
ops_not_after_pass = [
101
99
"executorch_exir_dialects_edge__ops_aten__log_softmax_default"
102
100
],
103
- pass_list = [DecomposeSoftmaxesPass ],
101
+ pass_list = [DecomposeSoftmaxPass ],
104
102
)
105
103
pipeline .run ()
You can’t perform that action at this time.
0 commit comments