8000 Refine test · pytorch/pytorch@90c1011 · GitHub
[go: up one dir, main page]
Skip to content

Commit 90c1011

Browse files
Xia-WeiwenXia-Weiwen
committed
Refine test
1 parent 008923d commit 90c1011

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

‎test/inductor/test_mkldnn_pattern_matcher.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1815,27 +1815,27 @@ def matcher_check_fn():
18151815
matcher_check_fn=matcher_check_fn,
18161816
is_qat=is_qat,
18171817
)
1818-
try:
1819-
# For python wrapper
1818+
if torch._inductor.config.cpp_wrapper:
1819+
# For CPP wrapper
18201820
self._test_code_common(
18211821
mod,
18221822
(v,),
18231823
[
1824-
"torch.ops.onednn.qlinear_pointwise.default",
1825-
"torch.ops.onednn.qlinear_pointwise.binary",
1824+
"op_qlinear_pointwise.call",
1825+
"op_qlinear_pointwise_binary.call",
18261826
],
18271827
[],
18281828
check_quantization=True,
18291829
num_include_ops=[2, 2],
18301830
)
1831-
except AssertionError:
1832-
# For CPP wrapper
1831+
else:
1832+
# For python wrapper
18331833
self._test_code_common(
18341834
mod,
18351835
(v,),
18361836
[
1837-
"op_qlinear_pointwise.call",
1838-
"op_qlinear_pointwise_binary.call",
1837+
"torch.ops.onednn.qlinear_pointwise.default",
1838+
"torch.ops.onednn.qlinear_pointwise.binary",
18391839
],
18401840
[],
18411841
check_quantization=True,

0 commit comments

Comments
 (0)
0