Update flang/lib/Optimizer/CodeGen/TargetRewrite.cpp

clementval · web-flow · commit 846334f26516 · 2024-12-10T09:30:10.000-08:00
diff --git a/flang/lib/Optimizer/CodeGen/TargetRewrite.cpp b/flang/lib/Optimizer/CodeGen/TargetRewrite.cpp
@@ -726,7 +726,7 @@ class TargetRewrite : public fir::impl::TargetRewritePassBase<TargetRewrite> {
       if (targetFeaturesAttr)
         fn->setAttr("target_features", targetFeaturesAttr);
 
-      convertSignature<mlir::func::ReturnOp>(fn);
+      convertSignature<mlir::func::ReturnOp, mlir::func::FuncOp>(fn);
     }
 
     for (auto gpuMod : mod.getOps<mlir::gpu::GPUModuleOp>()) {

Original file line number	Diff line number	Diff line change
`@@ -726,7 +726,7 @@ class TargetRewrite : public fir::impl::TargetRewritePassBase<TargetRewrite> {`
`726`	`726`	`if (targetFeaturesAttr)`
`727`	`727`	`fn->setAttr("target_features", targetFeaturesAttr);`
`728`	`728`
`729`		`- convertSignature<mlir::func::ReturnOp>(fn);`
	`729`	`+ convertSignature<mlir::func::ReturnOp, mlir::func::FuncOp>(fn);`
`730`	`730`	`}`
`731`	`731`
`732`	`732`	`for (auto gpuMod : mod.getOps<mlir::gpu::GPUModuleOp>()) {`