From f71e6a4186cadc94431bdef5b6778f959d984925 Mon Sep 17 00:00:00 2001
From: hong <43953930+phlrain@users.noreply.github.com>
Date: Sat, 1 Jul 2023 18:33:19 +0800
Subject: [PATCH] Fix op compat info (#55032)

* fix op compat info

* fix find first of
---
 paddle/fluid/ir_adaptor/translator/op_compat_info.h | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/paddle/fluid/ir_adaptor/translator/op_compat_info.h b/paddle/fluid/ir_adaptor/translator/op_compat_info.h
index 799e62c7544..a899f993151 100644
--- a/paddle/fluid/ir_adaptor/translator/op_compat_info.h
+++ b/paddle/fluid/ir_adaptor/translator/op_compat_info.h
@@ -82,9 +82,10 @@ class OpNameNormalizer {
       std::string target = "_grad";
       std::string data = "@GRAD";
 
-      size_t first_grad_pos = arg_name.find_first_of(target);
-      std::string legacy_name =
-          this->GetLegacyArgName(op_type, arg_name.substr(0, first_grad_pos));
+      size_t first_grad_pos = arg_name.find(target);
+      size_t type_pos = op_type.find(target);
+      std::string legacy_name = this->GetLegacyArgName(
+          op_type.substr(0, type_pos), arg_name.substr(0, first_grad_pos));
       legacy_name += arg_name.substr(first_grad_pos);
       for (size_t pos = 0;
            legacy_name.npos != (pos = legacy_name.find(target, pos));
-- 
GitLab