AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge "Added roctxRangeStart and roctxRangeEnd" into amd-master

Browse Source
This commit is contained in:
Rachida Kebichi
2020-08-12 10:32:29 -04:00
committed by Gerrit Code Review
parent f027cf1e22 da17a158f8
commit d3406bdb7b
5 changed files with 62 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/roctx/roctx.cpp
+28
View File
@@ -106,6 +106,7 @@ extern cb_table_t cb_table;
// Logger instantiation
roctracer::util::Logger::mutex_t roctracer::util::Logger::mutex_;
std::atomic<roctracer::util::Logger*> roctracer::util::Logger::instance_{};
std::atomic<int> roctx_range_counter(0);
///////////////////////////////////////////////////////////////////////////////////////////////////
// Public library methods
@@ -165,6 +166,33 @@ PUBLIC_API int roctxRangePop() {
API_METHOD_CATCH(-1)
}
PUBLIC_API roctx_range_id_t roctxRangeStartA(const char* message) {
API_METHOD_PREFIX
roctx_range_counter++;
roctx_api_data_t api_data{};
api_data.args.roctxRangeStartA.message = strdup(message);
api_data.args.roctxRangeStartA.id = roctx_range_counter;
activity_rtapi_callback_t api_callback_fun = NULL;
void* api_callback_arg = NULL;
roctx::cb_table.get(ROCTX_API_ID_roctxRangeStartA, &api_callback_fun, &api_callback_arg);
if (api_callback_fun) api_callback_fun(ACTIVITY_DOMAIN_ROCTX, ROCTX_API_ID_roctxRangeStartA, &api_data, api_callback_arg);
return roctx_range_counter;
API_METHOD_CATCH(-1);
}
PUBLIC_API void roctxRangeStop(roctx_range_id_t rangeId) {
API_METHOD_PREFIX
roctx_api_data_t api_data{};
api_data.args.roctxRangeStop.id = rangeId;
activity_rtapi_callback_t api_callback_fun = NULL;
void* api_callback_arg = NULL;
roctx::cb_table.get(ROCTX_API_ID_roctxRangeStop, &api_callback_fun, &api_callback_arg);
if (api_callback_fun) api_callback_fun(ACTIVITY_DOMAIN_ROCTX, ROCTX_API_ID_roctxRangeStop, &api_data, api_callback_arg);
API_METHOD_SUFFIX_NRET
}
PUBLIC_API void RangeStackIterate(roctx_range_iterate_cb_t callback, void* arg) {
for (const auto& entry : *roctx::thread_map) {
const auto tid = entry.first;