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

[codegen] cleaning-up / formalizing shared-memory passes

Browse Source
This commit is contained in:
Philippe Tillet
2019-09-20 16:01:12 -04:00
parent e35be1ddcf
commit 43d88154bd
26 changed files with 229 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/codegen/analysis/layout.cc
View File

@@ -20,10 +20,9 @@ std::set<int> layout::axes_of(ir::value *value) {
rank = ty->get_tile_rank();
// create result
std::set<int> result;
for(size_t d = 0; d < rank; d++){
for(size_t d = 0; d < rank; d++)
if(axes_->has_id(value, d))
result.insert(axes_->get_id(value, d));
}
return result;
}
@@ -54,6 +53,7 @@ const std::vector<ir::value*>& layout::values(unsigned id) const
size_t layout::get_num_groups() const
{ return values_.size(); }
// connect two values
void layout::connect(ir::value *x, ir::value *y) {
if(x == y)
return;
@@ -75,6 +75,7 @@ void layout::connect(ir::value *x, ir::value *y) {
}
}
// make graph
void layout::make_graph(ir::instruction *i) {
for(ir::value* opx: i->ops())
for(ir::value* opy: i->ops()){