Jonathan R. Madsen
589a729702
* roctracer: use multiple tracks for HIP streams Use different perfetto tracks for each stream, and set the name of these tracks to the stream pointer values. Setting the name like this matches the args in the API traces. This fixes overlapping work on multiple streams appearing as a call stack. * Fix -pedantic * Run clang-format * Add option to disable per stream tracks in perfetto * Updated scheme for roctracer activity + general roctracer fixes - Per-device tracks - Handle HSA OPS in ROCm 5.3 - the changes in ROCm 5.3 were causing HSA ops to get discarded - Default for OMNITRACE_ROCTRACER_DISCARD_INVALID is now zero - i.e. default behavior is to flip beg_ns and end_ns when beg_ns > end_ns * Flush perfetto at end of hip_activity_callback - fixes unterminated regions * GitHub Actions and run-ci script updates - improve reliability * Set OMNITRACE_TMPDIR in testing - files in /tmp get occasionally deleted during CI Co-authored-by: Gergely Meszaros <gergely@streamhpc.com>
85 строки
2.5 KiB
C++
85 строки
2.5 KiB
C++
// MIT License
|
|
//
|
|
// Copyright (c) 2022 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 "library/tracing.hpp"
|
|
#include "library/config.hpp"
|
|
#include "library/state.hpp"
|
|
#include "library/thread_info.hpp"
|
|
|
|
namespace omnitrace
|
|
{
|
|
namespace tracing
|
|
{
|
|
perfetto::TraceConfig&
|
|
get_perfetto_config()
|
|
{
|
|
static auto _v = ::perfetto::TraceConfig{};
|
|
return _v;
|
|
}
|
|
|
|
std::unique_ptr<perfetto::TracingSession>&
|
|
get_perfetto_session()
|
|
{
|
|
static auto _v = std::unique_ptr<perfetto::TracingSession>{};
|
|
return _v;
|
|
}
|
|
|
|
std::unordered_map<hash_value_t, std::string>&
|
|
get_perfetto_track_uuids()
|
|
{
|
|
static thread_local auto _v = std::unordered_map<hash_value_t, std::string>{};
|
|
return _v;
|
|
}
|
|
|
|
std::vector<std::function<void()>>&
|
|
get_finalization_functions()
|
|
{
|
|
static auto _v = std::vector<std::function<void()>>{};
|
|
return _v;
|
|
}
|
|
|
|
tim::hash_map_ptr_t&
|
|
get_timemory_hash_ids(int64_t _tid)
|
|
{
|
|
static auto _v = std::array<tim::hash_map_ptr_t, omnitrace::max_supported_threads>{};
|
|
return _v.at(_tid);
|
|
}
|
|
|
|
tim::hash_alias_ptr_t&
|
|
get_timemory_hash_aliases(int64_t _tid)
|
|
{
|
|
static auto _v =
|
|
std::array<tim::hash_alias_ptr_t, omnitrace::max_supported_threads>{};
|
|
return _v.at(_tid);
|
|
}
|
|
|
|
void
|
|
record_thread_start_time()
|
|
{
|
|
static thread_local std::once_flag _once{};
|
|
std::call_once(_once, []() {
|
|
thread_info::set_start(comp::wall_clock::record(), get_mode() != Mode::Sampling);
|
|
});
|
|
}
|
|
} // namespace tracing
|
|
} // namespace omnitrace
|