106 lines
2.3 KiB
C++
106 lines
2.3 KiB
C++
#ifndef TDL_INCLUDE_IR_FUNCTION_H
|
|
#define TDL_INCLUDE_IR_FUNCTION_H
|
|
|
|
#include <string>
|
|
#include <map>
|
|
#include "value.h"
|
|
#include "constant.h"
|
|
|
|
namespace triton{
|
|
namespace ir{
|
|
|
|
class function;
|
|
class function_type;
|
|
class module;
|
|
class basic_block;
|
|
|
|
/* Argument */
|
|
class argument: public value{
|
|
argument(type *ty, const std::string &name, function *parent, unsigned arg_no);
|
|
|
|
public:
|
|
static argument* create(type *ty, const std::string &name,
|
|
function *parent = nullptr, unsigned arg_no = 0);
|
|
|
|
private:
|
|
function *parent_;
|
|
unsigned arg_no_;
|
|
};
|
|
|
|
/* Attribute */
|
|
enum attribute_kind_t {
|
|
readonly,
|
|
writeonly,
|
|
noalias,
|
|
aligned,
|
|
multiple_of
|
|
};
|
|
|
|
class attribute {
|
|
public:
|
|
attribute(attribute_kind_t kind, unsigned value = 0):
|
|
kind_(kind), value_(value){}
|
|
|
|
bool operator<(const attribute& other) const {
|
|
return std::make_pair(kind_, value_) < std::make_pair(other.kind_, other.value_);
|
|
}
|
|
|
|
const attribute_kind_t get_kind() const {
|
|
return kind_;
|
|
}
|
|
|
|
const unsigned get_value() const {
|
|
return value_;
|
|
}
|
|
|
|
private:
|
|
attribute_kind_t kind_;
|
|
unsigned value_;
|
|
};
|
|
|
|
/* Function */
|
|
class function: public global_object{
|
|
typedef std::vector<argument*> args_t;
|
|
typedef args_t::iterator arg_iterator;
|
|
typedef args_t::const_iterator const_arg_iterator;
|
|
|
|
typedef std::vector<basic_block*> blocks_t;
|
|
typedef blocks_t::iterator block_iterator;
|
|
typedef blocks_t::const_iterator const_block_iterator;
|
|
|
|
typedef std::map<unsigned, std::set<attribute>> attr_map_t;
|
|
|
|
private:
|
|
function(function_type *ty, linkage_types_t linkage,
|
|
const std::string &name = "", module *parent = nullptr);
|
|
|
|
public:
|
|
// accessors
|
|
const args_t &args() { return args_; }
|
|
function_type* get_fn_type() { return fn_ty_; }
|
|
|
|
// factory methods
|
|
static function *create(function_type *ty, linkage_types_t linkage,
|
|
const std::string &name, module *mod);
|
|
// blocks
|
|
const blocks_t &blocks() { return blocks_; }
|
|
void insert_block(basic_block* block, basic_block *next = nullptr);
|
|
|
|
// attributes
|
|
void add_attr(unsigned arg_id, attribute attr) { attrs_[arg_id].insert(attr); }
|
|
const attr_map_t &attrs() { return attrs_; }
|
|
|
|
private:
|
|
module *parent_;
|
|
bool init_;
|
|
function_type *fn_ty_;
|
|
args_t args_;
|
|
blocks_t blocks_;
|
|
attr_map_t attrs_;
|
|
};
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|