Skip to content

Commit 963ac9b

Browse files
authored
fixed reduce bug (#880)
1 parent d29125f commit 963ac9b

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

tests/onnx/test_auto_scan_reduce_ops.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,11 @@
1919
import unittest
2020
import random
2121

22+
min_opset_version_map = {
23+
"ReduceL1": 7,
24+
"ReduceL2": 7,
25+
}
26+
2227

2328
class TestReduceOpsConvert(OPConvertAutoScanTest):
2429
"""
@@ -30,7 +35,7 @@ def sample_convert_config(self, draw):
3035
input_shape = draw(
3136
st.lists(
3237
st.integers(
33-
min_value=20, max_value=30), min_size=3, max_size=5))
38+
min_value=10, max_value=20), min_size=3, max_size=5))
3439

3540
input_dtype = draw(st.sampled_from(["float32", "int32", "int64"]))
3641

@@ -55,6 +60,10 @@ def sample_convert_config(self, draw):
5560
"delta": 1e-4,
5661
"rtol": 1e-4,
5762
}
63+
min_opset_versions = list()
64+
for op_name in config["op_names"]:
65+
min_opset_versions.append(min_opset_version_map[op_name])
66+
config["min_opset_version"] = min_opset_versions
5867

5968
attrs = {
6069
"axes": axes,

0 commit comments

Comments
 (0)