Make EliminateLibm work on EmitC::FuncOp

Author: josel-amd

mlir/include/mlir/Dialect/EmitC/IR/EmitC.td

@@ -752,6 +752,12 @@ def EmitC_FuncOp : EmitC_Op<"func", [
 
     /// Returns the result types of this function.
     ArrayRef<Type> getResultTypes() { return getFunctionType().getResults(); }
+
+    //===------------------------------------------------------------------===//
+    // SymbolOpInterface Methods
+    //===------------------------------------------------------------------===//
+
+    bool isDeclaration() { return isExternal(); }
   }];
   let hasCustomAssemblyFormat = 1;
   let hasVerifier = 1;

mlir/lib/Dialect/EmitC/Transforms/EliminateLibm.cpp

@@ -37,9 +37,9 @@ namespace {
 
 /// Replace all Libm calls (where callee has `libm` attribute + no definition)
 /// by opaque calls
-struct OpacifyLibmCall : public OpRewritePattern<func::CallOp> {
-  using OpRewritePattern<func::CallOp>::OpRewritePattern;
-  LogicalResult matchAndRewrite(func::CallOp callOp,
+struct OpacifyLibmCall : public OpRewritePattern<emitc::CallOp> {
+  using OpRewritePattern<emitc::CallOp>::OpRewritePattern;
+  LogicalResult matchAndRewrite(emitc::CallOp callOp,
                                 PatternRewriter &rewriter) const override {
 
     auto *st = SymbolTable::getNearestSymbolTable(callOp);
@@ -65,8 +65,8 @@ struct EliminateLibmPass : public impl::EliminateLibmBase<EliminateLibmPass> {
     MLIRContext *context = module->getContext();
 
     // Find the first math.h inclusion
-    SmallVector<func::FuncOp> libmPrototypes;
-    module.walk([&libmPrototypes](func::FuncOp funcOp) {
+    SmallVector<emitc::FuncOp> libmPrototypes;
+    module.walk([&libmPrototypes](emitc::FuncOp funcOp) {
       if (funcOp->hasAttr("libm") && funcOp.isDeclaration())
         libmPrototypes.push_back(funcOp);
     });

mlir/test/Dialect/EmitC/eliminate_libm.mlir

@@ -1,29 +1,29 @@
 // RUN: mlir-opt %s --eliminate-libm --verify-diagnostics --split-input-file | FileCheck %s
 
 // CHECK: emitc.include <"cmath">
-// CHECK-NOT: func.func private @expm1
-// CHECK-DAG: func.func @call_expm1(%[[IN:.*]]: f64) -> f64
+// CHECK-NOT: emitc.func private @expm1
+// CHECK-DAG: emitc.func @call_expm1(%[[IN:.*]]: f64) -> f64
 // CHECK-DAG: %[[RESULT:.*]] = emitc.call_opaque "expm1"(%[[IN]]) : (f64) -> f64
 // CHECK-DAG: return %[[RESULT]]
 module {
-  func.func private @expm1(f64) -> f64 attributes {libm, llvm.readnone}
-  func.func @call_expm1(%in : f64) -> f64 {
-    %e1 = func.call @expm1(%in) : (f64) -> f64
-    return %e1 : f64
+  emitc.func private @expm1(f64) -> f64 attributes {libm, llvm.readnone, specifiers = ["extern"]}
+  emitc.func @call_expm1(%in : f64) -> f64 {
+    %e1 = emitc.call @expm1(%in) : (f64) -> f64
+    emitc.return %e1 : f64
   }
 }
 
 // -----
 
 // CHECK-NOT: emitc.include <"cmath">
-// CHECK: func.func private @expm1
-// CHECK: func.func @call_expm1(%[[IN:.*]]: f64) -> f64
-// CHECK-NEXT: %[[RESULT:.*]] = call @expm1(%[[IN]]) : (f64) -> f64
+// CHECK: emitc.func private @expm1
+// CHECK: emitc.func @call_expm1(%[[IN:.*]]: f64) -> f64
+// CHECK-NEXT: %[[RESULT:.*]] = emitc.call @expm1(%[[IN]]) : (f64) -> f64
 // CHECK-NEXT: return %[[RESULT]]
 module {
-  func.func private @expm1(f64) -> f64 attributes {llvm.readnone}
-  func.func @call_expm1(%in : f64) -> f64 {
-    %e1 = func.call @expm1(%in) : (f64) -> f64
-    return %e1 : f64
+  emitc.func private @expm1(f64) -> f64 attributes {llvm.readnone}
+  emitc.func @call_expm1(%in : f64) -> f64 {
+    %e1 = emitc.call @expm1(%in) : (f64) -> f64
+    emitc.return %e1 : f64
   }
 }