fix HasAttr of mapper

PaddlePaddle · Feb 7, 2025 · 3332c80 · 3332c80
1 parent d111ea4
commit 3332c80
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/paddle2onnx/mapper/mapper.h b/paddle2onnx/mapper/mapper.h
@@ -302,10 +302,12 @@ class Mapper {
     if (in_pir_mode) {
       if (if_in_cf_block) {
         auto op = pir_parser_->sub_blocks_ops[pir_op_idx_];
-        return pir_parser_->OpHasAttr(op, name);
+        return pir_parser_->OpHasAttr(
+            op, pir_parser_->GetOpArgName(pir_op_idx_, name, true));
       } else {
         auto op = pir_parser_->global_blocks_ops[pir_op_idx_];
-        return pir_parser_->OpHasAttr(op, name);
+        return pir_parser_->OpHasAttr(
+            op, pir_parser_->GetOpArgName(pir_op_idx_, name, false));
       }
     } else {
       auto& op = parser_->GetOpDesc(block_idx_, op_idx_);