Commit 5781d56a authored by Mccaskey, Alex's avatar Mccaskey, Alex
Browse files

started on alias handling for qasm3 mlir compiler 



Signed-off-by: Mccaskey, Alex's avatarAlex McCaskey <mccaskeyaj@ornl.gov>
parent 83b19b0c

mlir/parsers/qasm3/CMakeLists.txt

+1 −0
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ file(GLOB SRC *.cpp antlr/generated/*.cpp utils/*.cpp 
   visitor_handlers/loop_stmt_handler.cpp

   visitor_handlers/conditional_handler.cpp

   visitor_handlers/subroutine_handler.cpp

   visitor_handlers/alias_handler.cpp

   )



add_library(${LIBRARY_NAME} SHARED ${SRC})

mlir/parsers/qasm3/qasm3_visitor.hpp

+4 −0
Original line number Diff line number Diff line
@@ -101,6 +101,10 @@ class qasm3_visitor : public qasm3::qasm3BaseVisitor { 
  antlrcpp::Any visitClassicalAssignment(

      qasm3Parser::ClassicalAssignmentContext* context) override;



  // Visit nodes related to aliasing, the 'let' keyword

  // see visitor_handlers/alias_handler.cpp

    antlrcpp::Any visitAliasStatement(

      qasm3Parser::AliasStatementContext* context) override;

 protected:

  // Reference to the MLIR OpBuilder and ModuleOp

  // this MLIRGen task

mlir/parsers/qasm3/visitor_handlers/alias_handler.cpp

0 → 100644
+72 −0
Original line number Diff line number Diff line 
#include "expression_handler.hpp"

#include "qasm3_visitor.hpp"



namespace qcor {

antlrcpp::Any qasm3_visitor::visitAliasStatement(

    qasm3Parser::AliasStatementContext* context) {

  auto location = get_location(builder, file_name, context);



  /** Aliasing **/

  // aliasStatement

  //     : 'let' Identifier EQUALS indexIdentifier SEMICOLON

  //     ;



  // /** Register Concatenation and Slicing **/



  // indexIdentifier

  //     : Identifier rangeDefinition

  //     | Identifier ( LBRACKET expressionList RBRACKET )?

  //     | indexIdentifier '||' indexIdentifier

  //     ;



  printErrorMessage(

      "Alias not yet supported yet - Thien, remove this for development.");



  // The name of the new alias register, pointing to previously allocated

  // register

  auto alias = context->Identifier()->getText();



  // Get the name and symbol Value of the original register

  auto allocated_variable = context->indexIdentifier()->Identifier()->getText();

  auto allocated_symbol = symbol_table.get_symbol(allocated_variable);



  // handle q[1, 3, 5] comma syntax

  if (context->indexIdentifier()->LBRACKET()) {

    // get the comma expression, count how many elements there are

    auto expressions =

        context->indexIdentifier()->expressionList()->expression();

    auto n_expressions = expressions.size();



    // Allocate a new array of qubits of given size

    auto str_attr = builder.getStringAttr(alias);

    auto integer_attr =

        mlir::IntegerAttr::get(builder.getI64Type(), n_expressions);

    mlir::Value allocation = builder.create<mlir::quantum::QallocOp>(

        location, array_type, integer_attr, str_attr);



    auto counter = 0;

    for (auto expr : expressions) {

      // GOAL HERE IS TO ASSIGN extracted qubits from original array

      // to the correct element of the alias array

      auto idx =

          symbol_table.evaluate_constant_integer_expression(expr->getText());



      // get the extracted element from the original register

      auto extracted = builder.create<mlir::quantum::ExtractQubitOp>(

          location, qubit_type, allocated_symbol,

          get_or_create_constant_integer_value(

              idx, location, builder.getI64Type(), symbol_table, builder));



      // use extracted with a new qassign dialect operation.

    }



  } else if (auto range_def = context->indexIdentifier()->rangeDefinition()) {

    // handle range definition

    // I think we can handle RANGE with a memref<3xi64>...



  } else {

    // handle concatenation

  }

  return 0;

}

}  // namespace qcor
 
 No newline at end of file