diff --git a/tools/get_op_list.py b/tools/get_op_list.py index 8d71af4a6e7e76c9180308575dbc5b5798e3e5d5..fcd8c5ef6fd09a651f928d7320571df7ab8026ec 100644 --- a/tools/get_op_list.py +++ b/tools/get_op_list.py @@ -55,10 +55,28 @@ def get_model_ops(model_file, ops_set): def get_model_phi_kernels(ops_set): phi_set = set() + phi_raw_list = [ + "add", + "subtract", + "multiply", + "multiply_sr", + "divide", + "maximum", + "minimum", + "remainder", + "floor_divide", + "elementwise_pow", + ] + phi_odd_dist = {"batch_norm": "batch_norm_infer"} for op in ops_set: print(op) - print(_get_phi_kernel_name(op)) - phi_set.add(_get_phi_kernel_name(op)) + phi_kernel = _get_phi_kernel_name(op) + print(phi_kernel) + phi_set.add(phi_kernel) + if phi_kernel in phi_raw_list: + phi_set.add(phi_kernel + "_raw") + if phi_kernel in phi_odd_dist.keys(): + phi_set.add(phi_odd_dist[phi_kernel]) return phi_set