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

[CODEGEN] Removed unnecessary coalescing rematerialization

Browse Source
This commit is contained in:
Philippe Tillet
2020-05-30 15:03:25 -04:00
committed by Philippe Tillet
parent 5995cbff8e
commit a8f1b85c5f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/codegen/transform/coalesce.cc
View File

@@ -116,9 +116,12 @@ void coalesce::run(ir::module &mod) {
// update list of values to rematerialize // update list of values to rematerialize
if(axes.empty()) if(axes.empty())
continue; continue;
for(auto it = ++axes.rbegin(); it != axes.rend(); it++) for(auto it = ++axes.rbegin(); it != axes.rend(); it++){
if(it->second.size() == 1)
continue;
remat.insert(remat.begin(), it->second.begin(), it->second.end()); remat.insert(remat.begin(), it->second.begin(), it->second.end());
} }
}
// rematerialize values // rematerialize values
for(ir::io_inst *r: remat) { for(ir::io_inst *r: remat) {
ir::builder& builder = mod.get_builder(); ir::builder& builder = mod.get_builder();