AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adding source snapshot and partial serialization (#99)

Browse Source 
* Adding source snapshot

* Adding option to serialize only on target kernel

* Fix for tidy

* Formatting

* Testing the new flag

---------

Co-authored-by: Giovanni Baraldi <gbaraldi@amd.com>

[ROCm/rocprofiler-sdk commit: a2fa188e14]
This commit is contained in:
Baraldi, Giovanni
2025-01-11 00:43:06 +01:00
committed by GitHub
parent e226e2a11a
commit 8abb65b166
7 changed files with 85 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler-sdk/source/lib/rocprofiler-sdk-att/code.cpp
+43
View File
@@ -27,6 +27,7 @@
#include <fstream>
#include <iostream>
#include <sstream>
#include <unordered_set>
namespace rocprofiler
{
@@ -34,6 +35,19 @@ namespace att_wrapper
{
#define ATT_CSV_NAME "att_output.csv"
// Builds a json filetree by recursively inserting "path" into the json object.
void
navigate(nlohmann::json& json, std::vector<std::string>& path, const std::string& filename)
{
if(path.size() == 1) json[path.at(0)] = filename;
if(path.size() <= 1) return;
auto& j = json[path.at(0)];
path.erase(path.begin());
navigate(j, path, filename);
}
CodeFile::CodeFile(const Fspath& _dir, std::shared_ptr<AddressTable> _table)
: dir(_dir)
, filename(_dir / "code.json")
@@ -90,6 +104,8 @@ CodeFile::~CodeFile()
nlohmann::json jcode;
std::unordered_set<std::string> snapshots{};
for(auto& line : vec)
{
auto& isa = *line.second;
@@ -110,6 +126,14 @@ CodeFile::~CodeFile()
<< ", " << isa.hitcount << ", " << isa.latency << "]";
jcode.push_back(nlohmann::json::parse(code.str()));
size_t lineref = isa.code_line->comment.rfind(':');
if(lineref == 0 || lineref == std::string::npos) continue;
auto source_ref = isa.code_line->comment.substr(0, lineref);
if(!source_ref.empty() && snapshots.find(source_ref) == snapshots.end())
snapshots.insert(std::move(source_ref));
}
nlohmann::json json;
@@ -117,6 +141,25 @@ CodeFile::~CodeFile()
json["version"] = TOOL_VERSION;
OutputFile(filename) << json;
nlohmann::json jsnapfiletree;
size_t num_snap = 0;
for(auto& source_ref : snapshots)
{
if(rocprofiler::common::filesystem::exists(source_ref))
{
Fspath filepath(source_ref);
std::stringstream newfile;
newfile << "source_" << (num_snap++) << '_' << filepath.filename().string();
std::vector<std::string> path_elements(filepath.begin(), filepath.end());
navigate(jsnapfiletree, path_elements, newfile.str());
rocprofiler::common::filesystem::copy(filepath, dir / newfile.str());
}
}
if(num_snap != 0) OutputFile(dir / "snapshots.json") << jsnapfiletree;
}
} // namespace att_wrapper