From 88d6a271e3ed66b28646f1797170010c412041d2 Mon Sep 17 00:00:00 2001
From: limingxing517 <limingxing517@163.com>
Date: Tue, 28 Feb 2023 16:44:01 +0800
Subject: [PATCH] update

---
 .../recommendation/WideDeep_ID2712_for_TensorFlow/train.py    | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/train.py b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/train.py
index 0865576db..241635268 100644
--- a/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/train.py
+++ b/TensorFlow/built-in/recommendation/WideDeep_ID2712_for_TensorFlow/train.py
@@ -335,6 +335,8 @@ if __name__ == '__main__':
     #autotune end
 
     custom_op.name = "NpuOptimizer"
+    custom_op.parameter_map["op_select_implmode"].s = tf.compat.as_bytes("high_performance")
+    custom_op.parameter_map["optypelist_for_implmode"].s = tf.compat.as_bytes("UnsortedSegmentSum,GatherV2")
     custom_op.parameter_map["enable_data_pre_proc"].b = True ##True getNext false在host侧
     #custom_op.parameter_map["mix_compile_mode"].b = True  #开启混合计算，根据实际情况配置
     custom_op.parameter_map["use_off_line"].b = True
@@ -540,4 +542,4 @@ if __name__ == '__main__':
         if mode == 'train' and rank_size > 1:
             sess.run(npu_shutdown)
          ############### for hccl $^?################
-        sess.close()
\ No newline at end of file
+        sess.close()
-- 
Gitee