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

Refactor lowering coversion impl. to cpp files for more efficient compilation 



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

mlir/transforms/CMakeLists.txt

+1 −1
Original line number Diff line number Diff line 


set(LIBRARY_NAME quantum-to-llvm-lowering)



file(GLOB SRC *.cpp)

file(GLOB SRC *.cpp lowering/*.cpp)



get_property(dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)

get_property(conversion_libs GLOBAL PROPERTY MLIR_CONVERSION_LIBS)

mlir/transforms/lowering/AssignQubitOpConversion.cpp

0 → 100644
+103 −0
Original line number Diff line number Diff line 
#include "AssignQubitOpConversion.hpp"

#include "mlir/Conversion/AffineToStandard/AffineToStandard.h"

#include "mlir/Conversion/SCFToStandard/SCFToStandard.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVMPass.h"

#include "mlir/Dialect/LLVMIR/LLVMDialect.h"

#include "mlir/Dialect/LLVMIR/LLVMTypes.h"

#include "mlir/Dialect/StandardOps/IR/Ops.h"

#include "mlir/IR/AsmState.h"

#include "mlir/IR/BuiltinOps.h"

#include "mlir/IR/MLIRContext.h"

#include "mlir/IR/Verifier.h"

#include "mlir/InitAllDialects.h"

#include "llvm/ADT/Sequence.h"

#include "llvm/ADT/StringRef.h"

#include "llvm/IR/Module.h"

#include "llvm/Support/CommandLine.h"

#include "llvm/Support/ErrorOr.h"

#include "llvm/Support/MemoryBuffer.h"

#include "llvm/Support/SourceMgr.h"

#include "llvm/Support/TargetSelect.h"

#include "llvm/Support/raw_ostream.h"

#include <iostream>



namespace qcor {

LogicalResult AssignQubitOpConversion::matchAndRewrite(

    Operation *op, ArrayRef<Value> operands,

    ConversionPatternRewriter &rewriter) const {

  // Local Declarations

  ModuleOp parentModule = op->getParentOfType<ModuleOp>();

  auto context = parentModule->getContext();

  auto location = parentModule->getLoc();

  // Unpack destination and source array and indices

  auto dest_array = operands[0];

  auto dest_idx = operands[1];

  auto src_array = operands[2];

  auto src_idx = operands[3];

  FlatSymbolRefAttr array_get_elem_fn_ptr = [&]() {

    static const std::string qir_get_qubit_from_array =

        "__quantum__rt__array_get_element_ptr_1d";

    if (parentModule.lookupSymbol<LLVM::LLVMFuncOp>(qir_get_qubit_from_array)) {

      return SymbolRefAttr::get(qir_get_qubit_from_array, context);

    } else {

      // prototype should be (int64* : qreg, int64 : element) -> int64* :

      // qubit

      auto qubit_array_type =

          LLVM::LLVMPointerType::get(get_quantum_type("Array", context));

      auto qubit_index_type = IntegerType::get(context, 64);

      // ret is i8*

      auto qbit_element_ptr_type =

          LLVM::LLVMPointerType::get(IntegerType::get(context, 8));



      auto get_ptr_qbit_ftype = LLVM::LLVMFunctionType::get(

          qbit_element_ptr_type,

          llvm::ArrayRef<Type>{qubit_array_type, qubit_index_type}, false);



      PatternRewriter::InsertionGuard insertGuard(rewriter);

      rewriter.setInsertionPointToStart(parentModule.getBody());

      rewriter.create<LLVM::LLVMFuncOp>(location, qir_get_qubit_from_array,

                                        get_ptr_qbit_ftype);



      return mlir::SymbolRefAttr::get(qir_get_qubit_from_array, context);

    }

  }();



  // Create the CallOp for the get element ptr 1d function

  auto get_dest_qbit_qir_call = rewriter.create<mlir::CallOp>(

      location, array_get_elem_fn_ptr,

      LLVM::LLVMPointerType::get(IntegerType::get(context, 8)),

      llvm::makeArrayRef(std::vector<mlir::Value>{dest_array, dest_idx}));



  auto get_src_qbit_qir_call = rewriter.create<mlir::CallOp>(

      location, array_get_elem_fn_ptr,

      LLVM::LLVMPointerType::get(IntegerType::get(context, 8)),

      llvm::makeArrayRef(std::vector<mlir::Value>{src_array, src_idx}));



  // Load source qubit

  auto src_bitcast = rewriter.create<LLVM::BitcastOp>(

      location,

      LLVM::LLVMPointerType::get(

          LLVM::LLVMPointerType::get(get_quantum_type("Qubit", context))),

      get_src_qbit_qir_call.getResult(0));



  auto real_casted_src_qubit = rewriter.create<LLVM::LoadOp>(

      location, LLVM::LLVMPointerType::get(get_quantum_type("Qubit", context)),

      src_bitcast.res());



  // Destination: just cast the raw ptr to Qubit** to store the source Qubit*

  // to. Get the destination raw ptr (int8) and cast to Qubit**

  auto dest_bitcast = rewriter.create<LLVM::BitcastOp>(

      location,

      LLVM::LLVMPointerType::get(

          LLVM::LLVMPointerType::get(get_quantum_type("Qubit", context))),

      get_dest_qbit_qir_call.getResult(0));



  // Store source (Qubit*) to destination (Qubit**)

  rewriter.create<LLVM::StoreOp>(location, real_casted_src_qubit, dest_bitcast);

  rewriter.eraseOp(op);

  // std::cout << "After assign:\n";

  // parentModule.dump();

  return success();

}

} // namespace qcor
 
 No newline at end of file

mlir/transforms/lowering/AssignQubitOpConversion.hpp

+1 −101
Original line number Diff line number Diff line 
#pragma once



#include "mlir/Conversion/AffineToStandard/AffineToStandard.h"

#include "mlir/Conversion/SCFToStandard/SCFToStandard.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVMPass.h"

#include "mlir/Dialect/LLVMIR/LLVMDialect.h"

#include "mlir/Dialect/LLVMIR/LLVMTypes.h"

#include "mlir/Dialect/StandardOps/IR/Ops.h"

#include "mlir/IR/AsmState.h"

#include "mlir/IR/BuiltinOps.h"

#include "mlir/IR/MLIRContext.h"

#include "mlir/IR/Verifier.h"

#include "mlir/InitAllDialects.h"

#include "quantum_to_llvm.hpp"

#include "llvm/ADT/Sequence.h"

#include "llvm/ADT/StringRef.h"

#include "llvm/IR/Module.h"

#include "llvm/Support/CommandLine.h"

#include "llvm/Support/ErrorOr.h"

#include "llvm/Support/MemoryBuffer.h"

#include "llvm/Support/SourceMgr.h"

#include "llvm/Support/TargetSelect.h"

#include "llvm/Support/raw_ostream.h"

#include <iostream>



namespace qcor {

class AssignQubitOpConversion : public ConversionPattern {
@@ -39,83 +16,6 @@ public: 


  LogicalResult

  matchAndRewrite(Operation *op, ArrayRef<Value> operands,

                  ConversionPatternRewriter &rewriter) const override {

    // Local Declarations

    ModuleOp parentModule = op->getParentOfType<ModuleOp>();

    auto context = parentModule->getContext();

    auto location = parentModule->getLoc();

    // Unpack destination and source array and indices

    auto dest_array = operands[0];

    auto dest_idx = operands[1];

    auto src_array = operands[2];

    auto src_idx = operands[3];

    FlatSymbolRefAttr array_get_elem_fn_ptr = [&]() {

      static const std::string qir_get_qubit_from_array =

          "__quantum__rt__array_get_element_ptr_1d";

      if (parentModule.lookupSymbol<LLVM::LLVMFuncOp>(

              qir_get_qubit_from_array)) {

        return SymbolRefAttr::get(qir_get_qubit_from_array, context);

      } else {

        // prototype should be (int64* : qreg, int64 : element) -> int64* :

        // qubit

        auto qubit_array_type =

            LLVM::LLVMPointerType::get(get_quantum_type("Array", context));

        auto qubit_index_type = IntegerType::get(context, 64);

        // ret is i8*

        auto qbit_element_ptr_type =

            LLVM::LLVMPointerType::get(IntegerType::get(context, 8));



        auto get_ptr_qbit_ftype = LLVM::LLVMFunctionType::get(

            qbit_element_ptr_type,

            llvm::ArrayRef<Type>{qubit_array_type, qubit_index_type}, false);



        PatternRewriter::InsertionGuard insertGuard(rewriter);

        rewriter.setInsertionPointToStart(parentModule.getBody());

        rewriter.create<LLVM::LLVMFuncOp>(location, qir_get_qubit_from_array,

                                          get_ptr_qbit_ftype);



        return mlir::SymbolRefAttr::get(qir_get_qubit_from_array, context);

      }

    }();



    // Create the CallOp for the get element ptr 1d function

    auto get_dest_qbit_qir_call = rewriter.create<mlir::CallOp>(

        location, array_get_elem_fn_ptr,

        LLVM::LLVMPointerType::get(IntegerType::get(context, 8)),

        llvm::makeArrayRef(std::vector<mlir::Value>{dest_array, dest_idx}));



    auto get_src_qbit_qir_call = rewriter.create<mlir::CallOp>(

        location, array_get_elem_fn_ptr,

        LLVM::LLVMPointerType::get(IntegerType::get(context, 8)),

        llvm::makeArrayRef(std::vector<mlir::Value>{src_array, src_idx}));



    // Load source qubit

    auto src_bitcast = rewriter.create<LLVM::BitcastOp>(

        location,

        LLVM::LLVMPointerType::get(

            LLVM::LLVMPointerType::get(get_quantum_type("Qubit", context))),

        get_src_qbit_qir_call.getResult(0));



    auto real_casted_src_qubit = rewriter.create<LLVM::LoadOp>(

        location,

        LLVM::LLVMPointerType::get(get_quantum_type("Qubit", context)),

        src_bitcast.res());



    // Destination: just cast the raw ptr to Qubit** to store the source Qubit*

    // to. Get the destination raw ptr (int8) and cast to Qubit**

    auto dest_bitcast = rewriter.create<LLVM::BitcastOp>(

        location,

        LLVM::LLVMPointerType::get(

            LLVM::LLVMPointerType::get(get_quantum_type("Qubit", context))),

        get_dest_qbit_qir_call.getResult(0));



    // Store source (Qubit*) to destination (Qubit**)

    rewriter.create<LLVM::StoreOp>(location, real_casted_src_qubit,

                                   dest_bitcast);

    rewriter.eraseOp(op);

    // std::cout << "After assign:\n";

    // parentModule.dump();

    return success();

  }

                  ConversionPatternRewriter &rewriter) const override;

};

} // namespace qcor
 
 No newline at end of file

mlir/transforms/lowering/CreateStringLiteralOpLowering.cpp

0 → 100644
+78 −0
Original line number Diff line number Diff line 
#include "CreateStringLiteralOpLowering.hpp"

#include "mlir/Conversion/AffineToStandard/AffineToStandard.h"

#include "mlir/Conversion/SCFToStandard/SCFToStandard.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVMPass.h"

#include "mlir/Dialect/LLVMIR/LLVMDialect.h"

#include "mlir/Dialect/LLVMIR/LLVMTypes.h"

#include "mlir/Dialect/StandardOps/IR/Ops.h"

#include "mlir/IR/AsmState.h"

#include "mlir/IR/BuiltinOps.h"

#include "mlir/IR/MLIRContext.h"

#include "mlir/IR/Verifier.h"

#include "mlir/InitAllDialects.h"

#include "llvm/ADT/Sequence.h"

#include "llvm/ADT/StringRef.h"

#include "llvm/IR/Module.h"

#include "llvm/Support/CommandLine.h"

#include "llvm/Support/ErrorOr.h"

#include "llvm/Support/MemoryBuffer.h"

#include "llvm/Support/SourceMgr.h"

#include "llvm/Support/TargetSelect.h"

#include "llvm/Support/raw_ostream.h"

#include <iostream>

namespace qcor {

/// Return a value representing an access into a global string with the given

/// name, creating the string if necessary.

Value CreateStringLiteralOpLowering::getOrCreateGlobalString(Location loc,

                                                             OpBuilder &builder,

                                                             StringRef name,

                                                             StringRef value,

                                                             ModuleOp module) {

  // Create the global at the entry of the module.

  LLVM::GlobalOp global;

  if (!(global = module.lookupSymbol<LLVM::GlobalOp>(name))) {

    OpBuilder::InsertionGuard insertGuard(builder);

    builder.setInsertionPointToStart(module.getBody());

    auto type = LLVM::LLVMArrayType::get(

        IntegerType::get(builder.getContext(), 8), value.size());

    global = builder.create<LLVM::GlobalOp>(loc, type, /*isConstant=*/true,

                                            LLVM::Linkage::Internal, name,

                                            builder.getStringAttr(value));

  }



  // Get the pointer to the first character in the global string.

  Value globalPtr = builder.create<LLVM::AddressOfOp>(loc, global);

  Value cst0 = builder.create<LLVM::ConstantOp>(

      loc, IntegerType::get(builder.getContext(), 64),

      builder.getIntegerAttr(builder.getIndexType(), 0));

  return builder.create<LLVM::GEPOp>(

      loc,

      LLVM::LLVMPointerType::get(IntegerType::get(builder.getContext(), 8)),

      globalPtr, ArrayRef<Value>({cst0, cst0}));

}



