Add 'projects/rocprofiler-sdk/' from commit 'bf0fad1d5406fbc51403ba1aa9621a9d4a9bce2b'
git-subtree-dir: projects/rocprofiler-sdk git-subtree-mainline:50a90550e9git-subtree-split:bf0fad1d54
This commit is contained in:
systems-assistant[bot]
@@ -0,0 +1,271 @@
|
||||
// MIT License
|
||||
//
|
||||
// Copyright (c) 2023-2025 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 "lib/common/logging.hpp"
|
||||
#include "lib/common/static_object.hpp"
|
||||
#include "lib/common/utility.hpp"
|
||||
|
||||
#include <rocprofiler-sdk-roctx/api_trace.h>
|
||||
#include <rocprofiler-sdk-roctx/defines.h>
|
||||
#include <rocprofiler-sdk-roctx/roctx.h>
|
||||
#include <rocprofiler-sdk-roctx/types.h>
|
||||
|
||||
#include <rocprofiler-register/rocprofiler-register.h>
|
||||
|
||||
#include <array>
|
||||
#include <atomic>
|
||||
#include <cassert>
|
||||
|
||||
#define ROCP_REG_VERSION \
|
||||
ROCPROFILER_REGISTER_COMPUTE_VERSION_3( \
|
||||
ROCTX_VERSION_MAJOR, ROCTX_VERSION_MINOR, ROCTX_VERSION_PATCH)
|
||||
|
||||
ROCPROFILER_REGISTER_DEFINE_IMPORT(roctx, ROCP_REG_VERSION)
|
||||
|
||||
namespace rocprofiler
|
||||
{
|
||||
namespace roctx
|
||||
{
|
||||
namespace
|
||||
{
|
||||
namespace common = ::rocprofiler::common;
|
||||
|
||||
using common::static_buffer_size;
|
||||
using atomic_roctx_range_id_t = std::atomic<roctx_range_id_t>;
|
||||
|
||||
constexpr auto atomic_range_id_size_v = static_buffer_size<atomic_roctx_range_id_t>();
|
||||
thread_local auto nested_range_level_buffer = std::array<std::byte, static_buffer_size<int>()>{};
|
||||
auto start_stop_range_id_buffer = std::array<std::byte, atomic_range_id_size_v>{};
|
||||
|
||||
auto&
|
||||
get_nested_range_level()
|
||||
{
|
||||
static thread_local auto* value = new(nested_range_level_buffer.data()) int{0};
|
||||
return *value;
|
||||
}
|
||||
|
||||
auto&
|
||||
get_start_stop_range_id()
|
||||
{
|
||||
static auto* value = new(start_stop_range_id_buffer.data()) atomic_roctx_range_id_t{0};
|
||||
return *value;
|
||||
}
|
||||
|
||||
int
|
||||
roctxGetThreadId(roctx_thread_id_t* tid)
|
||||
{
|
||||
*tid = common::get_tid();
|
||||
return 0;
|
||||
}
|
||||
|
||||
void
|
||||
roctxMarkA(const char* /*msg*/)
|
||||
{}
|
||||
|
||||
int
|
||||
roctxRangePushA(const char* /*msg*/)
|
||||
{
|
||||
return get_nested_range_level()++;
|
||||
}
|
||||
|
||||
int
|
||||
roctxRangePop()
|
||||
{
|
||||
if(get_nested_range_level() == 0) return -1;
|
||||
return --get_nested_range_level();
|
||||
}
|
||||
|
||||
roctx_range_id_t
|
||||
roctxRangeStartA(const char*)
|
||||
{
|
||||
auto range_id = ++get_start_stop_range_id();
|
||||
return range_id;
|
||||
}
|
||||
|
||||
void roctxRangeStop(roctx_range_id_t /*id*/) {}
|
||||
|
||||
int roctxProfilerPause(roctx_thread_id_t /*tid*/) { return 0; }
|
||||
|
||||
int roctxProfilerResume(roctx_thread_id_t /*tid*/) { return 0; }
|
||||
|
||||
int
|
||||
roctxNameOsThread(const char* /*name*/)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameHsaAgent(const char* /*name*/, const struct hsa_agent_s* /*agent*/)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameHipDevice(const char* /*name*/, int /*devid*/)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameHipStream(const char* /*name*/, const struct ihipStream_t* /*stream*/)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct roctx_api_table
|
||||
{
|
||||
roctxCoreApiTable_t core = common::init_public_api_struct(roctxCoreApiTable_t{});
|
||||
roctxControlApiTable_t control = common::init_public_api_struct(roctxControlApiTable_t{});
|
||||
roctxNameApiTable_t name = common::init_public_api_struct(roctxNameApiTable_t{});
|
||||
};
|
||||
|
||||
auto*&
|
||||
get_table_impl()
|
||||
{
|
||||
rocprofiler::common::init_logging("ROCTX");
|
||||
|
||||
auto*& tbl = common::static_object<roctx_api_table>::construct();
|
||||
|
||||
tbl->core = roctxCoreApiTable_t{sizeof(roctxCoreApiTable_t),
|
||||
&::rocprofiler::roctx::roctxMarkA,
|
||||
&::rocprofiler::roctx::roctxRangePushA,
|
||||
&::rocprofiler::roctx::roctxRangePop,
|
||||
&::rocprofiler::roctx::roctxRangeStartA,
|
||||
&::rocprofiler::roctx::roctxRangeStop,
|
||||
&::rocprofiler::roctx::roctxGetThreadId};
|
||||
|
||||
tbl->control = roctxControlApiTable_t{sizeof(roctxControlApiTable_t),
|
||||
&::rocprofiler::roctx::roctxProfilerPause,
|
||||
&::rocprofiler::roctx::roctxProfilerResume};
|
||||
|
||||
tbl->name = roctxNameApiTable_t{sizeof(roctxNameApiTable_t),
|
||||
&::rocprofiler::roctx::roctxNameOsThread,
|
||||
&::rocprofiler::roctx::roctxNameHsaAgent,
|
||||
&::rocprofiler::roctx::roctxNameHipDevice,
|
||||
&::rocprofiler::roctx::roctxNameHipStream};
|
||||
|
||||
auto table_array = std::array<void*, 3>{&tbl->core, &tbl->control, &tbl->name};
|
||||
auto lib_id = rocprofiler_register_library_indentifier_t{};
|
||||
auto rocp_reg_status =
|
||||
rocprofiler_register_library_api_table("roctx",
|
||||
&ROCPROFILER_REGISTER_IMPORT_FUNC(roctx),
|
||||
ROCP_REG_VERSION,
|
||||
table_array.data(),
|
||||
table_array.size(),
|
||||
&lib_id);
|
||||
|
||||
ROCP_INFO << "[rocprofiler-sdk-roctx][" << getpid() << "] rocprofiler-register returned code "
|
||||
<< rocp_reg_status << ": " << rocprofiler_register_error_string(rocp_reg_status);
|
||||
|
||||
ROCP_WARNING_IF(rocp_reg_status != ROCP_REG_SUCCESS && rocp_reg_status != ROCP_REG_NO_TOOLS)
|
||||
<< "[rocprofiler-sdk-roctx][" << getpid()
|
||||
<< "] rocprofiler-register failed with error code " << rocp_reg_status << ": "
|
||||
<< rocprofiler_register_error_string(rocp_reg_status);
|
||||
|
||||
return tbl;
|
||||
}
|
||||
|
||||
const auto*
|
||||
get_table()
|
||||
{
|
||||
static auto*& tbl = get_table_impl();
|
||||
return tbl;
|
||||
}
|
||||
} // namespace
|
||||
} // namespace roctx
|
||||
} // namespace rocprofiler
|
||||
|
||||
ROCTX_EXTERN_C_INIT
|
||||
|
||||
void
|
||||
roctxMarkA(const char* message)
|
||||
{
|
||||
::rocprofiler::roctx::get_table()->core.roctxMarkA_fn(message);
|
||||
}
|
||||
|
||||
int
|
||||
roctxRangePushA(const char* message)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->core.roctxRangePushA_fn(message);
|
||||
}
|
||||
|
||||
int
|
||||
roctxRangePop()
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->core.roctxRangePop_fn();
|
||||
}
|
||||
|
||||
roctx_range_id_t
|
||||
roctxRangeStartA(const char* message)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->core.roctxRangeStartA_fn(message);
|
||||
}
|
||||
|
||||
void
|
||||
roctxRangeStop(roctx_range_id_t id)
|
||||
{
|
||||
::rocprofiler::roctx::get_table()->core.roctxRangeStop_fn(id);
|
||||
}
|
||||
|
||||
int
|
||||
roctxGetThreadId(roctx_thread_id_t* tid)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->core.roctxGetThreadId_fn(tid);
|
||||
}
|
||||
|
||||
int
|
||||
roctxProfilerPause(roctx_thread_id_t tid)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->control.roctxProfilerPause_fn(tid);
|
||||
}
|
||||
|
||||
int
|
||||
roctxProfilerResume(roctx_thread_id_t tid)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->control.roctxProfilerResume_fn(tid);
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameOsThread(const char* name)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->name.roctxNameOsThread_fn(name);
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameHsaAgent(const char* name, const struct hsa_agent_s* agent)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->name.roctxNameHsaAgent_fn(name, agent);
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameHipDevice(const char* name, int device_id)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->name.roctxNameHipDevice_fn(name, device_id);
|
||||
}
|
||||
|
||||
int
|
||||
roctxNameHipStream(const char* name, const struct ihipStream_t* stream)
|
||||
{
|
||||
return ::rocprofiler::roctx::get_table()->name.roctxNameHipStream_fn(name, stream);
|
||||
}
|
||||
|
||||
ROCTX_EXTERN_C_FINI
|
||||
Reference in New Issue
Block a user