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

bla

Browse Source
This commit is contained in:
Philippe Tillet
2019-02-15 22:03:09 -05:00
parent 5f5959dc6e
commit cf1a583dbf
6 changed files with 36 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/codegen/tune.cpp
View File

@@ -67,17 +67,15 @@ void tune::init_c_graph(ir::instruction *v) {
}
// Element-wise
else if(dynamic_cast<ir::user*>(v)){
std::cout << typeid(*v).name() << std::endl;
for(unsigned i = 0; i < shapes.size(); i ++)
for(ir::value* op: v->ops())
add_constraint({v, i}, {op, i});
}
/* Add mask constraints */
if(ir::value *mask = v->get_mask()){
std::cout << typeid(*mask).name() << " " << typeid(*v->ops()[0]).name() << std::endl;
if(ir::value *pred = v->get_mask().pred){
for(unsigned i = 0; i < shapes.size(); i++)
add_constraint({v->ops()[0], i}, {mask, i});
add_constraint({v->ops()[0], i}, {pred, i});
}
}