summaryrefslogtreecommitdiff
path: root/0003-Add-python-bindings-for-kernels.patch
diff options
context:
space:
mode:
authorCoprDistGit <infra@openeuler.org>2025-11-12 09:24:22 +0000
committerCoprDistGit <infra@openeuler.org>2025-11-12 09:24:22 +0000
commit49b098769a059436d97d14b053d08f7d7e9ca9fe (patch)
tree19b8cfee8080e52235153fe3883c06ca2dc9bed3 /0003-Add-python-bindings-for-kernels.patch
parent2b940cc82a7c6490b3ca84d2981ba9afaac7d4ea (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.patch192
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
+