Commit e634a2dd authored by Nguyen, Thien Minh's avatar Nguyen, Thien Minh
Browse files

Work on Affine for loop 



Signed-off-by: default avatarThien Nguyen <nguyentm@ornl.gov>
parent 0ee10a59

mlir/parsers/qasm3/visitor_handlers/loop_stmt_handler.cpp

+19 −0
Original line number Diff line number Diff line
@@ -5,6 +5,25 @@ using symbol_table_t = exprtk::symbol_table<double>; 
using expression_t = exprtk::expression<double>;

using parser_t = exprtk::parser<double>;



namespace {

/// Creates a single affine "for" loop, iterating from lbs to ubs with

/// the given step.

/// to construct the body of the loop and is passed the induction variable.

void affineLoopBuilder(mlir::ValueRange lbs, mlir::ValueRange ubs, int64_t step,

                       std::function<void(mlir::Value)> bodyBuilderFn,

                       mlir::OpBuilder &builder, mlir::Location &loc) {

  // Create the actual loop

  builder.create<mlir::AffineForOp>(

      loc, lbs, builder.getMultiDimIdentityMap(lbs.size()), ubs,

      builder.getMultiDimIdentityMap(ubs.size()), step, llvm::None,

      [&](mlir::OpBuilder &nestedBuilder, mlir::Location nestedLoc,

          mlir::Value iv, mlir::ValueRange itrArgs) {

        mlir::OpBuilder::InsertionGuard guard(nestedBuilder);

        bodyBuilderFn(iv);

        nestedBuilder.create<mlir::AffineYieldOp>(nestedLoc);

      });

}

} // namespace

namespace qcor {

antlrcpp::Any qasm3_visitor::visitLoopStatement(

    qasm3Parser::LoopStatementContext* context) {

mlir/transforms/quantum_to_llvm.cpp

+3 −0
Original line number Diff line number Diff line
@@ -70,6 +70,9 @@ void QuantumToLLVMLoweringPass::runOnOperation() { 


  // Lower arctan correctly

  patterns.insert<StdAtanOpLowering>(&getContext());

  // Affine to Standard

  populateAffineToStdConversionPatterns(patterns, &getContext());

  populateLoopToStdConversionPatterns(patterns, &getContext());



  // Add Standard to LLVM

  populateStdToLLVMConversionPatterns(typeConverter, patterns);