github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[ROCM] Use llvm::dbgs() and llvm::errs() streams instead of std::cout.

Browse Source
This commit is contained in:
dfukalov
2023-04-14 16:56:11 +02:00
parent 07d6540507
commit d76b343aac
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lib/Target/HSACO/HSACOTranslation.cpp
View File

@@ -68,7 +68,7 @@ initialize_module(llvm::Module *module, const std::string &triple,
auto target =
llvm::TargetRegistry::lookupTarget(module->getTargetTriple(), error);
if (target == nullptr) {
std::cout << "LookupTarget fail: " << error << std::endl;
llvm::errs() << "LookupTarget fail: " << error << '\n';
return nullptr;
}
llvm::TargetOptions opt;
@@ -108,7 +108,7 @@ std::string generate_amdgcn_assembly(llvm::Module *module,
std::string amdgcn(buffer.begin(), buffer.end());
if (::triton::tools::getBoolEnv("AMDGCN_ENABLE_DUMP")) {
std::cout << "// -----// AMDGCN Dump //----- //\n" << amdgcn << std::endl;
llvm::dbgs() << "// -----// AMDGCN Dump //----- //\n" << amdgcn << '\n';
}
return amdgcn;
@@ -131,8 +131,9 @@ std::string generate_hsaco(llvm::Module *module, const std::string &triple,
std::unique_ptr<llvm::raw_fd_ostream> isabin_fs(
new llvm::raw_fd_ostream(isabin_path, ec, llvm::sys::fs::OF_Text));
if (ec) {
std::cout << isabin_path << " was not created. error code: " << ec
<< std::endl;
llvm::errs() << isabin_path
<< " was not created. error code: " << ec.category().name()
<< ':' << ec.value() << '\n';
}
// emit
llvm::legacy::PassManager pass;
@@ -151,9 +152,9 @@ std::string generate_hsaco(llvm::Module *module, const std::string &triple,
"-shared", "-o", hsaco_path, isabin_path},
std::nullopt, {}, 0, 0, &error_message);
if (lld_result) {
std::cout << "ld.lld execute fail: " << std::endl;
std::cout << error_message << std::endl;
std::cout << lld_result << std::endl;
llvm::errs() << "ld.lld execute fail: " << '\n'
<< error_message << "Code: "
<< lld_result << '\n';
}
return hsaco_path;

2
lib/Target/LLVMIR/LLVMIRTranslation.cpp
View File

@@ -337,7 +337,7 @@ translateTritonGPUToLLVMIR(llvm::LLVMContext *llvmContext,
std::unique_ptr<llvm::raw_string_ostream> ir_ss(
new llvm::raw_string_ostream(mod_string));
llvmIR->print(*ir_ss, nullptr);
std::cout << "// -----// LLVM IR Dump //----- //\n" << mod_string << std::endl;
llvm::dbgs() << "// -----// LLVM IR Dump //----- //\n" << mod_string << "\n";
}
return llvmIR;