Skip to content

Commit b14a37d

Browse files
committed
Fix after second rebase
1 parent 15dbcbc commit b14a37d

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

backends/nxp/tests/generic_tests/test_profiling.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,13 +56,14 @@ def forward(self, x):
5656

5757
class TestProfiling:
5858
@pytest.mark.xfail(reason="SoftMax support PR is not merged so far.", strict=True)
59-
def test__softmax(self, caplog):
59+
def test__softmax(self, caplog, request):
6060
caplog.set_level(logging.INFO)
6161
model = SoftmaxModule(-1)
6262
lower_run_compare(
6363
model,
6464
(10,),
6565
dlg_model_verifier=BaseGraphVerifier(1, []),
66+
request=request,
6667
use_profiling=True,
6768
output_comparator=NumericalStatsOutputComparator(),
6869
)
@@ -77,14 +78,15 @@ def test__softmax(self, caplog):
7778
3: (), # Neutron Dump
7879
}
7980

80-
def test__parallel_pool(self, caplog):
81+
def test__parallel_pool(self, caplog, request):
8182
caplog.set_level(logging.INFO)
8283
input_shape = (1, 3, 32, 32)
8384
model = ParallelPoolModel(input_shape[1])
8485
lower_run_compare(
8586
model,
8687
input_shape,
8788
dlg_model_verifier=BaseGraphVerifier(1, []),
89+
request=request,
8890
output_comparator=NumericalStatsOutputComparator(),
8991
use_neutron_for_format_conversion=False,
9092
use_profiling=True,
@@ -105,14 +107,15 @@ def test__parallel_pool(self, caplog):
105107
}
106108

107109
@pytest.mark.xfail(reason="SoftMax support PR is not merged so far.", strict=True)
108-
def test__cifar(self, caplog):
110+
def test__cifar(self, caplog, request):
109111
caplog.set_level(logging.INFO)
110112
input_shape = (1, 3, 32, 32)
111113
model = CifarNetModel()
112114
lower_run_compare(
113115
model,
114116
input_shape,
115117
dlg_model_verifier=BaseGraphVerifier(1, []),
118+
request=request,
116119
output_comparator=NumericalStatsOutputComparator(),
117120
use_neutron_for_format_conversion=False,
118121
use_profiling=True,
@@ -133,14 +136,15 @@ def test__cifar(self, caplog):
133136
11: (), # Neutron Dump
134137
}
135138

136-
def test__avg_pool(self, caplog):
139+
def test__avg_pool(self, caplog, request):
137140
caplog.set_level(logging.INFO)
138141
input_shape = (2, 9, 6, 15)
139142
model = AvgPool2dModule(False, 0)
140143
lower_run_compare(
141144
model,
142145
input_shape,
143146
dlg_model_verifier=BaseGraphVerifier(1, []),
147+
request=request,
144148
output_comparator=NumericalStatsOutputComparator(),
145149
use_neutron_for_format_conversion=False,
146150
use_profiling=True,

0 commit comments

Comments
 (0)