Files
rocm-systems/source/lib/omnitrace/api.cpp
T
Jonathan R. Madsen 642b6b95ca Support external (i.e. user-defined) trace annotations (#195)
* Support external (i.e. user-defined) trace annotations

- tweaked the python examples to be more balanced
- updated the user-api example to conform to user API changes
- moved the get/set for State and ThreadState to state.{hpp,cpp}
- introduced user-provided trace annotations
- added perfetto python category
- moved coverage impl files around
- created enumerations for mapping category enums to category types
- created enumerations for mapping annotation type enums to annotation values
- moved tracing::add_perfetto_annotations to tracing/annotation.hpp
- utility make_index_sequence_range
- libomnitrace-dl: omnitrace_push_category_region
- libomnitrace-dl: omnitrace_pop_category_region
- libomnitrace-user: omnitrace_user_push_annotated_region
- libomnitrace-user: omnitrace_user_pop_annotated_region
- libpyomnitrace: support extra annotations via annotate_trace config value
  - filename
  - line
  - last attempted instr in bytecode (lasti)
  - argcount
  - num local variables
  - stacksize
- omnitrace-python: -a / --annotate-traces option

* tweak ubuntu-focal workflow

* Fix installation of omnitrace-user headers

* ubuntu-focal-codecov workflow update

- Install texinfo

* Update timemory submodule
2022-11-11 07:31:14 -06:00

157 строки
4.1 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 "api.hpp"
#include "library/debug.hpp"
#include <exception>
#include <stdexcept>
extern "C" void
omnitrace_push_trace(const char* _name)
{
omnitrace_push_trace_hidden(_name);
}
extern "C" void
omnitrace_pop_trace(const char* _name)
{
omnitrace_pop_trace_hidden(_name);
}
extern "C" int
omnitrace_push_region(const char* _name)
{
try
{
omnitrace_push_region_hidden(_name);
} catch(std::exception& _e)
{
OMNITRACE_VERBOSE_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" int
omnitrace_pop_region(const char* _name)
{
try
{
omnitrace_pop_region_hidden(_name);
} catch(std::exception& _e)
{
OMNITRACE_VERBOSE_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" int
omnitrace_push_category_region(omnitrace_category_t _category, const char* _name,
omnitrace_annotation_t* _annotations,
size_t _annotation_count)
{
try
{
omnitrace_push_category_region_hidden(_category, _name, _annotations,
_annotation_count);
} catch(std::exception& _e)
{
OMNITRACE_VERBOSE_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" int
omnitrace_pop_category_region(omnitrace_category_t _category, const char* _name,
omnitrace_annotation_t* _annotations,
size_t _annotation_count)
{
try
{
omnitrace_pop_category_region_hidden(_category, _name, _annotations,
_annotation_count);
} catch(std::exception& _e)
{
OMNITRACE_VERBOSE_F(1, "Exception caught: %s\n", _e.what());
return -1;
}
return 0;
}
extern "C" void
omnitrace_init_library(void)
{
omnitrace_init_library_hidden();
}
extern "C" void
omnitrace_init_tooling(void)
{
omnitrace_init_tooling_hidden();
}
extern "C" void
omnitrace_init(const char* _mode, bool _rewrite, const char* _arg0)
{
omnitrace_init_hidden(_mode, _rewrite, _arg0);
}
extern "C" void
omnitrace_finalize(void)
{
omnitrace_finalize_hidden();
}
extern "C" void
omnitrace_reset_preload(void)
{
omnitrace_reset_preload_hidden();
}
extern "C" void
omnitrace_set_env(const char* env_name, const char* env_val)
{
omnitrace_set_env_hidden(env_name, env_val);
}
extern "C" void
omnitrace_set_mpi(bool use, bool attached)
{
omnitrace_set_mpi_hidden(use, attached);
}
extern "C" void
omnitrace_register_source(const char* file, const char* func, size_t line, size_t address,
const char* source)
{
omnitrace_register_source_hidden(file, func, line, address, source);
}
extern "C" void
omnitrace_register_coverage(const char* file, const char* func, size_t address)
{
omnitrace_register_coverage_hidden(file, func, address);
}