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>
91 lines
3.2 KiB
C++
91 lines
3.2 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.
|
|
|
|
#pragma once
|
|
|
|
#include "library/components/roctracer.hpp"
|
|
#include "library/config.hpp"
|
|
#include "library/debug.hpp"
|
|
#include "library/perfetto.hpp"
|
|
#include "library/ptl.hpp"
|
|
|
|
#include <roctracer.h>
|
|
|
|
#include <iostream>
|
|
#include <memory>
|
|
|
|
// Macro to check ROC-tracer calls status
|
|
#define OMNITRACE_ROCTRACER_CALL(call) \
|
|
{ \
|
|
OMNITRACE_DEBUG_F(#call); \
|
|
int err = call; \
|
|
if(err != 0) \
|
|
{ \
|
|
OMNITRACE_PRINT_F("%s in: %s\n", roctracer_error_string(), #call); \
|
|
} \
|
|
}
|
|
|
|
namespace omnitrace
|
|
{
|
|
using roctracer_bundle_t =
|
|
tim::component_bundle<project::omnitrace, comp::roctracer_data, comp::wall_clock>;
|
|
using roctracer_hsa_bundle_t =
|
|
tim::component_bundle<project::omnitrace, comp::roctracer_data>;
|
|
using roctracer_functions_t = std::vector<std::pair<std::string, std::function<void()>>>;
|
|
|
|
// HSA API callback function
|
|
void
|
|
hsa_api_callback(uint32_t domain, uint32_t cid, const void* callback_data, void* arg);
|
|
|
|
void
|
|
hsa_activity_callback(uint32_t op, const activity_record_t* record, void* arg);
|
|
|
|
void
|
|
hip_exec_activity_callbacks(int64_t _tid);
|
|
|
|
// HIP API callback function
|
|
void
|
|
hip_api_callback(uint32_t domain, uint32_t cid, const void* callback_data, void* arg);
|
|
|
|
void
|
|
roctx_api_callback(uint32_t domain, uint32_t cid, const void* callback_data, void* arg);
|
|
|
|
// Activity tracing callback
|
|
void
|
|
hip_activity_callback(const char* begin, const char* end, void*);
|
|
|
|
bool&
|
|
roctracer_is_init();
|
|
|
|
bool&
|
|
roctracer_is_setup();
|
|
|
|
int64_t
|
|
get_clock_skew();
|
|
|
|
roctracer_functions_t&
|
|
roctracer_setup_routines();
|
|
|
|
roctracer_functions_t&
|
|
roctracer_shutdown_routines();
|
|
} // namespace omnitrace
|