Runtime Initialization Tracing (#1105)
* Runtime initialization tracing - calbacks and buffer entries notifying when a runtime has been initialized * Minor cleanup to registration.cpp * JSON tool implementation * Increase perfetto_reader timeout * Handle perfetto_reader timeout when attr doesn't exist * clang-tidy fixes to memory_allocation.cpp
Esse commit está contido em:
Jonathan R. Madsen
GitHub
@@ -0,0 +1,162 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023 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 "lib/rocprofiler-sdk/runtime_initialization.hpp"
|
||||
#include "lib/common/utility.hpp"
|
||||
#include "lib/rocprofiler-sdk/context/context.hpp"
|
||||
#include "lib/rocprofiler-sdk/tracing/fwd.hpp"
|
||||
#include "lib/rocprofiler-sdk/tracing/tracing.hpp"
|
||||
|
||||
#include <rocprofiler-sdk/fwd.h>
|
||||
|
||||
#include <cstdint>
|
||||
#include <mutex>
|
||||
#include <stdexcept>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace runtime_init
|
||||
{
|
||||
namespace
|
||||
{
|
||||
template <size_t OpIdx>
|
||||
struct runtime_init_info;
|
||||
|
||||
#define SPECIALIZE_RUNTIME_INIT_INFO(NAME, PRETTY_NAME) \
|
||||
template <> \
|
||||
struct runtime_init_info<ROCPROFILER_RUNTIME_INITIALIZATION_##NAME> \
|
||||
{ \
|
||||
static constexpr auto operation_idx = ROCPROFILER_RUNTIME_INITIALIZATION_##NAME; \
|
||||
static constexpr auto name = "RUNTIME_INITIALIZATION_" #NAME; \
|
||||
static constexpr auto pretty_name = PRETTY_NAME; \
|
||||
};
|
||||
|
||||
SPECIALIZE_RUNTIME_INIT_INFO(NONE, "<unknown-runtime>")
|
||||
SPECIALIZE_RUNTIME_INIT_INFO(HSA, "HSA runtime")
|
||||
SPECIALIZE_RUNTIME_INIT_INFO(HIP, "HIP runtime")
|
||||
SPECIALIZE_RUNTIME_INIT_INFO(MARKER, "Marker (ROCTx) runtime")
|
||||
SPECIALIZE_RUNTIME_INIT_INFO(RCCL, "RCCL runtime")
|
||||
|
||||
#undef SPECIALIZE_RUNTIME_INIT_INFO
|
||||
|
||||
template <size_t Idx, size_t... IdxTail>
|
||||
std::pair<const char*, const char*>
|
||||
name_by_id(const uint32_t id, std::index_sequence<Idx, IdxTail...>)
|
||||
{
|
||||
if(Idx == id) return {runtime_init_info<Idx>::name, runtime_init_info<Idx>::pretty_name};
|
||||
if constexpr(sizeof...(IdxTail) > 0)
|
||||
return name_by_id(id, std::index_sequence<IdxTail...>{});
|
||||
else
|
||||
return {nullptr, nullptr};
|
||||
}
|
||||
|
||||
template <size_t... Idx>
|
||||
void
|
||||
get_ids(std::vector<uint32_t>& _id_list, std::index_sequence<Idx...>)
|
||||
{
|
||||
auto _emplace = [](auto& _vec, uint32_t _v) {
|
||||
if(_v < static_cast<uint32_t>(ROCPROFILER_RUNTIME_INITIALIZATION_LAST))
|
||||
_vec.emplace_back(_v);
|
||||
};
|
||||
|
||||
(_emplace(_id_list, runtime_init_info<Idx>::operation_idx), ...);
|
||||
}
|
||||
} // namespace
|
||||
|
||||
const char*
|
||||
name_by_id(uint32_t id)
|
||||
{
|
||||
return name_by_id(id, std::make_index_sequence<ROCPROFILER_RUNTIME_INITIALIZATION_LAST>{})
|
||||
.first;
|
||||
}
|
||||
|
||||
const char*
|
||||
pretty_name_by_id(uint32_t id)
|
||||
{
|
||||
return name_by_id(id, std::make_index_sequence<ROCPROFILER_RUNTIME_INITIALIZATION_LAST>{})
|
||||
.second;
|
||||
}
|
||||
|
||||
std::vector<uint32_t>
|
||||
get_ids()
|
||||
{
|
||||
auto _data = std::vector<uint32_t>{};
|
||||
_data.reserve(ROCPROFILER_RUNTIME_INITIALIZATION_LAST);
|
||||
get_ids(_data, std::make_index_sequence<ROCPROFILER_RUNTIME_INITIALIZATION_LAST>{});
|
||||
return _data;
|
||||
}
|
||||
|
||||
void
|
||||
initialize(rocprofiler_runtime_initialization_operation_t operation_idx,
|
||||
uint64_t lib_version,
|
||||
uint64_t lib_instance)
|
||||
{
|
||||
using runtime_init_record_t = rocprofiler_buffer_tracing_runtime_initialization_record_t;
|
||||
using runtime_init_data_t = rocprofiler_callback_tracing_runtime_initialization_data_t;
|
||||
|
||||
constexpr auto callback_domain_idx = ROCPROFILER_CALLBACK_TRACING_RUNTIME_INITIALIZATION;
|
||||
constexpr auto buffered_domain_idx = ROCPROFILER_BUFFER_TRACING_RUNTIME_INITIALIZATION;
|
||||
constexpr auto corr_id = rocprofiler_correlation_id_t{0, rocprofiler_user_data_t{.value = 0}};
|
||||
|
||||
ROCP_INFO << pretty_name_by_id(operation_idx) << " has been initialized";
|
||||
|
||||
auto thr_id = common::get_tid();
|
||||
auto data = tracing::tracing_data{};
|
||||
|
||||
tracing::populate_contexts(callback_domain_idx, buffered_domain_idx, operation_idx, data);
|
||||
|
||||
auto tracer_data = common::init_public_api_struct(runtime_init_data_t{});
|
||||
auto buffer_record = common::init_public_api_struct(runtime_init_record_t{});
|
||||
|
||||
{
|
||||
tracer_data.version = lib_version;
|
||||
tracer_data.instance = lib_instance;
|
||||
tracing::execute_phase_none_callbacks(data.callback_contexts,
|
||||
thr_id,
|
||||
corr_id.internal,
|
||||
data.external_correlation_ids,
|
||||
callback_domain_idx,
|
||||
operation_idx,
|
||||
tracer_data);
|
||||
}
|
||||
|
||||
{
|
||||
buffer_record.version = lib_version;
|
||||
buffer_record.instance = lib_instance;
|
||||
buffer_record.timestamp = common::timestamp_ns();
|
||||
tracing::execute_buffer_record_emplace(data.buffered_contexts,
|
||||
thr_id,
|
||||
corr_id.internal,
|
||||
data.external_correlation_ids,
|
||||
buffered_domain_idx,
|
||||
operation_idx,
|
||||
buffer_record);
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
finalize()
|
||||
{}
|
||||
} // namespace runtime_init
|
||||
} // namespace rocprofiler
|
||||
Referência em uma Nova Issue
Bloquear um usuário