Commit 5c7fe67d authored by Mccaskey, Alex's avatar Mccaskey, Alex
Browse files

started work on mapping clang ast nodes to xacc ir 



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

compiler/clang/QCorPluginASTAction.cpp

+2 −1
Original line number Diff line number Diff line
@@ -14,7 +14,8 @@ bool QCorPluginASTAction::ParseArgs(const CompilerInstance &ci, 
  if (!xacc::isInitialized()) {

    xacc::Initialize(args);

  }

  for (auto a : args) xacc::info("ARG: " + a);

  for (auto a : args)

    xacc::info("qcor argument: " + a);

  return true;

}

compiler/clang/QcorASTVisitor.hpp

+120 −4
Original line number Diff line number Diff line
@@ -3,24 +3,101 @@ 


#include "clang/AST/AST.h"

#include "clang/AST/ASTConsumer.h"

#include "clang/AST/Expr.h"

#include "clang/AST/RecursiveASTVisitor.h"

#include "clang/AST/Stmt.h"

#include "clang/Frontend/CompilerInstance.h"

#include "clang/Lex/Lexer.h"

#include "clang/Lex/Preprocessor.h"

#include "clang/Rewrite/Core/Rewriter.h"

#include "clang/AST/Stmt.h"

#include "clang/AST/Expr.h"



#include "QuantumKernelHandler.hpp"



#include "clang/Parse/ParseAST.h"



#include "XACC.hpp"

#include "IRProvider.hpp"



using namespace clang;

using namespace xacc;



namespace qcor {

static const SourceLocation nopos;



class CppToXACCIRVisitor : public RecursiveASTVisitor<CppToXACCIRVisitor> {

protected:



   ASTContext& context;



   std::shared_ptr<Function> function;

   std::shared_ptr<IRProvider> provider;



   std::string gateName = "";

   std::vector<int> bits;

   std::vector<InstructionParameter> parameters;



public:

   CppToXACCIRVisitor(ASTContext& c): context(c) {

       provider = xacc::getService<IRProvider>("gate");

       function = provider->createFunction("tmp", {});

   }



   bool VisitCallExpr(CallExpr* expr) {

       if (gateName != "") {

           // create the gate instruction

           if (gateName == "CX") {gateName = "CNOT";}



           auto inst = provider->createInstruction(gateName, bits, parameters);

           function->addInstruction(inst);

       }

       gateName = "";

       bits.clear();

       parameters.clear();

       return true;

   }



   bool VisitDeclRefExpr(DeclRefExpr* expr) {



       if (expr->getType() == context.DependentTy) {

         gateName = expr->getNameInfo().getAsString();



        //  std::cout << "VISITING " << gateName << ", " << expr->getType().getAsString() << "\n";

       } else if (expr->getType() == context.DoubleTy) {

           InstructionParameter p(expr->getNameInfo().getAsString());

           parameters.push_back(p);

       }



       return true;

   }



   bool VisitIntegerLiteral(IntegerLiteral* literal) {

       bits.push_back(literal->getValue().getLimitedValue());

       return true;

   }



   bool VisitFloatingLiteral(FloatingLiteral* literal) {

       InstructionParameter p(literal->getValue().convertToDouble());

       parameters.push_back(p);

       return true;

   }



   std::shared_ptr<Function> getFunction() {

       // add the last one

        if (gateName != "") {

           // create the gate instruction

           if (gateName == "CX") {gateName = "CNOT";}



           auto inst = provider->createInstruction(gateName, bits, parameters);

           function->addInstruction(inst);

       }

       return function;

   }

};



class QcorASTVisitor : public RecursiveASTVisitor<QcorASTVisitor> {



public:



  QcorASTVisitor(CompilerInstance &c) : ci(c) {}



