Skip to content

Commit d1b832e

Browse files
anijain2305pytorchmergebot
authored andcommitted
[inductor][mkl][inline-inbuilt-nn-modules] Change assertion (pytorch#130219)
Fixes the test in the next PR - `python test/inductor/test_mkldnn_pattern_matcher.py -k TestDynamicPatternMatcher.test_conv3d_unary_dynamic_shapes` Pull Request resolved: pytorch#130219 Approved by: https://github.com/leslie-fang-intel
1 parent 940e447 commit d1b832e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/_inductor/mkldnn_ir.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def _original_deconv_weight_size(
113113
else:
114114
assert 0 < len(output_padding) <= dims
115115
output_padding = pad_listlike(output_padding, dims)
116-
assert isinstance(groups, int)
116+
assert isinstance(groups, (int, sympy.core.numbers.Integer))
117117
if transposed:
118118
# When transposed, the size of the prepacked oneDNN weight is different
119119
# from the PyTorch weight. We're not able to run aten conv with such

0 commit comments

Comments
 (0)