1 //===- MLIRGen.h - MLIR Generation from a Toy AST -------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file declares a simple interface to perform IR generation targeting MLIR 10 // from a Module AST for the Toy language. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef TOY_MLIRGEN_H 15 #define TOY_MLIRGEN_H 16 17 #include <memory> 18 19 namespace mlir { 20 class MLIRContext; 21 template <typename OpTy> 22 class OwningOpRef; 23 class ModuleOp; 24 } // namespace mlir 25 26 namespace toy { 27 class ModuleAST; 28 29 /// Emit IR for the given Toy moduleAST, returns a newly created MLIR module 30 /// or nullptr on failure. 31 mlir::OwningOpRef<mlir::ModuleOp> mlirGen(mlir::MLIRContext &context, 32 ModuleAST &moduleAST); 33 } // namespace toy 34 35 #endif // TOY_MLIRGEN_H 36