  bool VisitLambdaExpr(LambdaExpr *LE) {
@@ -29,13 +106,34 @@ public: 
    lo.CPlusPlus11 = 1;

    auto xaccKernelLambdaStr =

        Lexer::getSourceText(CharSourceRange(LE->getSourceRange(), true), SM,

                             lo).str();

                             lo)

            .str();



    std::cout << "Check it out, I got the Lambda as a source string :)\n";

    xacc::info(xaccKernelLambdaStr);



    CppToXACCIRVisitor visitor(ci.getASTContext());



    //  LE->getType().dump();



    // create empty statement, does nothing

    // Stmt *tmp = (Stmt *)new (ci.getASTContext()) NullStmt(nopos);

    // std::vector<Stmt *> stmts;

    // stmts.push_back(tmp);

    // replace LE's compound statement with that statement

    // LE->getBody()->setLastStmt(ReturnStmt::CreateEmpty(

    //     ci.getASTContext(),

    //     false));



    Stmt* q_kernel_body = LE->getBody();

    q_kernel_body->dump();



    std::cout << "TRAVERSING STMT:\n";

    visitor.TraverseStmt(LE->getBody());

    auto function = visitor.getFunction();

    std::cout << "XACC IR Qasm:\n" << function->toString() << "\n";





    return true;

  }
@@ -45,3 +143,21 @@ private: 


} // namespace qcor

#endif



    // const auto& parents = ci.getASTContext().getParents(old_stmt);

    // auto it = parents.begin();

    // if (it == parents.end()) std::cout << "NO PARENTS:\n";

    // else {

    //     std::cout << "this has " << parents.size() << " parents\n";



    //     const Stmt* parent = parents[0].get<Stmt>();

    //     parent->dump();



    //     Stmt& new_stmt = *ReturnStmt::CreateEmpty(

    //        ci.getASTContext(),

    //        false);



    //     // Stmt::const_child_iterator it = std::find(parent->child_begin(), parent->child_end(), old_stmt);

    //     // *it = new_stmt;

    //     // std::replace(parent->child_begin(), parent->child_end(), old_stmt, new_stmt);

    // }
 
 No newline at end of file

compiler/clang/QuantumKernelHandler.hpp

+17 −5
Original line number Diff line number Diff line
@@ -20,6 +20,7 @@ public: 
  QuantumKernelHandler(ASTContext &context) : m_Context(context) {

    auto irProvider = xacc::getService<xacc::IRProvider>("gate");

    validInstructions = irProvider->getInstructions();

    validInstructions.push_back("CX");

  }



  bool LookupUnqualified(clang::LookupResult &R, clang::Scope *S) override {
@@ -34,12 +35,23 @@ public: 


      IdentifierInfo *II = Name.getAsIdentifierInfo();

      SourceLocation Loc = R.getNameLoc();

      VarDecl *Result =

          VarDecl::Create(m_Context, R.getSema().getFunctionLevelDeclContext(),

                          Loc, Loc, II, m_Context.DependentTy, 0, SC_None);

    //   VarDecl *Result =

    //       VarDecl::Create(m_Context, R.getSema().getFunctionLevelDeclContext(),

    //                       Loc, Loc, II, m_Context.DependentTy, 0, SC_None);

      auto fdecl = FunctionDecl::Create(

          m_Context, R.getSema().getFunctionLevelDeclContext(), Loc, Loc, Name,

          m_Context.DependentTy, 0, SC_None);

    //   CompoundStmt *CS =

    //       new (m_Context) clang::CompoundStmt(clang::SourceLocation());



      if (Result) {

        R.addDecl(Result);

      Stmt *S = new (m_Context) NullStmt(Stmt::EmptyShell()); //::CreateEmpty(m_Context, false);



      std::cout << "HAS BODY: " << fdecl->hasBody() << "\n";

      fdecl->setBody(S);

      std::cout << "HAS BODY: " << fdecl->hasBody() << "\n";



      if (fdecl) {

        R.addDecl(fdecl);

        return true;

      } else {

        return false;