[aievec] Fixup issue creating func.func for lut impl. (#2631)

Author: erwei-xilinx

lib/Dialect/AIEVec/Transforms/VectorToAIEVecConversions.cpp

@@ -453,14 +453,14 @@ generateAIEVecOpsForReductionOp(ConversionPatternRewriter &rewriter,
 }
 
 static func::FuncOp getOrInsertFuncDecl(ConversionPatternRewriter &rewriter,
-                                        mlir::ModuleOp parentModuleOp,
+                                        Operation *parentSymbolTableOp,
                                         StringRef funcName, TypeRange inTypes,
                                         TypeRange outTypes) {
 
   mlir::OpBuilder::InsertionGuard insertGuard(rewriter);
   rewriter.setInsertionPointToStart(
-      &parentModuleOp.getRegion().getBlocks().front());
-  SymbolTable st = SymbolTable(parentModuleOp);
+      &parentSymbolTableOp->getRegions().front().getBlocks().front());
+  SymbolTable st = SymbolTable(parentSymbolTableOp);
   func::FuncOp fnOpLookup = st.lookup<func::FuncOp>(funcName);
   func::FuncOp fnOp;
   // if the function is already declared, use the existing function, don't
@@ -473,8 +473,8 @@ static func::FuncOp getOrInsertFuncDecl(ConversionPatternRewriter &rewriter,
     NamedAttribute funcAccess = NamedAttribute(t1, t2);
     FunctionType fnType =
         mlir::FunctionType::get(rewriter.getContext(), inTypes, outTypes);
-    fnOp = rewriter.create<func::FuncOp>(parentModuleOp.getLoc(), funcName,
-                                         fnType, funcAccess);
+    fnOp = rewriter.create<func::FuncOp>(parentSymbolTableOp->getLoc(),
+                                         funcName, fnType, funcAccess);
   }
   return fnOp;
 }
@@ -1981,12 +1981,12 @@ struct ComputeExpOpByLUTLLVMPattern : OpConversionPattern<math::ExpOp> {
 
     auto srcType = dyn_cast<VectorType>(adaptor.getOperand().getType());
     StringRef funcName = "getExpBf16";
-    auto moduleOp = expOp->getParentOfType<mlir::ModuleOp>();
 
     VectorType v16bf16Ty = mlir::VectorType::get({16}, rewriter.getBF16Type());
     VectorType v8i64Ty = mlir::VectorType::get({8}, rewriter.getI64Type());
     func::FuncOp fnOp = getOrInsertFuncDecl(
-        rewriter, moduleOp, funcName, TypeRange{v16bf16Ty}, TypeRange{v8i64Ty});
+        rewriter, expOp->getParentWithTrait<OpTrait::SymbolTable>(), funcName,
+        TypeRange{v16bf16Ty}, TypeRange{v8i64Ty});
 
     SmallVector<Value> expOperands = {adaptor.getOperand()};