// Match any Operation that is the QallocOp

LogicalResult CreateStringLiteralOpLowering::matchAndRewrite(

    Operation *op, ArrayRef<Value> operands,

    ConversionPatternRewriter &rewriter) const {

  // Local Declarations, get location, parentModule

  // and the context

  auto loc = op->getLoc();

  ModuleOp parentModule = op->getParentOfType<ModuleOp>();

  auto slOp = cast<mlir::quantum::CreateStringLiteralOp>(op);

  auto slOpText = slOp.text();

  auto slVarName = slOp.varname();



  Value new_global_str = getOrCreateGlobalString(

      loc, rewriter, slVarName,

      StringRef(slOpText.str().c_str(), slOpText.str().length() + 1),

      parentModule);



  variables.insert({slVarName.str(), new_global_str});



  rewriter.eraseOp(op);



  return success();

}

} // namespace qcor
 
 No newline at end of file

mlir/transforms/lowering/CreateStringLiteralOpLowering.hpp

+3 −66
Original line number Diff line number Diff line 
#pragma once



#include "mlir/Conversion/AffineToStandard/AffineToStandard.h"

#include "mlir/Conversion/SCFToStandard/SCFToStandard.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h"

#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVMPass.h"

#include "mlir/Dialect/LLVMIR/LLVMDialect.h"

