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

[codegen] [allocation] fixed issues in HMMA

Browse Source
This commit is contained in:
Philippe Tillet
2019-09-23 17:54:42 -04:00
parent b95ac15d48
commit f0013f8bf1
10 changed files with 94 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/codegen/transform/peephole.cc
View File

@@ -104,12 +104,12 @@ bool peephole::rewrite_dot_hmma(ir::dot_inst *dot, ir::builder& builder, bool tr
BB = ((ir::trans_inst*)B)->get_operand(0);
}
else{
if(auto *T = dynamic_cast<ir::trans_inst*>(A)){
if(auto *T = dynamic_cast<ir::trans_inst*>(B)){
std::vector<ir::constant_int*> perm(T->get_perm());
std::swap(perm[0], perm[1]);
AA = builder.create_trans(T->get_operand(0), perm);
T->replace_all_uses_with(AA);
trans_a = true;
BB = builder.create_trans(T->get_operand(0), perm);
T->replace_all_uses_with(BB);
trans_b = true;
}
}
if(!trans_a && !trans_b)