File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 4
4
# This source code is licensed under the BSD-style license found in the
5
5
# LICENSE file in the root directory of this source tree.
6
6
7
+ import logging
7
8
import unittest
8
9
9
10
from typing import Tuple
13
14
from executorch .backends .arm .test .test_models import TosaProfile
14
15
from executorch .backends .arm .test .tester .arm_tester import ArmBackendSelector , ArmTester
15
16
17
+ logger = logging .getLogger (__name__ )
18
+ logger .setLevel (logging .INFO )
19
+
16
20
"""
17
21
This file contain unit tests where conv are combined with other ops.
18
22
"""
@@ -145,7 +149,7 @@ def _test_conv_combo_tosa_MI_pipeline(
145
149
if common .TOSA_REF_MODEL_INSTALLED :
146
150
tester .run_method ().compare_outputs ()
147
151
else :
148
- common . logger .warning (
152
+ logger .warning (
149
153
"TOSA ref model tool not installed, skip numerical correctness tests"
150
154
)
151
155
@@ -175,7 +179,7 @@ def _test_conv_combo_tosa_BI_pipeline(
175
179
if common .TOSA_REF_MODEL_INSTALLED :
176
180
tester .run_method ().compare_outputs (atol = atol , rtol = rtol , qtol = 1 )
177
181
else :
178
- common . logger .warning (
182
+ logger .warning (
179
183
"TOSA ref model tool not installed, skip numerical correctness tests"
180
184
)
181
185
You can’t perform that action at this time.
0 commit comments