gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #45 from daadaada/master

Browse Source 
[LANG] Add support for PREFIX_INC, PREFIX_DEC, POSTFIX_INC and POSTFIX_DEC
This commit is contained in:
Philippe Tillet
2020-12-09 02:13:31 -05:00
committed by Philippe Tillet
parent 547a99a5d4 27dc780871
commit d70f54fd6a
3 changed files with 47 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/triton/lang/code_gen.h
View File

@@ -48,6 +48,7 @@ protected:
void set_ret(ir::value* value);
ir::value *GenUnaryMinus(ir::value* arg);
ir::value *GenUnaryInc(UnaryOp* arg, bool is_postfix, bool is_inc);
public:
Generator(Parser* parser) : parser_(parser) {}

2
lib/ir/constant.cc
View File

@@ -43,6 +43,8 @@ constant_int::constant_int(type *ty, uint64_t value)
: constant(ty, 0), value_(value){ }
constant_int *constant_int::get(type *ty, uint64_t value) {
if (!ty->is_integer_ty())
throw std::runtime_error("Cannot create constant_int with non integer ty");
context_impl *impl = ty->get_context().p_impl.get();
constant_int *& cst = impl->int_constants_[std::make_pair(ty, value)];
if(cst == nullptr)

48
lib/lang/code_gen.cc
View File

@@ -182,6 +182,46 @@ ir::value* Generator::GenUnaryMinus(ir::value* arg) {
return bld_->create_sub(_0, arg);
}
ir::value* Generator::GenUnaryInc(UnaryOp* expr, bool is_postfix,
bool is_inc) {
Visit(expr->operand_);
ir::value* arg = ret_;
ir::value *_1 = nullptr;
ir::value *instr = nullptr;
if (arg->get_type()->is_floating_point_ty()) {
_1 = ir::constant_fp::get(arg->get_type(), 1.0);
if (is_inc)
instr = bld_->create_fadd(arg, _1);
else
instr = bld_->create_fsub(arg, _1);
} else if (arg->get_type()->is_integer_ty()) {
_1 = ir::constant_int::get(arg->get_type(), 1);
if (is_inc)
instr = bld_->create_add(arg, _1);
else
instr = bld_->create_sub(arg, _1);
} else if (arg->get_type()->is_pointer_ty()) {
ir::type *ty = ir::type::get_int64_ty(*ctx_);
_1 = ir::constant_int::get(ty, 1);
if (is_inc)
instr = bld_->create_gep(arg, {_1});
else {
ir::value *neg_1 = ir::constant_int::get(ty, -1);
instr = bld_->create_gep(arg, {neg_1});
}
} else
error_not_implemented("data type not supported for unary inc");
mod_->set_value(arg->get_name(), instr);
if (is_postfix)
return arg;
else
return instr;
}
void Generator::VisitUnaryOp(UnaryOp* unary) {
// recursion
Visit(unary->operand_);
@@ -190,10 +230,10 @@ void Generator::VisitUnaryOp(UnaryOp* unary) {
ir::type *arg_scal_ty = arg_ty->get_scalar_ty();
// return
switch (unary->op_) {
case Token::PREFIX_INC: return error_not_implemented("prefix increment not implemented");
case Token::PREFIX_DEC: return error_not_implemented("prefix decrement not implemented");
case Token::POSTFIX_INC: return error_not_implemented("postfix increment not implemented");
case Token::POSTFIX_DEC: return error_not_implemented("postfix decrement not implemented");
case Token::PREFIX_INC: return set_ret(GenUnaryInc(unary, false, true));
case Token::PREFIX_DEC: return set_ret(GenUnaryInc(unary, false, false));
case Token::POSTFIX_INC: return set_ret(GenUnaryInc(unary, true, true));
case Token::POSTFIX_DEC: return set_ret(GenUnaryInc(unary, true, false));
case Token::ADDR: return error_not_implemented("unary & not implemented");
case Token::DEREF: return set_ret(bld_->create_load(arg));
case Token::PLUS: return error_not_implemented("unary + not implemented");