539 lines
17 KiB
C++
539 lines
17 KiB
C++
|
/*
|
||
|
|
Copyright (c) 2022 - Present Advanced Micro Devices, Inc. All rights reserved.
|
||
|
|
|
||
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
|
of this software and associated documentation files (the "Software"), to deal
|
||
|
|
in the Software without restriction, including without limitation the rights
|
||
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
|
|
copies of the Software, and to permit persons to whom the Software is
|
||
|
|
furnished to do so, subject to the following conditions:
|
||
|
|
|
||
|
|
The above copyright notice and this permission notice shall be included in
|
||
|
|
all copies or substantial portions of the Software.
|
||
|
|
|
||
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||
|
|
THE SOFTWARE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "hiprtcComgrHelper.hpp"
|
||
|
|
|
||
|
|
namespace hiprtc {
|
||
|
|
|
||
|
|
namespace helpers {
|
||
|
|
bool addCodeObjData(amd_comgr_data_set_t& input, const std::vector<char>& source,
|
||
|
|
const std::string& name, const amd_comgr_data_kind_t type) {
|
||
|
|
amd_comgr_data_t data;
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::create_data(type, &data); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::set_data(data, source.size(), source.data());
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::release_data(data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::set_data_name(data, name.c_str()); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::release_data(data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::data_set_add(input, data); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::release_data(data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
amd::Comgr::release_data(data); // Release from our end after setting the input
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool extractBuildLog(amd_comgr_data_set_t dataSet, std::string& buildLog) {
|
||
|
|
size_t count;
|
||
|
|
if (auto res = amd::Comgr::action_data_count(dataSet, AMD_COMGR_DATA_KIND_LOG, &count);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<char> log;
|
||
|
|
if (count > 0) {
|
||
|
|
if (!extractByteCodeBinary(dataSet, AMD_COMGR_DATA_KIND_LOG, log)) return false;
|
||
|
|
buildLog.insert(buildLog.end(), log.data(), log.data() + log.size());
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool extractByteCodeBinary(const amd_comgr_data_set_t inDataSet,
|
||
|
|
const amd_comgr_data_kind_t dataKind, std::vector<char>& bin) {
|
||
|
|
amd_comgr_data_t binaryData;
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::action_data_get_data(inDataSet, dataKind, 0, &binaryData);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t binarySize = 0;
|
||
|
|
if (auto res = amd::Comgr::get_data(binaryData, &binarySize, NULL);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::release_data(binaryData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t bufSize = (dataKind == AMD_COMGR_DATA_KIND_LOG) ? binarySize + 1 : binarySize;
|
||
|
|
|
||
|
|
char* binary = new char[bufSize];
|
||
|
|
if (binary == nullptr) {
|
||
|
|
amd::Comgr::release_data(binaryData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::get_data(binaryData, &binarySize, binary);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
delete[] binary;
|
||
|
|
amd::Comgr::release_data(binaryData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dataKind == AMD_COMGR_DATA_KIND_LOG) {
|
||
|
|
binary[binarySize] = '\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
amd::Comgr::release_data(binaryData);
|
||
|
|
|
||
|
|
bin.reserve(binarySize);
|
||
|
|
bin.assign(binary, binary + binarySize);
|
||
|
|
delete[] binary;
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool createAction(amd_comgr_action_info_t& action, std::vector<std::string>& options,
|
||
|
|
const std::string& isa, const amd_comgr_language_t lang) {
|
||
|
|
if (auto res = amd::Comgr::create_action_info(&action); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (lang != AMD_COMGR_LANGUAGE_NONE) {
|
||
|
|
if (auto res = amd::Comgr::action_info_set_language(action, lang);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::action_info_set_isa_name(action, isa.c_str());
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<const char*> optionsArgv;
|
||
|
|
optionsArgv.reserve(options.size());
|
||
|
|
for (auto& option : options) {
|
||
|
|
optionsArgv.push_back(option.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res =
|
||
|
|
amd::Comgr::action_info_set_option_list(action, optionsArgv.data(), optionsArgv.size());
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return res;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::action_info_set_logging(action, true);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return res;
|
||
|
|
}
|
||
|
|
|
||
|
|
return AMD_COMGR_STATUS_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool compileToBitCode(const amd_comgr_data_set_t compileInputs, const std::string& isa,
|
||
|
|
std::vector<std::string>& compileOptions, std::string& buildLog,
|
||
|
|
std::vector<char>& LLVMBitcode) {
|
||
|
|
amd_comgr_language_t lang = AMD_COMGR_LANGUAGE_HIP;
|
||
|
|
amd_comgr_action_info_t action;
|
||
|
|
amd_comgr_data_set_t output;
|
||
|
|
amd_comgr_data_set_t input = compileInputs;
|
||
|
|
|
||
|
|
if (auto res = createAction(action, compileOptions, isa, lang); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::create_data_set(&output); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res =
|
||
|
|
amd::Comgr::do_action(AMD_COMGR_ACTION_COMPILE_SOURCE_TO_BC, action, input, output);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
extractBuildLog(output, buildLog);
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractBuildLog(output, buildLog)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractByteCodeBinary(output, AMD_COMGR_DATA_KIND_BC, LLVMBitcode)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Clean up
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool linkLLVMBitcode(const amd_comgr_data_set_t linkInputs, const std::string& isa,
|
||
|
|
std::vector<std::string>& linkOptions, std::string& buildLog,
|
||
|
|
std::vector<char>& LinkedLLVMBitcode) {
|
||
|
|
amd_comgr_language_t lang = AMD_COMGR_LANGUAGE_HIP;
|
||
|
|
amd_comgr_action_info_t action;
|
||
|
|
amd_comgr_data_set_t dataSetDevLibs;
|
||
|
|
|
||
|
|
if (auto res = createAction(action, linkOptions, isa, AMD_COMGR_LANGUAGE_HIP);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::create_data_set(&dataSetDevLibs); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::do_action(AMD_COMGR_ACTION_ADD_DEVICE_LIBRARIES, action, linkInputs,
|
||
|
|
dataSetDevLibs);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
extractBuildLog(dataSetDevLibs, buildLog);
|
||
|
|
LogPrintfInfo("%s", buildLog.c_str());
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractBuildLog(dataSetDevLibs, buildLog)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd_comgr_data_set_t output;
|
||
|
|
if (auto res = amd::Comgr::create_data_set(&output); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res =
|
||
|
|
amd::Comgr::do_action(AMD_COMGR_ACTION_LINK_BC_TO_BC, action, dataSetDevLibs, output);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractBuildLog(output, buildLog)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractByteCodeBinary(output, AMD_COMGR_DATA_KIND_BC, LinkedLLVMBitcode)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(dataSetDevLibs);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool createExecutable(const amd_comgr_data_set_t linkInputs, const std::string& isa,
|
||
|
|
std::vector<std::string>& exeOptions, std::string& buildLog,
|
||
|
|
std::vector<char>& executable) {
|
||
|
|
amd_comgr_action_info_t action;
|
||
|
|
|
||
|
|
if (auto res = createAction(action, exeOptions, isa); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd_comgr_data_set_t relocatableData;
|
||
|
|
if (auto res = amd::Comgr::create_data_set(&relocatableData); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::do_action(AMD_COMGR_ACTION_CODEGEN_BC_TO_RELOCATABLE, action,
|
||
|
|
linkInputs, relocatableData);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
extractBuildLog(relocatableData, buildLog);
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractBuildLog(relocatableData, buildLog)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
std::vector<std::string> emptyOpt;
|
||
|
|
if (auto res = createAction(action, emptyOpt, isa); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd_comgr_data_set_t output;
|
||
|
|
if (auto res = amd::Comgr::create_data_set(&output); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::do_action(AMD_COMGR_ACTION_LINK_RELOCATABLE_TO_EXECUTABLE, action,
|
||
|
|
relocatableData, output);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
extractBuildLog(output, buildLog);
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractBuildLog(output, buildLog)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!extractByteCodeBinary(output, AMD_COMGR_DATA_KIND_EXECUTABLE, executable)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(output);
|
||
|
|
amd::Comgr::destroy_data_set(relocatableData);
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool dumpIsaFromBC(const amd_comgr_data_set_t isaInputs, const std::string& isa,
|
||
|
|
std::vector<std::string>& exeOptions, std::string name, std::string& buildLog) {
|
||
|
|
if (name.size() == 0) return false;
|
||
|
|
amd_comgr_action_info_t action;
|
||
|
|
|
||
|
|
if (auto res = createAction(action, exeOptions, isa); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd_comgr_data_set_t isaData;
|
||
|
|
if (auto res = amd::Comgr::create_data_set(&isaData); res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::do_action(AMD_COMGR_ACTION_CODEGEN_BC_TO_ASSEMBLY, action, isaInputs,
|
||
|
|
isaData);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
extractBuildLog(isaData, buildLog);
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(isaData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<char> isaOutput;
|
||
|
|
if (!extractByteCodeBinary(isaData, AMD_COMGR_DATA_KIND_SOURCE, isaOutput)) {
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(isaData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto isaFileName = name + ".s";
|
||
|
|
std::ofstream f(isaFileName.c_str(), std::ios::trunc | std::ios::binary);
|
||
|
|
if (f.is_open()) {
|
||
|
|
f.write(isaOutput.data(), isaOutput.size());
|
||
|
|
f.close();
|
||
|
|
} else {
|
||
|
|
buildLog += "Warning: writing isa file failed.\n";
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(isaData);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
amd::Comgr::destroy_action_info(action);
|
||
|
|
amd::Comgr::destroy_data_set(isaData);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool demangleName(const std::string& mangledName, std::string& demangledName) {
|
||
|
|
amd_comgr_data_t mangled_data;
|
||
|
|
amd_comgr_data_t demangled_data;
|
||
|
|
|
||
|
|
if (AMD_COMGR_STATUS_SUCCESS != amd::Comgr::create_data(AMD_COMGR_DATA_KIND_BYTES, &mangled_data))
|
||
|
|
return false;
|
||
|
|
|
||
|
|
if (AMD_COMGR_STATUS_SUCCESS !=
|
||
|
|
amd::Comgr::set_data(mangled_data, mangledName.size(), mangledName.c_str())) {
|
||
|
|
amd::Comgr::release_data(mangled_data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (AMD_COMGR_STATUS_SUCCESS != amd::Comgr::demangle_symbol_name(mangled_data, &demangled_data)) {
|
||
|
|
amd::Comgr::release_data(mangled_data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t demangled_size = 0;
|
||
|
|
if (AMD_COMGR_STATUS_SUCCESS != amd::Comgr::get_data(demangled_data, &demangled_size, NULL)) {
|
||
|
|
amd::Comgr::release_data(mangled_data);
|
||
|
|
amd::Comgr::release_data(demangled_data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
demangledName.resize(demangled_size);
|
||
|
|
|
||
|
|
if (AMD_COMGR_STATUS_SUCCESS !=
|
||
|
|
amd::Comgr::get_data(demangled_data, &demangled_size,
|
||
|
|
const_cast<char*>(demangledName.data()))) {
|
||
|
|
amd::Comgr::release_data(mangled_data);
|
||
|
|
amd::Comgr::release_data(demangled_data);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd::Comgr::release_data(mangled_data);
|
||
|
|
amd::Comgr::release_data(demangled_data);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string handleMangledName(std::string loweredName) {
|
||
|
|
if (loweredName.empty()) {
|
||
|
|
return loweredName;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (loweredName.find(".kd") != std::string::npos) {
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
|
||
|
|
if (loweredName.find("void ") == 0) {
|
||
|
|
loweredName.erase(0, strlen("void "));
|
||
|
|
}
|
||
|
|
|
||
|
|
auto dx{loweredName.find_first_of("(<")};
|
||
|
|
|
||
|
|
if (dx == std::string::npos) {
|
||
|
|
return loweredName;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (loweredName[dx] == '<') {
|
||
|
|
uint32_t count = 1;
|
||
|
|
do {
|
||
|
|
++dx;
|
||
|
|
count += (loweredName[dx] == '<') ? 1 : ((loweredName[dx] == '>') ? -1 : 0);
|
||
|
|
} while (count);
|
||
|
|
|
||
|
|
loweredName.erase(++dx);
|
||
|
|
} else {
|
||
|
|
loweredName.erase(dx);
|
||
|
|
}
|
||
|
|
|
||
|
|
return loweredName;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool fillDemangledNames(std::vector<char>& executable, std::vector<std::string>& mangledNames) {
|
||
|
|
amd_comgr_data_t dataObject;
|
||
|
|
if (auto res = amd::Comgr::create_data(AMD_COMGR_DATA_KIND_EXECUTABLE, &dataObject);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto res = amd::Comgr::set_data(dataObject, executable.size(), executable.data())) {
|
||
|
|
amd::Comgr::release_data(dataObject);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
auto callback = [](amd_comgr_symbol_t symbol, void* data) {
|
||
|
|
if (data == nullptr) return AMD_COMGR_STATUS_ERROR_INVALID_ARGUMENT;
|
||
|
|
size_t len = 0;
|
||
|
|
if (auto res = amd::Comgr::symbol_get_info(symbol, AMD_COMGR_SYMBOL_INFO_NAME_LENGTH, &len);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS)
|
||
|
|
return res;
|
||
|
|
std::string name(len, 0);
|
||
|
|
if (auto res = amd::Comgr::symbol_get_info(symbol, AMD_COMGR_SYMBOL_INFO_NAME, &name[0]);
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS)
|
||
|
|
return res;
|
||
|
|
auto storage = reinterpret_cast<std::vector<std::string>*>(data);
|
||
|
|
storage->push_back(name);
|
||
|
|
return AMD_COMGR_STATUS_SUCCESS;
|
||
|
|
};
|
||
|
|
|
||
|
|
if (auto res =
|
||
|
|
amd::Comgr::iterate_symbols(dataObject, callback, reinterpret_cast<void*>(&mangledNames));
|
||
|
|
res != AMD_COMGR_STATUS_SUCCESS) {
|
||
|
|
amd::Comgr::release_data(dataObject);
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
amd::Comgr::release_data(dataObject);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool getMangledNames(const std::vector<std::string>& mangledNames,
|
||
|
|
std::map<std::string, std::string>& strippedNames,
|
||
|
|
std::map<std::string, std::string>& demangledNames) {
|
||
|
|
for (auto& i : mangledNames) {
|
||
|
|
std::string demangledName;
|
||
|
|
if (!demangleName(i, demangledName)) return false;
|
||
|
|
demangledName = handleMangledName(demangledName);
|
||
|
|
|
||
|
|
demangledName.erase(std::remove_if(demangledName.begin(), demangledName.end(),
|
||
|
|
[](unsigned char c) { return std::isspace(c); }),
|
||
|
|
demangledName.end());
|
||
|
|
|
||
|
|
if (auto res = strippedNames.find(demangledName); res != strippedNames.end()) {
|
||
|
|
auto& strippedName = res->second;
|
||
|
|
if (auto dres = demangledNames.find(strippedName); dres != demangledNames.end()) {
|
||
|
|
dres->second = i;
|
||
|
|
continue;
|
||
|
|
} else {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto dres = demangledNames.find(demangledName); dres != demangledNames.end()) {
|
||
|
|
dres->second = i;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
} // namespace helpers
|
||
|
|
} // namespace hiprtc
|