60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
#include "triton/ir/basic_block.h"
|
|
#include "triton/ir/module.h"
|
|
#include "triton/ir/type.h"
|
|
#include "triton/ir/constant.h"
|
|
#include "triton/ir/function.h"
|
|
#include "triton/ir/instructions.h"
|
|
#include "triton/ir/print.h"
|
|
|
|
namespace triton{
|
|
namespace ir{
|
|
|
|
std::string get_name(ir::value *v, unsigned i) {
|
|
if(v->get_name().empty()){
|
|
std::string name = "%" + std::to_string(i);
|
|
v->set_name(name);
|
|
}
|
|
return v->get_name();
|
|
}
|
|
|
|
|
|
void print(module &mod, std::ostream& os) {
|
|
unsigned cnt = 0;
|
|
for(ir::function *fn: mod.get_function_list()){
|
|
os << "{" << std::endl;
|
|
for(ir::basic_block *block: fn->blocks()){
|
|
auto const &predecessors = block->get_predecessors();
|
|
os << block->get_name() << ":";
|
|
if(!predecessors.empty()){
|
|
os << " ";
|
|
os << "; preds = ";
|
|
auto const &predecessors = block->get_predecessors();
|
|
for(ir::basic_block *pred: predecessors)
|
|
os << pred->get_name() << (pred!=predecessors.back()?", ":"");
|
|
}
|
|
os << std::endl;
|
|
for(ir::instruction *inst: block->get_inst_list()){
|
|
os << " ";
|
|
if(ir::value *pred = inst->get_mask_pred())
|
|
os << "@" << get_name(pred, cnt++) << " ";
|
|
if(!inst->get_type()->is_void_ty())
|
|
os << get_name(inst, cnt++) << " = ";
|
|
os << inst->repr();
|
|
ir::instruction::ops_t ops = inst->ops();
|
|
size_t num_ops = inst->get_num_operands();
|
|
if(num_ops > 0)
|
|
os << " ";
|
|
for(unsigned i = 0; i < num_ops; i++)
|
|
os << get_name(ops[i], cnt++) << (i < num_ops - 1?", ":"");
|
|
os << ";" << std::endl;
|
|
}
|
|
os << std::endl;
|
|
}
|
|
os << "}" << std::endl;
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|