#include "mlir/Dialect/LLVMIR/LLVMTypes.h"

#include "mlir/Dialect/StandardOps/IR/Ops.h"

#include "mlir/IR/AsmState.h"

#include "mlir/IR/BuiltinOps.h"

#include "mlir/IR/MLIRContext.h"

#include "mlir/IR/Verifier.h"

#include "mlir/InitAllDialects.h"

#include "quantum_to_llvm.hpp"

#include "llvm/ADT/Sequence.h"

#include "llvm/ADT/StringRef.h"

#include "llvm/IR/Module.h"

#include "llvm/Support/CommandLine.h"

#include "llvm/Support/ErrorOr.h"

#include "llvm/Support/MemoryBuffer.h"

#include "llvm/Support/SourceMgr.h"

#include "llvm/Support/TargetSelect.h"

#include "llvm/Support/raw_ostream.h"

#include <iostream>



namespace qcor {

class CreateStringLiteralOpLowering : public ConversionPattern {

private:
@@ -32,29 +10,7 @@ private: 
  /// name, creating the string if necessary.

  static Value getOrCreateGlobalString(Location loc, OpBuilder &builder,

                                       StringRef name, StringRef value,

                                       ModuleOp module) {

    // Create the global at the entry of the module.

    LLVM::GlobalOp global;

    if (!(global = module.lookupSymbol<LLVM::GlobalOp>(name))) {

      OpBuilder::InsertionGuard insertGuard(builder);

      builder.setInsertionPointToStart(module.getBody());

      auto type = LLVM::LLVMArrayType::get(

          IntegerType::get(builder.getContext(), 8), value.size());

      global = builder.create<LLVM::GlobalOp>(loc, type, /*isConstant=*/true,

                                              LLVM::Linkage::Internal, name,

                                              builder.getStringAttr(value));

    }



    // Get the pointer to the first character in the global string.

    Value globalPtr = builder.create<LLVM::AddressOfOp>(loc, global);

    Value cst0 = builder.create<LLVM::ConstantOp>(

        loc, IntegerType::get(builder.getContext(), 64),

        builder.getIntegerAttr(builder.getIndexType(), 0));

    return builder.create<LLVM::GEPOp>(

        loc,

        LLVM::LLVMPointerType::get(IntegerType::get(builder.getContext(), 8)),

        globalPtr, ArrayRef<Value>({cst0, cst0}));

  }

                                       ModuleOp module);



public:

  // Constructor, store seen variables
@@ -68,25 +24,6 @@ public: 
  // Match any Operation that is the QallocOp

  LogicalResult

  matchAndRewrite(Operation *op, ArrayRef<Value> operands,

                  ConversionPatternRewriter &rewriter) const override {

    // Local Declarations, get location, parentModule

    // and the context

    auto loc = op->getLoc();

    ModuleOp parentModule = op->getParentOfType<ModuleOp>();

    auto slOp = cast<mlir::quantum::CreateStringLiteralOp>(op);

    auto slOpText = slOp.text();

    auto slVarName = slOp.varname();



    Value new_global_str = getOrCreateGlobalString(

        loc, rewriter, slVarName,

        StringRef(slOpText.str().c_str(), slOpText.str().length() + 1),

        parentModule);



    variables.insert({slVarName.str(), new_global_str});



    rewriter.eraseOp(op);



    return success();

  }

                  ConversionPatternRewriter &rewriter) const override;

};

} // namespace qcor
 
 No newline at end of file