Jonathan R. Madsen
f6241af5ee
* Update invoke.hpp
* Update OMNITRACE_FUNCTION
* Update library debug messages
* ptl verbosity
* Update timemory submodule
* mpi_gotcha calls omnitrace_finalize_hidden
* omnitrace_{push,pop}_region returns error code
* omnitrace-user updates
- doxygen documentation
- omnitrace_get_user_callbacks
- omnitrace_user_error_string
- omnitrace-user functions return error codes
* Update user-api example
* Tweak to workflows and tests
* Fix for OMNITRACE_FUNCTION
- conditional impl if __GNUC__ < 9
* focal-external-rocm workflow update
116 linhas
3.7 KiB
C++
116 linhas
3.7 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 "common/defines.h"
|
|
#include "common/join.hpp"
|
|
|
|
#include <atomic>
|
|
#include <cstring>
|
|
#include <functional>
|
|
#include <string>
|
|
|
|
#if !defined(OMNITRACE_COMMON_LIBRARY_NAME)
|
|
# define OMNITRACE_COMMON_LIBRARY_NAME "common"
|
|
# error OMNITRACE_COMMON_LIBRARY_NAME must be defined
|
|
#endif
|
|
|
|
namespace omnitrace
|
|
{
|
|
inline namespace common
|
|
{
|
|
namespace
|
|
{
|
|
template <typename FuncT, typename... Args>
|
|
inline auto
|
|
invoke(const char* _name, FuncT&& _func, Args... _args) OMNITRACE_HIDDEN_API;
|
|
|
|
inline int32_t&
|
|
get_guard()
|
|
{
|
|
static thread_local int32_t _v = 0;
|
|
return _v;
|
|
}
|
|
|
|
inline int64_t
|
|
get_thread_index()
|
|
{
|
|
static std::atomic<int64_t> _c{ 0 };
|
|
static thread_local auto _v = _c++;
|
|
return _v;
|
|
}
|
|
|
|
template <typename FuncT, typename... Args>
|
|
auto
|
|
invoke(const char* _name, int _verbose, FuncT&& _func, Args... _args)
|
|
{
|
|
if(_func)
|
|
{
|
|
struct decrement_guard
|
|
{
|
|
// decrement the guard as it exits the scope
|
|
~decrement_guard() { --get_guard(); }
|
|
} _unlk{};
|
|
|
|
// if _lk is ever greater than zero on the same thread, this
|
|
// means a function within the current function is calling
|
|
// our instrumentation so we ignore the call
|
|
int32_t _lk = get_guard()++;
|
|
if(_lk == 0)
|
|
{
|
|
if(_verbose >= 3)
|
|
{
|
|
fflush(stderr);
|
|
fprintf(stderr,
|
|
"[omnitrace][" OMNITRACE_COMMON_LIBRARY_NAME
|
|
"][%li][%i] %s(%s)\n",
|
|
get_thread_index(), _lk, _name, join(", ", _args...).c_str());
|
|
fflush(stderr);
|
|
}
|
|
return std::invoke(std::forward<FuncT>(_func), _args...);
|
|
}
|
|
else if(_verbose >= 2)
|
|
{
|
|
fflush(stderr);
|
|
fprintf(stderr,
|
|
"[omnitrace][" OMNITRACE_COMMON_LIBRARY_NAME
|
|
"][%li] %s(%s) was guarded :: value = %i\n",
|
|
get_thread_index(), _name, join(", ", _args...).c_str(), _lk);
|
|
fflush(stderr);
|
|
}
|
|
}
|
|
else if(_verbose >= 0)
|
|
{
|
|
fprintf(stderr,
|
|
"[omnitrace][" OMNITRACE_COMMON_LIBRARY_NAME
|
|
"][%li] %s(%s) ignored :: null function pointer\n",
|
|
get_thread_index(), _name, join(", ", _args...).c_str());
|
|
}
|
|
|
|
using return_type = decltype(std::invoke(std::forward<FuncT>(_func), _args...));
|
|
if constexpr(!std::is_void<return_type>::value) return return_type();
|
|
}
|
|
} // namespace
|
|
} // namespace common
|
|
} // namespace omnitrace
|