Jonathan R. Madsen
ebb29ac54a
* Miscellaneous updates
- Updated README
- Updated VERSION
- Header include tweaks
- get_verbose() + get_verbose_env()
- fixes to omnitrace-avail
- exclude all cuda/cupti settings
- apply available_only to hw counters
- config file warnings
- config displayed at verbose > 0
- fix to MPI_Finalize when only using MPI headers
* Updated LICENSE
* CPack tweak
[ROCm/rocprofiler-systems commit: 8648410309]
91 lines
3.9 KiB
C++
91 lines
3.9 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/defines.hpp"
|
|
|
|
#if defined(PERFETTO_CATEGORIES)
|
|
# error "PERFETTO_CATEGORIES is already defined. Please include \"" __FILE__ "\" before including any timemory files"
|
|
#endif
|
|
|
|
#if !defined(TIMEMORY_USE_PERFETTO)
|
|
# include <perfetto.h>
|
|
# define PERFETTO_CATEGORIES \
|
|
perfetto::Category("host").SetDescription("Host-side function tracing"), \
|
|
perfetto::Category("device").SetDescription("Device-side function tracing"), \
|
|
perfetto::Category("host-critical-trace") \
|
|
.SetDescription("Host-side critical traces"), \
|
|
perfetto::Category("device-critical-trace") \
|
|
.SetDescription("Device-side critical traces")
|
|
#else
|
|
# define PERFETTO_CATEGORIES \
|
|
perfetto::Category("host").SetDescription("Host-side function tracing"), \
|
|
perfetto::Category("device").SetDescription("Device-side function tracing"), \
|
|
perfetto::Category("host-critical-trace") \
|
|
.SetDescription("Host-side critical traces"), \
|
|
perfetto::Category("device-critical-trace") \
|
|
.SetDescription("Device-side critical traces"), \
|
|
perfetto::Category("timemory") \
|
|
.SetDescription("Events from the timemory API")
|
|
# define TIMEMORY_PERFETTO_CATEGORIES PERFETTO_CATEGORIES
|
|
#endif
|
|
|
|
#if !defined(TIMEMORY_USE_PERFETTO)
|
|
PERFETTO_DEFINE_CATEGORIES(PERFETTO_CATEGORIES);
|
|
#endif
|
|
|
|
namespace omnitrace
|
|
{
|
|
#if defined(CUSTOM_DATA_SOURCE)
|
|
class CustomDataSource : public perfetto::DataSource<CustomDataSource>
|
|
{
|
|
public:
|
|
void OnSetup(const SetupArgs&) override
|
|
{
|
|
// Use this callback to apply any custom configuration to your data source
|
|
// based on the TraceConfig in SetupArgs.
|
|
PRINT_HERE("%s", "setup");
|
|
}
|
|
|
|
void OnStart(const StartArgs&) override
|
|
{
|
|
// This notification can be used to initialize the GPU driver, enable
|
|
// counters, etc. StartArgs will contains the DataSourceDescriptor,
|
|
// which can be extended.
|
|
PRINT_HERE("%s", "start");
|
|
}
|
|
|
|
void OnStop(const StopArgs&) override
|
|
{
|
|
// Undo any initialization done in OnStart.
|
|
PRINT_HERE("%s", "stop");
|
|
}
|
|
|
|
// Data sources can also have per-instance state.
|
|
int my_custom_state = 0;
|
|
};
|
|
|
|
PERFETTO_DECLARE_DATA_SOURCE_STATIC_MEMBERS(CustomDataSource);
|
|
#endif
|
|
} // namespace omnitrace
|