From e7d504a58608e79eec40d04989fade55d9a9a592 Mon Sep 17 00:00:00 2001 From: majunwang <9018303+majunwang@user.noreply.gitee.com> Date: Thu, 29 May 2025 02:14:22 +0000 Subject: [PATCH 1/4] =?UTF-8?q?update=20tf=5Fadapter/ops/aicore/npu=5Faico?= =?UTF-8?q?re=5Fops.cc.=20embdding=5Fhash=5Ftable=5Flook=5Fup=5For=5Finser?= =?UTF-8?q?t=E7=AE=97=E5=AD=90=E6=96=B0=E5=A2=9E=E5=B1=9E=E6=80=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: majunwang <9018303+majunwang@user.noreply.gitee.com> --- tf_adapter/ops/aicore/npu_aicore_ops.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tf_adapter/ops/aicore/npu_aicore_ops.cc b/tf_adapter/ops/aicore/npu_aicore_ops.cc index bb5e99e7f..06973394f 100644 --- a/tf_adapter/ops/aicore/npu_aicore_ops.cc +++ b/tf_adapter/ops/aicore/npu_aicore_ops.cc @@ -527,6 +527,8 @@ REGISTER_OP("EmbeddingHashTableLookupOrInsert") .Attr("default_key_or_value:bool = false") .Attr("default_key: int = 0") .Attr("default_value: float = 0.0") + .Attr("filter_key_flag:bool = false") + .Attr("filter_key: int = -1") .SetIsStateful() .SetShapeFn([](InferenceContext* c) { int64 num = 0; -- Gitee From 7a58cddacc50c6c5757ec2ab9cf13f8c266e3b09 Mon Sep 17 00:00:00 2001 From: majunwang <9018303+majunwang@user.noreply.gitee.com> Date: Thu, 29 May 2025 02:17:28 +0000 Subject: [PATCH 2/4] =?UTF-8?q?update=20tf=5Fadapter/python/npu=5Fbridge/n?= =?UTF-8?q?pu=5Fcpu/npu=5Fcpu=5Fops.py.=20embdding=5Fhash=5Ftable=5Flook?= =?UTF-8?q?=5Fup=5For=5Finsert=E7=AE=97=E5=AD=90=E6=96=B0=E5=A2=9E?= =?UTF-8?q?=E5=B1=9E=E6=80=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: majunwang <9018303+majunwang@user.noreply.gitee.com> --- tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py b/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py index e7c5fa61f..b319f5e90 100644 --- a/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py +++ b/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py @@ -37,12 +37,12 @@ gen_npu_cpu_ops = helper.get_gen_ops() # @param default_value float 类型 # @return values float 类型 def embedding_hashtable_lookup_or_insert(table_handle, keys, bucket_size, embedding_dim, filter_mode, filter_freq, - default_key_or_value, default_key, default_value): + default_key_or_value, default_key, default_value, filter_key_flag, filter_key): """ device embedding feature mapping lookup or insert. """ result = gen_npu_cpu_ops.EmbeddingHashTableLookupOrInsert( table_handle=table_handle, keys=keys, bucket_size=bucket_size, embedding_dim=embedding_dim, filter_mode=filter_mode, filter_freq=filter_freq, default_key_or_value=default_key_or_value, - default_key=default_key, default_value=default_value) + default_key=default_key, default_value=default_value,filter_key_flag=filter_key_flag,filter_key=filter_key) return result -- Gitee From 4c95b4d754ad66ace1d386e92e935d77d616280d Mon Sep 17 00:00:00 2001 From: majunwang <9018303+majunwang@user.noreply.gitee.com> Date: Thu, 29 May 2025 02:22:25 +0000 Subject: [PATCH 3/4] update tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py. Signed-off-by: majunwang <9018303+majunwang@user.noreply.gitee.com> --- tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py b/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py index b319f5e90..8b79fb817 100644 --- a/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py +++ b/tf_adapter/python/npu_bridge/npu_cpu/npu_cpu_ops.py @@ -42,7 +42,7 @@ def embedding_hashtable_lookup_or_insert(table_handle, keys, bucket_size, embedd result = gen_npu_cpu_ops.EmbeddingHashTableLookupOrInsert( table_handle=table_handle, keys=keys, bucket_size=bucket_size, embedding_dim=embedding_dim, filter_mode=filter_mode, filter_freq=filter_freq, default_key_or_value=default_key_or_value, - default_key=default_key, default_value=default_value,filter_key_flag=filter_key_flag,filter_key=filter_key) + default_key=default_key, default_value=default_value, filter_key_flag=filter_key_flag, filter_key=filter_key) return result -- Gitee From 9b6ef22f2c131148630d1125ed416b8235da29f6 Mon Sep 17 00:00:00 2001 From: majunwang <9018303+majunwang@user.noreply.gitee.com> Date: Thu, 29 May 2025 02:24:02 +0000 Subject: [PATCH 4/4] update tf_adapter/ops/aicore/npu_aicore_ops.cc. Signed-off-by: majunwang <9018303+majunwang@user.noreply.gitee.com> --- tf_adapter/ops/aicore/npu_aicore_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_adapter/ops/aicore/npu_aicore_ops.cc b/tf_adapter/ops/aicore/npu_aicore_ops.cc index 06973394f..91913fd56 100644 --- a/tf_adapter/ops/aicore/npu_aicore_ops.cc +++ b/tf_adapter/ops/aicore/npu_aicore_ops.cc @@ -527,7 +527,7 @@ REGISTER_OP("EmbeddingHashTableLookupOrInsert") .Attr("default_key_or_value:bool = false") .Attr("default_key: int = 0") .Attr("default_value: float = 0.0") - .Attr("filter_key_flag:bool = false") + .Attr("filter_key_flag: bool = false") .Attr("filter_key: int = -1") .SetIsStateful() .SetShapeFn([](InferenceContext* c) { -- Gitee