From b75a83039d308839db6f738789eabd58096b2ad8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E7=B2=BE=E7=A7=91?= <2351312060@qq.com>
Date: Wed, 30 Mar 2022 03:20:19 +0000
Subject: [PATCH 1/2] 
 /MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg.

---
 .../train/fusion_switch.cfg                              | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg

diff --git a/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg
new file mode 100644
index 000000000..57e2c717c
--- /dev/null
+++ b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/fusion_switch.cfg
@@ -0,0 +1,9 @@
+{
+    "Switch":{
+        "GraphFusion":{
+            "RealDiv2MulsFusionPass":"off"
+        },
+        "UBFusion":{
+        }
+    }
+}
-- 
Gitee


From 6e242c9b69ebc53d00527f68c04ada4415a25c77 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E7=B2=BE=E7=A7=91?= <2351312060@qq.com>
Date: Wed, 30 Mar 2022 03:22:15 +0000
Subject: [PATCH 2/2] /MMoE_Transformer_ID0390_for_TensorFlow/train/main.py.

---
 .../MMoE_Transformer_ID0390_for_TensorFlow/train/main.py     | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py
index ae177ccdf..ff30bfc74 100644
--- a/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py
+++ b/TensorFlow/built-in/recommendation/MMoE_Transformer_ID0390_for_TensorFlow/train/main.py
@@ -254,7 +254,10 @@ if __name__ == '__main__':
         custom_op.parameter_map["use_off_line"].b = True
         custom_op.parameter_map["precision_mode"].s = tf.compat.as_bytes('allow_mix_precision')
         custom_op.parameter_map["dynamic_input"].b = True
-        custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile") 
+        custom_op.parameter_map["dynamic_graph_execute_mode"].s = tf.compat.as_bytes("lazy_recompile")
+        pwdname = os.getcwd()
+        swith_file = pwdname+"/train/fusion_switch.cfg"
+        custom_op.parameter_map["fusion_switch_file"].s = tf.compat.as_bytes(swith_file) 
         if iterations_per_loop_mode:
             custom_op.parameter_map["iterations_per_loop"].i = 10
         sess_config.graph_options.rewrite_options.remapping = RewriterConfig.OFF
-- 
Gitee