Expose header file utils.h op_compat_info.h (#68471)

0x3878f · web-flow · commit a5ca1f9d0d89 · 2024-09-27T10:44:49.000+08:00
* expose header file utils.h

* expose header file paddle/fluid/ir_adaptor/translator/op_compat_info.h

* Add getters in op_compat_info.h
diff --git a/paddle/fluid/ir_adaptor/translator/op_compat_info.h b/paddle/fluid/ir_adaptor/translator/op_compat_info.h
@@ -54,6 +54,26 @@ class OpNameNormalizer {
     return OpNameNormalizer;
   }
 
+  std::unordered_map<std::string, std::string> GetOpNameMappings() const {
+    return op_name_mappings;
+  }
+
+  std::unordered_map<std::string, std::unordered_map<std::string, std::string>>
+  GetOpArgNameMappings() const {
+    return op_arg_name_mappings;
+  }
+
+  std::unordered_map<std::string,
+                     std::unordered_map<std::string, MutableAttributeInfo>>
+  GetOpMutableAttributeInfos() const {
+    return op_mutable_attribute_infos;
+  }
+
+  std::unordered_map<std::string, std::unordered_set<std::string>>
+  GetOpMutableAttributes() const {
+    return op_mutable_attributes;
+  }
+
   std::string operator[](const std::string& op_type) {
     if (op_name_mappings.find(op_type) == op_name_mappings.end()) {
       return op_type;
diff --git a/python/setup.py.in b/python/setup.py.in
@@ -1131,7 +1131,9 @@ headers = (
     list(find_files('dense_tensor.inl','@PADDLE_SOURCE_DIR@/paddle/phi/core'))+
     list(find_files('op_yaml_info.h','@PADDLE_SOURCE_DIR@/paddle/fluid/pir/dialect/operator/interface'))+
     list(find_files('op_yaml_info_util.h','@PADDLE_SOURCE_DIR@/paddle/fluid/pir/dialect/operator/utils/'))+
-    list(find_files('op_yaml_info_parser.h','@PADDLE_SOURCE_DIR@/paddle/fluid/pir/dialect/operator/utils/')))
+    list(find_files('op_yaml_info_parser.h','@PADDLE_SOURCE_DIR@/paddle/fluid/pir/dialect/operator/utils/'))+
+    list(find_files('utils.h','@PADDLE_SOURCE_DIR@/paddle/fluid/pir/dialect/operator/utils/'))+
+    list(find_files('op_compat_info.h','@PADDLE_SOURCE_DIR@/paddle/fluid/ir_adaptor/translator/')))
 jit_layer_headers = ['layer.h', 'serializer.h', 'serializer_utils.h', 'all.h', 'function.h']
 for f in jit_layer_headers:
     headers += list(find_files(f, '@PADDLE_SOURCE_DIR@/paddle/fluid/jit', recursive=True))
diff --git a/setup.py b/setup.py
@@ -1742,12 +1742,24 @@ def get_headers():
                 paddle_source_dir + '/paddle/fluid/pir/dialect/operator/utils',
             )
         )
-        + list(  # serialize and deserialize interface headers
+        + list(  # op yaml parser interface headers
             find_files(
                 'op_yaml_info_parser.h',
                 paddle_source_dir + '/paddle/fluid/pir/dialect/operator/utils',
             )
         )
+        + list(  # pir op utils interface headers
+            find_files(
+                'utils.h',
+                paddle_source_dir + '/paddle/fluid/pir/dialect/operator/utils',
+            )
+        )
+        + list(  # ir translator interface headers
+            find_files(
+                'op_compat_info.h',
+                paddle_source_dir + '/paddle/fluid/ir_adaptor/translator/',
+            )
+        )
     )
 
     jit_layer_headers = [

Original file line number	Diff line number	Diff line change
`@@ -1742,12 +1742,24 @@ def get_headers():`
`1742`	`1742`	`paddle_source_dir + '/paddle/fluid/pir/dialect/operator/utils',`
`1743`	`1743`	`)`
`1744`	`1744`	`)`
`1745`		`- + list( # serialize and deserialize interface headers`
	`1745`	`+ + list( # op yaml parser interface headers`
`1746`	`1746`	`find_files(`
`1747`	`1747`	`'op_yaml_info_parser.h',`
`1748`	`1748`	`paddle_source_dir + '/paddle/fluid/pir/dialect/operator/utils',`
`1749`	`1749`	`)`
`1750`	`1750`	`)`
	`1751`	`+ + list( # pir op utils interface headers`
	`1752`	`+ find_files(`
	`1753`	`+ 'utils.h',`
	`1754`	`+ paddle_source_dir + '/paddle/fluid/pir/dialect/operator/utils',`
	`1755`	`+ )`
	`1756`	`+ )`
	`1757`	`+ + list( # ir translator interface headers`
	`1758`	`+ find_files(`
	`1759`	`+ 'op_compat_info.h',`
	`1760`	`+ paddle_source_dir + '/paddle/fluid/ir_adaptor/translator/',`
	`1761`	`+ )`
	`1762`	`+ )`
`1751`	`1763`	`)`
`1752`	`1764`
`1753`	`1765`	`jit_layer_headers = [`