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

[codegen] more progress

Browse Source
This commit is contained in:
Philippe Tillet
2019-10-03 14:11:50 -04:00
parent 1bf0c8adeb
commit a1e0512703
6 changed files with 63 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
lib/codegen/analysis/liveness.cc
View File

@@ -91,46 +91,36 @@ void liveness::connected_components(node_t x, std::set<node_t> &nodes, graph_t &
bool liveness::do_pad(ir::value *x) {
// alignment for matrix product
// if(auto* dot = dynamic_cast<ir::dot_inst*>(x)) {
// auto order = tiles_->order(x);
// // a
// ir::value *a = dot->get_operand(0);\
// size_t previous_a = pad_[a];
// bool a_trans = dynamic_cast<ir::trans_inst*>(a);
// bool a_row = order[0] == 0;
// if(tiles_->hmma(x) == HMMA_A_ROW)
// pad_[a] = 16;
// else if(tiles_->hmma(x) == HMMA_A_COL)
// pad_[a] = 8;
// else if(a_trans ^ a_row)
// pad_[a] = 4;
// else
// pad_[a] = 0;
// // b
// ir::value *b = dot->get_operand(1);
// size_t previous_b = pad_[b];
// bool b_trans = dynamic_cast<ir::trans_inst*>(b);
// bool b_col = order[0] == 0;
// if(tiles_->hmma(x) == HMMA_B_COL)
// pad_[b] = 16;
// if(tiles_->hmma(x) == HMMA_B_ROW)
// pad_[b] = 8;
// if(b_trans ^ b_col)
// pad_[b] = 4;
// else
// pad_[b] = 0;
// return previous_a != pad_[a] || previous_b != pad_[b];
// }
if(auto* dot = dynamic_cast<ir::dot_inst*>(x)) {
// a
ir::value *a = dot->get_operand(0);\
size_t previous_a = pad_[a];
if(tiles_->hmma(a) == HMMA_A_ROW)
pad_[a] = 16;
else if(tiles_->hmma(a) == HMMA_A_COL)
pad_[a] = 8;
else
pad_[a] = 0;
// b
ir::value *b = dot->get_operand(1);
size_t previous_b = pad_[b];
if(tiles_->hmma(b) == HMMA_B_COL)
pad_[b] = 16;
if(tiles_->hmma(b) == HMMA_B_ROW)
pad_[b] = 8;
else
pad_[b] = 0;
return previous_a != pad_[a] || previous_b != pad_[b];
}
if(auto* cts = dynamic_cast<ir::copy_to_shared_inst*>(x)) {
auto cts_order = tiles_->order(cts);
ir::value *arg = cts->get_operand(0);
auto arg_order = tiles_->order(arg);
size_t previous = pad_[cts];
if(cts_order != arg_order)
pad_[cts] = 4;
return pad_[cts] != previous;
}
// if(auto* tr = dynamic_cast<ir::trans_inst*>(x)) {
// pad_[tr] = 4;
// }
// padding for phi-nodes
if(auto* phi = dynamic_cast<ir::phi_node*>(x)) {
bool has_changed = false;
@@ -142,7 +132,7 @@ bool liveness::do_pad(ir::value *x) {
}
return has_changed;
}
// default -- no pading
// default -- no padding
size_t previous = pad_[x];
pad_[x] = std::max<int>(previous, 0);
return pad_[x] != previous;