SWDEV-492607: Adding ATT wrapper (#40)
* Adding att parser wrapper * Adding ATT tests as optional * Adding decoder API for query capability * Removed samples * Formatting * adding new line * Removed perfetto and moved to static library * using default search for lib * Updated to SDK * Namespace changes * Added tests * Small refactor * Updated API to receive agent_id * Fixing tests * Tidy fixes * Not write to file * Switch to filesystem.hpp * Compilation fixes * Formatting * Tidy fix * Removed likely * Adding tests * Added gfx9 test * Adding gfx12 tests * Formatting * Enable tidy * Fix tests * Fix deadlock on agent test * Workaround ASAN * Moving query outside class. * Fix standalone tool * Addressing comments * Formatting * Change query name * Fixed some tests. Updated PR comments. * Formatting * Improved coverage * Formatting * Fix for comments * Formatting * Adding some description. Fix error type. --------- Co-authored-by: Giovanni Baraldi <gbaraldi@amd.com>
This commit is contained in:
Baraldi, Giovanni
committed by
GitHub
GitHub
parent
9de284a568
commit
2c8e88a76b
@@ -0,0 +1,197 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2024 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 "att_lib_wrapper.hpp"
|
||||
#include "dl.hpp"
|
||||
#include "filenames.hpp"
|
||||
#include "occupancy.hpp"
|
||||
#include "profile_interface.hpp"
|
||||
#include "wave.hpp"
|
||||
#include "wstates.hpp"
|
||||
|
||||
#include "lib/rocprofiler-sdk/registration.hpp"
|
||||
|
||||
#include <dlfcn.h>
|
||||
#include <algorithm>
|
||||
#include <fstream>
|
||||
#include <iostream>
|
||||
#include <mutex>
|
||||
#include <nlohmann/json.hpp>
|
||||
#include <vector>
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace att_wrapper
|
||||
{
|
||||
auto
|
||||
get_lib_names()
|
||||
{
|
||||
std::vector<std::pair<tool_att_capability_t, const char*>> lib_names = {
|
||||
{tool_att_capability_t::ATT_CAPABILITIES_DEBUG, "libatt_decoder_debug.so"},
|
||||
{tool_att_capability_t::ATT_CAPABILITIES_TRACE, "libatt_decoder_trace.so"},
|
||||
{tool_att_capability_t::ATT_CAPABILITIES_SUMMARY, "libatt_decoder_summary.so"},
|
||||
{tool_att_capability_t::ATT_CAPABILITIES_TESTING, "libatt_decoder_testing.so"},
|
||||
};
|
||||
return lib_names;
|
||||
}
|
||||
|
||||
ATTFileMgr::ATTFileMgr(Fspath _dir, std::shared_ptr<DL> _dl)
|
||||
: dir(std::move(_dir))
|
||||
, dl(std::move(_dl))
|
||||
{
|
||||
rocprofiler::common::filesystem::create_directories(dir);
|
||||
table = std::make_shared<AddressTable>();
|
||||
codefile = std::make_shared<CodeFile>(dir, table);
|
||||
filenames = std::make_shared<FilenameMgr>(dir);
|
||||
for(size_t i = 0; i < ATT_WAVE_STATE_LAST; i++)
|
||||
wstates.at(i) = std::make_shared<WstatesFile>(i, dir);
|
||||
}
|
||||
|
||||
ATTFileMgr::~ATTFileMgr() { OccupancyFile::OccupancyFile(dir, table, occupancy); }
|
||||
|
||||
void
|
||||
ATTFileMgr::parseShader(int se_id, const std::vector<char>& data)
|
||||
{
|
||||
WaveConfig config(se_id, filenames, codefile, wstates);
|
||||
ToolData tooldata(data, config, dl);
|
||||
|
||||
if(config.occupancy.size()) occupancy.emplace(se_id, std::move(config.occupancy));
|
||||
|
||||
for(auto& [pc, kernel] : config.kernel_names)
|
||||
codefile->kernel_names.emplace(pc, std::move(kernel));
|
||||
}
|
||||
|
||||
int
|
||||
get_shader_id(const std::string& name)
|
||||
{
|
||||
auto run_pos = name.rfind('_');
|
||||
if(run_pos == std::string::npos) throw std::runtime_error("Invalid name");
|
||||
|
||||
std::string_view stripped = name.substr(0, run_pos);
|
||||
auto se_number_pos = stripped.rfind('_');
|
||||
if(se_number_pos == std::string::npos || se_number_pos + 1 >= stripped.size())
|
||||
throw std::runtime_error("Invalid name");
|
||||
|
||||
return std::stoi(std::string(stripped.substr(se_number_pos + 1)));
|
||||
}
|
||||
|
||||
std::vector<tool_att_capability_t>
|
||||
query_att_decode_capability()
|
||||
{
|
||||
std::vector<tool_att_capability_t> ret;
|
||||
|
||||
for(auto& [cap, libname] : get_lib_names())
|
||||
{
|
||||
if(auto handle = dlopen(libname, RTLD_NOW | RTLD_LOCAL))
|
||||
{
|
||||
dlclose(handle);
|
||||
ret.push_back(cap);
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
ATTDecoder::ATTDecoder(tool_att_capability_t capability)
|
||||
{
|
||||
for(auto& [cap, libname] : get_lib_names())
|
||||
{
|
||||
if(cap == capability)
|
||||
{
|
||||
dl = std::make_shared<DL>(libname);
|
||||
return;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
void
|
||||
ATTDecoder::parse(const Fspath& input_dir,
|
||||
const Fspath& output_dir,
|
||||
const std::vector<std::string>& att_files,
|
||||
const std::vector<CodeobjLoadInfo>& codeobj_files,
|
||||
const std::string& output_formats)
|
||||
{
|
||||
auto& formats = GlobalDefs::get().output_formats;
|
||||
formats = output_formats;
|
||||
std::transform(formats.begin(), formats.end(), formats.begin(), [](unsigned char c) {
|
||||
return std::tolower(c);
|
||||
});
|
||||
|
||||
ATTFileMgr mgr(output_dir, dl);
|
||||
|
||||
for(auto& file : codeobj_files)
|
||||
{
|
||||
if(file.name.find("memory://") == 0)
|
||||
{
|
||||
ROCP_WARNING << file.name << " was not loaded";
|
||||
continue;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
mgr.table->addDecoder((input_dir / file.name).c_str(), file.id, file.addr, file.size);
|
||||
} catch(std::exception& e)
|
||||
{
|
||||
ROCP_ERROR << file.id << ':' << file.name << " - " << e.what() << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
for(auto& shader : att_files)
|
||||
{
|
||||
int shader_id = 0;
|
||||
try
|
||||
{
|
||||
shader_id = get_shader_id(shader);
|
||||
} catch(std::exception& e)
|
||||
{
|
||||
ROCP_WARNING << "Could not retrieve shader_id: " << e.what();
|
||||
continue;
|
||||
}
|
||||
|
||||
std::vector<char> shader_data;
|
||||
|
||||
{
|
||||
std::ifstream file(input_dir / shader, std::ios::binary);
|
||||
if(!file.is_open())
|
||||
{
|
||||
ROCP_WARNING << "could not open " << shader;
|
||||
continue;
|
||||
}
|
||||
|
||||
file.seekg(0, std::ios::end);
|
||||
shader_data.resize(file.tellg());
|
||||
file.seekg(0);
|
||||
file.read(shader_data.data(), shader_data.size());
|
||||
}
|
||||
|
||||
mgr.parseShader(shader_id, shader_data);
|
||||
}
|
||||
}
|
||||
|
||||
bool
|
||||
ATTDecoder::valid() const
|
||||
{
|
||||
return dl && dl->att_parse_data_fn && dl->att_info_fn && dl->att_status_fn;
|
||||
}
|
||||
|
||||
} // namespace att_wrapper
|
||||
} // namespace rocprofiler
|
||||
Reference in New Issue
Block a user