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

add print_llvm_module

Browse Source
This commit is contained in:
Michael Melesse
2022-10-28 20:07:35 +00:00
parent 531ef18cb6
commit 7fce2bc5f1
1 changed files with 39 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
include/print_helper.h
View File

@@ -57,24 +57,52 @@ inline void print_vector(std::vector<T> &vec, std::string name = "")
} }
// dump llvm ir to tmp file // dump llvm ir to tmp file
inline void write_llvm_ir(llvm::Module &llvm_module, std::string filename = "", bool tracked = false) inline std::string print_llvm_module(llvm::Module *llvm_module, bool print_to_cout = true)
{
std::cout << "\t" << "print_llvm_module" << std::endl;
// get module as a string
std::error_code ec;
std::string mod_string;
std::unique_ptr<llvm::raw_string_ostream> ir_ss(
new llvm::raw_string_ostream(mod_string));
llvm_module->print(*ir_ss, nullptr);
// print module
if (print_to_cout)
{
if (!mod_string.empty())
std::cout << "\t" << mod_string << std::endl;
else
std::cout << "\t" << llvm_module->getModuleIdentifier() << ": "
<< "is empty" << std::endl;
}
return mod_string;
}
// dump llvm ir to tmp file
inline void write_llvm_ir(llvm::Module *llvm_module, std::string filename = "", bool tracked = false)
{ {
if (filename.empty()) // get module string
filename = llvm_module.getModuleIdentifier(); std::string module_string = print_llvm_module(llvm_module, false);
// get file name and path
if (filename.empty())
filename = llvm_module->getModuleIdentifier();
std::string count_str = ""; std::string count_str = "";
if (tracked) if (tracked)
{ {
count_str = "_" + std::to_string(print_count); count_str = "_" + std::to_string(print_count);
} }
std::string ir_path = std::string("/tmp/") + filename + count_str + std::string(".ll"); std::string ir_path = std::string("/tmp/") + filename + count_str + std::string(".ll");
std::error_code ec;
std::unique_ptr<llvm::raw_fd_ostream> ir_fs( // write file
new llvm::raw_fd_ostream(ir_path, ec)); std::ofstream output_file(ir_path);
llvm_module.print(*ir_fs, nullptr); output_file << module_string;
ir_fs->flush(); output_file.close();
// increament counter
if (tracked) if (tracked)
{ {
print_count += 1; print_count += 1;
@@ -110,7 +138,8 @@ inline void print_llvm_value(llvm::Value *llvm_value, std::string name = "")
if (llvm_value) if (llvm_value)
std::cout << "\t" << name << ": " << get_llvm_value_as_str(llvm_value) << std::endl; std::cout << "\t" << name << ": " << get_llvm_value_as_str(llvm_value) << std::endl;
else else
std::cout << "\t" << name << ": " << "is nullptr" << std::endl; std::cout << "\t" << name << ": "
<< "is nullptr" << std::endl;
} }
inline void print_llvm_type(llvm::Type *llvm_type, std::string name = "") inline void print_llvm_type(llvm::Type *llvm_type, std::string name = "")