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

[examples] multiple transposition schemes now supported

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-24 13:06:20 -07:00
parent 0b1c389894
commit 96b4d5e411
3 changed files with 76 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -112,8 +112,12 @@ bool peephole::rewrite_dot_hmma(ir::dot_inst *dot, ir::builder& builder, bool tr
trans_a = true;
}
}
if(!trans_a && !trans_b)
return false;
ir::instruction *dot_atbt = builder.insert(ir::dot_inst::create(AA, BB, D, trans_a, trans_b));
dot->replace_all_uses_with(dot_atbt);
return true;
}
@@ -186,8 +190,9 @@ bool peephole::rewrite_dot(ir::instruction *value, ir::builder& builder){
if(dot->is_a_trans() || dot->is_b_trans())
return false;
// hmma
if(is_hmma(dot))
if(is_hmma(dot)){
return rewrite_dot_hmma(dot, builder, trans_a, trans_b, A, B, D);
}
else
return rewrite_dot_fp32(dot, builder, trans_a, trans_b, A, B, D);
}