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

[codegen][selection] more flexible instruction selection for reduce_inst

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-04 16:34:36 -07:00
parent 6be532c6a2
commit d869d9a924
14 changed files with 167 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/ir/builder.cpp
View File

@@ -322,8 +322,8 @@ value *builder::create_sqrt(value *A, const std::string &name) {
return insert(sqrt_inst::create(A, name));
}
value *builder::create_reduce(value *A, const std::string &name) {
return insert(reduce_inst::create(A, name));
value *builder::create_reduce(value *A, unsigned axis, const std::string &name) {
return insert(reduce_inst::create(A, axis, name));
}
value *builder::create_select(value *pred, value *if_value, value *else_value, const std::string &name){