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

[LANG] Add support for POSTFIX_INC and POSTFIX_DEC, and pointer type

Browse Source
This commit is contained in:
Yan Da
2020-12-09 13:42:21 +08:00
committed by Philippe Tillet
parent 05b95b7fa6
commit e9b2335224
1 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lib/lang/code_gen.cc
View File

@@ -188,27 +188,37 @@ ir::value* Generator::GenUnaryInc(UnaryOp* expr, bool is_postfix,
ir::value* arg = ret_;
ir::value *_1 = nullptr;
if (arg->get_type()->is_half_ty() ||
arg->get_type()->is_float_ty() ||
arg->get_type()->is_double_ty())
_1 = ir::constant_fp::get(arg->get_type(), 1.0);
else if (arg->get_type()->is_integer_ty())
_1 = ir::constant_int::get(arg->get_type(), 1);
else
error_not_implemented("data type not supported for unary inc");
if (is_postfix) {
error_not_implemented("postfix inc not implemented");
return nullptr;
} else {
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);
mod_->set_value(arg->get_name(), instr);
return instr;
} else if (arg->get_type()->is_pointer_ty()) {
_1 = ir::constant_int::get(arg->get_type(), 1);
if (is_inc)
instr = bld_->create_gep(arg, {_1});
else {
ir::value *neg_1 = ir::constant_int::get(arg->get_type(), -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) {