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

[ir] deleted mask/merge instructions; will be replaced by masked_load/store and select

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-25 15:06:15 -07:00
parent 6ce82dfcdb
commit 2a377bc8b1
27 changed files with 387 additions and 407 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
lib/lang/statement.cpp
View File

@@ -29,21 +29,35 @@ ir::value* compound_statement::codegen(ir::module* mod) const{
/* Expression statement */
ir::value* expression_statement::codegen(ir::module *mod) const{
ir::builder &builder = mod->get_builder();
ir::value *expr = expr_->codegen(mod);
if(pred_ == nullptr)
return expr;
ir::value *pred = pred_->codegen(mod);
if(auto *x = dynamic_cast<ir::load_inst*>(expr))
x->set_mask(pred);
else if(auto *x = dynamic_cast<ir::store_inst*>(expr))
x->set_mask(pred);
else
expr = builder.create_select(pred, expr, ir::undef_value::get(expr->get_type()));
// get name if applicable
std::string name = "";
ir::value *current = nullptr;
if(assignment_expression *assignment = dynamic_cast<assignment_expression*>(expr_))
if(auto *named = dynamic_cast<named_expression*>(assignment)){
std::string name = named->lvalue()->id()->name();
mod->set_value(name, expr);
if(const named_expression* named = dynamic_cast<const named_expression*>(assignment->lvalue())){
name = named->id()->name();
current = mod->get_value(name);
}
// lower expression
ir::value *expr = expr_->codegen(mod);
// modify expression if predicated
if(pred_) {
ir::value *pred = pred_->codegen(mod);
if(!current)
current = ir::undef_value::get(expr->get_type());
if(auto *x = dynamic_cast<ir::load_inst*>(expr)){
x->erase_from_parent();
expr = builder.create_masked_load(x->get_pointer_operand(), pred, current);
}
else if(auto *x = dynamic_cast<ir::store_inst*>(expr)){
x->erase_from_parent();
expr =builder.create_masked_store(x->get_pointer_operand(), x->get_value_operand(), pred);
}
else
expr = builder.create_select(pred, expr, current);
}
// update symbols table
if(!name.empty())
mod->set_value(name, expr);
return expr;
}