From 0bcbfe8353ebd822ae23530b52f1693d50e8c960 Mon Sep 17 00:00:00 2001
From: ming1753 <61511741+ming1753@users.noreply.github.com>
Date: Fri, 7 Jul 2023 16:34:33 +0800
Subject: [PATCH] add odd rules for getting kernels (#55178)

---
 tools/get_op_list.py | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/tools/get_op_list.py b/tools/get_op_list.py
index 8d71af4a6e7..fcd8c5ef6fd 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
 
-- 
GitLab