tritonToLLVM.hpp

Go to the documentation of this file.

 
/*
**  Copyright (C) - Triton
**
**  This program is under the terms of the Apache License 2.0.
*/
 
#ifndef TRITON_TRITONTOLLVM_HPP
#define TRITON_TRITONTOLLVM_HPP
 
#include <map>
#include <memory>
#include <ostream>
#include <string>
#include <unordered_map>
 
#include <triton/ast.hpp>
#include <triton/dllexport.hpp>
 
#include <llvm/IR/IRBuilder.h>
#include <llvm/IR/LLVMContext.h>
#include <llvm/IR/LegacyPassManager.h>
#include <llvm/IR/Module.h>
#include <llvm/Transforms/IPO/PassManagerBuilder.h>
 
 
 
namespace triton {
  namespace ast {
 
    class TritonToLLVM {
      private:
        llvm::LLVMContext& llvmContext;
 
        std::shared_ptr<llvm::Module> llvmModule;
 
        llvm::IRBuilder<> llvmIR;
 
        std::map<triton::ast::SharedAbstractNode, llvm::Value*> llvmVars;
 
        void createFunction(const triton::ast::SharedAbstractNode& node, const char* fname);
 
        llvm::Value* do_convert(const triton::ast::SharedAbstractNode& node, std::unordered_map<triton::ast::SharedAbstractNode, llvm::Value*>* results);
 
      public:
        TRITON_EXPORT TritonToLLVM(llvm::LLVMContext& llvmContext);
 
        TRITON_EXPORT std::shared_ptr<llvm::Module> convert(const triton::ast::SharedAbstractNode& node, const char* fname="__triton", bool optimize=false);
    };
 
  };
};
 
#endif /* TRITON_TRITONTOLLVM_HPP */