From 1692af99eade96eba49830b4712feee695557d74 Mon Sep 17 00:00:00 2001
From: kangguangli <kangguangli@hotmail.com>
Date: Wed, 30 Aug 2023 14:22:10 +0800
Subject: [PATCH] [NewIR] add_arg_mapping_for_fetch (#56752)

* add_arg_mapping_for_fetch

* fix

* fix
---
 paddle/fluid/ir_adaptor/translator/op_compat_gen.py | 1 +
 paddle/fluid/ir_adaptor/translator/op_translator.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/paddle/fluid/ir_adaptor/translator/op_compat_gen.py b/paddle/fluid/ir_adaptor/translator/op_compat_gen.py
index 7c1a78305e7..60a4a092040 100644
--- a/paddle/fluid/ir_adaptor/translator/op_compat_gen.py
+++ b/paddle/fluid/ir_adaptor/translator/op_compat_gen.py
@@ -128,6 +128,7 @@ def OpNameNormalizerInitialization(
 
     # special mapping list
     op_arg_name_mappings["set_value_grad"]["values_grad"] = "ValueTensor@GRAD"
+    op_arg_name_mappings["fetch"] = {"x": "X"}
 
     op_name_normailzer_template = env.get_template("op_compat_info.cc.j2")
     with open(output_source_file, 'wt') as f:
diff --git a/paddle/fluid/ir_adaptor/translator/op_translator.cc b/paddle/fluid/ir_adaptor/translator/op_translator.cc
index cad900ee7f6..38d833fc312 100644
--- a/paddle/fluid/ir_adaptor/translator/op_translator.cc
+++ b/paddle/fluid/ir_adaptor/translator/op_translator.cc
@@ -1699,6 +1699,7 @@ OpTranslator::OpTranslator() {
   special_handlers["cast"] = CastOpTranscriber();
   special_handlers["feed"] = FeedOpTranscriber();
   special_handlers["data"] = DataOpTranscriber();
+  special_handlers["fetch"] = FetchOpTranscriber();
   special_handlers["fetch_v2"] = FetchOpTranscriber();
   special_handlers["fill_constant"] = FillConstantTranscriber();
   special_handlers["grad_add"] = GradAddOpTranscriber();
-- 
GitLab