f81012a8cf
* [CODEGEN] Fixed atomic_add issue * [CODEGEN] Fixed liveness analysis bug for instructions that are not DCE'd but have no users (e.g., atomic_cas)
60 lines
1.4 KiB
C++
60 lines
1.4 KiB
C++
#include <climits>
|
|
#include <iostream>
|
|
#include "triton/codegen/analysis/liveness.h"
|
|
#include "triton/codegen/analysis/layout.h"
|
|
#include "triton/ir/function.h"
|
|
#include "triton/ir/module.h"
|
|
#include "triton/ir/utils.h"
|
|
|
|
namespace triton{
|
|
namespace codegen{
|
|
namespace analysis{
|
|
|
|
|
|
void liveness::run(ir::module &mod) {
|
|
intervals_.clear();
|
|
|
|
// Assigns index to each instruction
|
|
std::map<ir::value*, slot_index> indices;
|
|
for(ir::function *fn: mod.get_function_list()){
|
|
slot_index index = 0;
|
|
for(ir::basic_block *block: fn->blocks())
|
|
for(ir::instruction *instr: block->get_inst_list()){
|
|
index += 1;
|
|
indices.insert({instr, index});
|
|
}
|
|
}
|
|
|
|
// create live intervals
|
|
for(auto &x: layouts_->get_all()) {
|
|
shared_layout* layout = x.second->to_shared();
|
|
if(!layout)
|
|
continue;
|
|
// users
|
|
std::set<ir::user*> users;
|
|
for(ir::value *v: layout->get_values()){
|
|
for(ir::user *u: v->get_users())
|
|
users.insert(u);
|
|
}
|
|
// compute intervals
|
|
unsigned start = INT32_MAX;
|
|
for(ir::value *v: layout->get_values())
|
|
if(indices.find(v) != indices.end())
|
|
start = std::min(start, indices.at(v));
|
|
unsigned end = 0;
|
|
for(ir::user *u: users)
|
|
if(indices.find(u) != indices.end())
|
|
end = std::max(end, indices.at(u));
|
|
if(end == 0)
|
|
end = start + 1;
|
|
intervals_[layout] = segment{start, end};
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|