diff --git a/benchmark/opperf/utils/op_registry_utils.py b/benchmark/opperf/utils/op_registry_utils.py index 4e3e7eb1264f..43158dbc4edf 100644 --- a/benchmark/opperf/utils/op_registry_utils.py +++ b/benchmark/opperf/utils/op_registry_utils.py @@ -222,8 +222,6 @@ def get_all_misc_binary_operators(): binary_misc_mx_operators[op_name] = mx_operators[op_name] elif "reshape_like" == op_name: binary_misc_mx_operators[op_name] = mx_operators[op_name] - elif "ElementWiseSum" == op_name: - binary_misc_mx_operators[op_name] = mx_operators[op_name] return binary_misc_mx_operators @@ -244,6 +242,8 @@ def get_all_elemen_wise_binary_operators(): "lhs" in op_params["params"]["arg_names"] and \ "rhs" in op_params["params"]["arg_names"]: binary_elemen_wise_mx_operators[op_name] = mx_operators[op_name] + elif "ElementWiseSum" == op_name: + binary_elemen_wise_mx_operators[op_name] = mx_operators[op_name] return binary_elemen_wise_mx_operators