diff options
| author | CoprDistGit <infra@openeuler.org> | 2025-11-13 13:32:56 +0000 |
|---|---|---|
| committer | CoprDistGit <infra@openeuler.org> | 2025-11-13 13:32:56 +0000 |
| commit | 64b81c67daaf63b283970a385e69cf77a8ddd74c (patch) | |
| tree | 19b8cfee8080e52235153fe3883c06ca2dc9bed3 /0003-Add-python-bindings-for-kernels.patch | |
| parent | 5563f430e2858b98d8d1e87745611e3e80347359 (diff) | |
automatic import of ANNCopeneuler22.03_LTS_SP4
Diffstat (limited to '0003-Add-python-bindings-for-kernels.patch')
| -rw-r--r-- | 0003-Add-python-bindings-for-kernels.patch | 192 |
1 files changed, 192 insertions, 0 deletions
diff --git a/0003-Add-python-bindings-for-kernels.patch b/0003-Add-python-bindings-for-kernels.patch new file mode 100644 index 0000000..22f70dc --- /dev/null +++ b/0003-Add-python-bindings-for-kernels.patch @@ -0,0 +1,192 @@ +From 97c972045ab8bf8f4f4db39da0728b2c41183eff Mon Sep 17 00:00:00 2001 +From: zhengchenhui <zhengchenhui1@huawei.com> +Date: Wed, 12 Nov 2025 16:38:50 +0800 +Subject: [PATCH] Add python bindings for kernels. + +--- + ...i_def_KPFusedEmbeddingActionIdGather.pbtxt | 3 ++ + .../api_def_KPFusedEmbeddingPadding.pbtxt | 3 ++ + .../api_def_KPFusedEmbeddingPaddingFast.pbtxt | 3 ++ + .../api_def/api_def_KPFusedGather.pbtxt | 3 ++ + .../api_def_KPFusedSparseDynamicStitch.pbtxt | 3 ++ + .../api_def_KPFusedSparseReshape.pbtxt | 3 ++ + .../api_def_KPFusedSparseSegmentReduce.pbtxt | 3 ++ + ...ef_KPFusedSparseSegmentReduceNonzero.pbtxt | 3 ++ + .../api_def/api_def_KPFusedSparseSelect.pbtxt | 3 ++ + annc/tensorflow/tf_annc_optimizer.patch | 53 +++++++++++++++++-- + 10 files changed, 76 insertions(+), 4 deletions(-) + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedEmbeddingActionIdGather.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedEmbeddingPadding.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedEmbeddingPaddingFast.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedGather.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedSparseDynamicStitch.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedSparseReshape.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduce.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduceNonzero.pbtxt + create mode 100644 annc/tensorflow/api_def/api_def_KPFusedSparseSelect.pbtxt + +diff --git a/annc/tensorflow/api_def/api_def_KPFusedEmbeddingActionIdGather.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedEmbeddingActionIdGather.pbtxt +new file mode 100644 +index 0000000..bb221ad +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedEmbeddingActionIdGather.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedEmbeddingActionIdGather" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedEmbeddingPadding.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedEmbeddingPadding.pbtxt +new file mode 100644 +index 0000000..5aef378 +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedEmbeddingPadding.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedEmbeddingPadding" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedEmbeddingPaddingFast.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedEmbeddingPaddingFast.pbtxt +new file mode 100644 +index 0000000..17e7ebc +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedEmbeddingPaddingFast.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedEmbeddingPaddingFast" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedGather.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedGather.pbtxt +new file mode 100644 +index 0000000..63c4eef +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedGather.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedGather" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedSparseDynamicStitch.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedSparseDynamicStitch.pbtxt +new file mode 100644 +index 0000000..5b01d70 +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedSparseDynamicStitch.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedSparseDynamicStitch" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedSparseReshape.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedSparseReshape.pbtxt +new file mode 100644 +index 0000000..8fe3c2f +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedSparseReshape.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedSparseReshape" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduce.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduce.pbtxt +new file mode 100644 +index 0000000..b13aa7d +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduce.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedSparseSegmentReduce" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduceNonzero.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduceNonzero.pbtxt +new file mode 100644 +index 0000000..c0945af +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedSparseSegmentReduceNonzero.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedSparseSegmentReduceNonzero" ++} +diff --git a/annc/tensorflow/api_def/api_def_KPFusedSparseSelect.pbtxt b/annc/tensorflow/api_def/api_def_KPFusedSparseSelect.pbtxt +new file mode 100644 +index 0000000..bafc915 +--- /dev/null ++++ b/annc/tensorflow/api_def/api_def_KPFusedSparseSelect.pbtxt +@@ -0,0 +1,3 @@ ++op { ++ graph_op_name: "KPFusedSparseSelect" ++} +diff --git a/annc/tensorflow/tf_annc_optimizer.patch b/annc/tensorflow/tf_annc_optimizer.patch +index c92330a..ec0199e 100644 +--- a/annc/tensorflow/tf_annc_optimizer.patch ++++ b/annc/tensorflow/tf_annc_optimizer.patch +@@ -1,5 +1,5 @@ + diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD +-index 538574360ba..b43e0455802 100644 ++index 538574360ba..3864064d18c 100644 + --- a/tensorflow/core/BUILD + +++ b/tensorflow/core/BUILD + @@ -629,6 +629,7 @@ cc_library( +@@ -10,6 +10,14 @@ index 538574360ba..b43e0455802 100644 + ] + if_mkl([ + "//tensorflow/core/kernels/mkl:mkl_concat_op", + "//tensorflow/core/kernels/mkl:mkl_dequantize_op", ++@@ -907,6 +908,7 @@ filegroup( ++ "candidate_sampling_ops_op_lib", ++ "checkpoint_ops_op_lib", ++ "clustering_ops_op_lib", +++ "embedding_fused_ops_op_lib", ++ "collective_ops_op_lib", ++ "control_flow_ops_op_lib", ++ "count_ops_op_lib", + diff --git a/tensorflow/core/grappler/optimizers/BUILD b/tensorflow/core/grappler/optimizers/BUILD + index ecd559734ea..97a918ead6d 100644 + --- a/tensorflow/core/grappler/optimizers/BUILD +@@ -189,6 +197,45 @@ index 91d80b6c2b5..b00b2a5d027 100644 + ] + select({ + # Non-tpu platforms don't need tpu dependency. + "//tensorflow:chromiumos": [], ++diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD ++index 9810f8acd2e..3b0b23fec0a 100644 ++--- a/tensorflow/python/BUILD +++++ b/tensorflow/python/BUILD ++@@ -269,6 +269,7 @@ py_strict_library( ++ "//tensorflow/python/ops/signal:util_ops", ++ "//tensorflow/python/ops/signal:window_ops", ++ "//tensorflow/python/ops/structured:structured_ops", +++ "//tensorflow/python/ops:embedding_fused_ops_gen", ++ "//tensorflow/python/platform:_pywrap_stacktrace_handler", ++ "//tensorflow/python/platform:app", ++ "//tensorflow/python/platform:client_testlib", ++@@ -389,6 +390,7 @@ py_strict_library( ++ "//tensorflow/python/ops:bitwise_ops", ++ "//tensorflow/python/ops:boosted_trees_ops_gen", ++ "//tensorflow/python/ops:clustering_ops_gen", +++ "//tensorflow/python/ops:embedding_fused_ops_gen", ++ "//tensorflow/python/ops:composite_tensor_ops", ++ "//tensorflow/python/ops:cond_v2", ++ "//tensorflow/python/ops:cudnn_rnn_ops_gen", ++diff --git a/tensorflow/python/ops/BUILD b/tensorflow/python/ops/BUILD ++index b3a3d612a01..185da7da4a8 100644 ++--- a/tensorflow/python/ops/BUILD +++++ b/tensorflow/python/ops/BUILD ++@@ -228,6 +228,14 @@ tf_gen_op_strict_wrapper_private_py( ++ ], ++ ) ++ +++tf_gen_op_strict_wrapper_private_py( +++ name = "embedding_fused_ops_gen", +++ visibility = ["//tensorflow:internal"], +++ deps = [ +++ "//tensorflow/core:embedding_fused_ops_op_lib", +++ ], +++) +++ ++ tf_gen_op_strict_wrapper_private_py( ++ name = "collective_ops_gen", ++ visibility = ["//tensorflow:internal"], + diff --git a/third_party/xla/xla/service/cpu/BUILD b/third_party/xla/xla/service/cpu/BUILD + index 6e0ea613435..47c346b4e93 100644 + --- a/third_party/xla/xla/service/cpu/BUILD +@@ -216,6 +263,4 @@ index 6e0ea613435..47c346b4e93 100644 + + + cc_library( + name = "cpu_compiler_pure", +- srcs = ["cpu_compiler.cc"], +- +- ++ srcs = ["cpu_compiler.cc"], +\ No newline at end of file +-- +2.33.0 + |
