add mutator for GetAddr

liuruyan · liuruyan · commit 5a7c7c25799b · 2025-04-11T02:32:37.000Z
diff --git a/paddle/cinn/ir/ir_mutator.h b/paddle/cinn/ir/ir_mutator.h
@@ -313,6 +313,10 @@ void IRMutator<T>::Visit(const IntrinsicOp *expr, T op) {
         Visit(&expr, &expr);
       }
     } break;
+    case ir::IntrinsicKind::kGetAddr: {
+      auto *n = llvm::dyn_cast<intrinsics::GetAddr>(node);
+      Visit(&n->data, &n->data);
+    } break;
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -313,6 +313,10 @@ void IRMutator<T>::Visit(const IntrinsicOp *expr, T op) {`
`313`	`313`	`Visit(&expr, &expr);`
`314`	`314`	`}`
`315`	`315`	`} break;`
	`316`	`+ case ir::IntrinsicKind::kGetAddr: {`
	`317`	`+ auto *n = llvm::dyn_cast<intrinsics::GetAddr>(node);`
	`318`	`+ Visit(&n->data, &n->data);`
	`319`	`+ } break;`
`316`	`320`	`}`
`317`	`321`	`}`
`318`	`322`