67 lines
1.6 KiB
C++
67 lines
1.6 KiB
C++
#include <algorithm>
|
|
#include "triton/ir/function.h"
|
|
#include "triton/ir/type.h"
|
|
#include "triton/ir/module.h"
|
|
|
|
namespace triton{
|
|
namespace ir{
|
|
|
|
|
|
/* Argument */
|
|
|
|
argument::argument(type *ty, const std::string &name, function *parent, unsigned arg_no)
|
|
: value(ty, name), parent_(parent), arg_no_(arg_no) { }
|
|
|
|
argument *argument::create(type *ty, const std::string &name,
|
|
function *parent, unsigned arg_no) {
|
|
return new argument(ty, name, parent, arg_no);
|
|
}
|
|
|
|
function* argument::get_parent() const {
|
|
return parent_;
|
|
}
|
|
|
|
unsigned argument::get_arg_no() const {
|
|
return arg_no_;
|
|
}
|
|
|
|
void argument::accept(visitor *v) {
|
|
v->visit_argument(this);
|
|
}
|
|
|
|
|
|
/* function */
|
|
function::function(function_type *ty, linkage_types_t linkage,
|
|
const std::string &name, module *parent)
|
|
: global_object(ty, 0, linkage, name), parent_(parent), fn_ty_(ty), is_kernel_(false) {
|
|
unsigned num_params = fn_ty_->get_num_params();
|
|
if(parent)
|
|
parent->push_function(this);
|
|
// skip if no parameter
|
|
if(num_params == 0)
|
|
return;
|
|
// create arguments
|
|
args_.resize(num_params);
|
|
for(unsigned i = 0; i < num_params; i++){
|
|
type *param_ty = fn_ty_->get_param_ty(i);
|
|
args_[i] = argument::create(param_ty, "", this, i);
|
|
}
|
|
}
|
|
|
|
/* basic block */
|
|
void function::insert_block(basic_block *block, basic_block *next) {
|
|
auto it = std::find(blocks_.begin(), blocks_.end(), next);
|
|
blocks_.insert(it, block);
|
|
}
|
|
|
|
|
|
function *function::create(function_type *ty, linkage_types_t linkage,
|
|
const std::string &name, module *mod) {
|
|
return new function(ty, linkage, name, mod);
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
|