Skip to content

Commit

Permalink
update test_auto_scan_reduce_ops
Browse files Browse the repository at this point in the history
  • Loading branch information
Zheng-Bicheng committed May 16, 2024
1 parent fadacc9 commit b8c1341
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tests/test_auto_scan_reduce_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,16 +21,16 @@
import random

op_api_map = {
"reduce_max": paddle.max,
# "reduce_max": paddle.max,
"reduce_min": paddle.min,
"reduce_mean": paddle.mean,
"reduce_sum": paddle.sum,
"reduce_prod": paddle.prod,
# "reduce_mean": paddle.mean,
# "reduce_sum": paddle.sum,
# "reduce_prod": paddle.prod,
}

opset_version_map = {
"reduce_max": [11, 12, 13, 18],
"reduce_min": [11, 12, 13, 18],
"reduce_max": [12, 13, 18],
"reduce_min": [12, 13, 18],
"reduce_mean": [11, 13, 18],
"reduce_sum": [13, 18],
# "reduce_prod": [11, 18],
Expand Down

0 comments on commit b8c1341

Please sign in to comment.