267954fdd3
* Update include/rocprofiler
- remove rocprofiler_external_correlation_id_t
- redefine rocprofiler_correlation_id_t to include internal id and external user data
- associate rocprofiler_push_external_correlation_id and rocprofiler_pop_external_correlation_id with a context
* Update include/rocprofiler/rocprofiler.h
- rocprofiler_get_status_name
- rocprofiler_get_status_string
* Update lib/rocprofiler/rocprofiler.cpp
- implement rocprofiler_get_status_name and rocprofiler_get_status_string
* Update lib/rocprofiler/tests/status.cpp
- unit test for status string and name
* Update lib/rocprofiler/tests/registration.cpp
- update to new rocprofiler_correlation_id_t
* Update samples
- update to new rocprofiler_correlation_id_t
* Add lib/rocprofiler/external_correlation.cpp
- placeholder for external correlation push/pop
* Update lib/rocprofiler/hsa/agent_cache.cpp
- slight tweak to when HSA_AMD_AGENT_INFO_NEAREST_CPU is defined
* Update context implementation and hsa.cpp
- get_active_contexts is array of const context pointers
- update hsa_api_impl<Idx>::functor to new rocprofiler_correlation_id_t
* Update include/rocprofiler/fwd.h
- add ROCPROFILER_STATUS_ERROR_INVALID_ARGUMENT
- reorder enum for consistency
* Update include/rocprofiler/external_correlation.h
- doxygen comments
- thread id parameter
* Update include/rocprofiler/rocprofiler.h
- add rocprofiler_get_thread_id function (needed for external corr id)
* Update lib/common/synchronized.hpp
- explicit LockedType
- define all copy/move ctor and assignment
- update rlock/wlock/ulock to support arguments and return values
- Support additional template parameter for special case of synchronized instance which is the mapped type of a sychronized map
* Update lib/rocprofiler/external_correlation.cpp
- implement rocprofiler_{push,pop}_external_correlation_id
* Update lib/rocprofiler/CMakeLists.txt
- external_correlation.hpp
* Update lib/rocprofiler/rocprofiler.cpp
- status string for ROCPROFILER_STATUS_ERROR_INVALID_ARGUMENT
- implement rocprofiler_get_thread_id
* Update lib/rocprofiler/tests (external correlation)
- add external_correlation unit tests
* Update include/rocprofiler/callback_tracing.h
- doxygen comments
- callback invoked in callback tracing has user_data pointer passed to it
* Update samples/api_callback_tracing/client.cpp
- add rocprofiler_user_data_t* to tool_tracing_callback
* Update lib/rocprofiler/tests/registration.cpp
- add rocprofiler_user_data_t* to tool_tracing_callback
* Update lib/rocprofiler/context/context.{hpp,cpp}
- update correlation_tracing_service
- external_correlation instance
- rename get_unique_record_id to get_unique_internal_id
* Update lib/tests/common/demangling.cpp
- tweak mangled definitions due to changing function get_unique_record_id to get_unique_internal_id
* Update lib/rocprofiler/hsa/hsa.cpp
- handle updates to external correlation id
- handle updates to callback signature in callback tracing
* Update CMakeLists.txt
- CMAKE_BUILD_TYPE=Coverage defines CODECOV=1
* Update samples/api_callback_tracing/client.cpp
[ROCm/rocprofiler-sdk commit: 6a3f79e626]
543 linhas
22 KiB
C++
543 linhas
22 KiB
C++
// MIT License
|
|
//
|
|
// Copyright (c) 2023 ROCm Developer Tools
|
|
//
|
|
// 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 <rocprofiler/registration.h>
|
|
#include <rocprofiler/rocprofiler.h>
|
|
|
|
#include "lib/common/environment.hpp"
|
|
#include "lib/common/units.hpp"
|
|
#include "lib/common/utility.hpp"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <dlfcn.h>
|
|
#include <pthread.h>
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
#include <filesystem>
|
|
#include <iostream>
|
|
#include <random>
|
|
#include <sstream>
|
|
#include <string_view>
|
|
#include <typeinfo>
|
|
#include <vector>
|
|
|
|
#define ROCPROFILER_CALL(ARG, MSG) \
|
|
{ \
|
|
auto _status = (ARG); \
|
|
EXPECT_EQ(_status, ROCPROFILER_STATUS_SUCCESS) << MSG << " :: " << #ARG; \
|
|
}
|
|
|
|
namespace
|
|
{
|
|
struct callback_data
|
|
{
|
|
rocprofiler_client_id_t* client_id = nullptr;
|
|
rocprofiler_client_finalize_t client_fini_func = nullptr;
|
|
rocprofiler_context_id_t client_ctx = {};
|
|
rocprofiler_buffer_id_t client_buffer = {};
|
|
rocprofiler_callback_thread_t client_thread = {};
|
|
uint64_t client_workflow_count = {};
|
|
uint64_t client_callback_count = {};
|
|
int64_t current_depth = 0;
|
|
int64_t max_depth = 0;
|
|
};
|
|
|
|
struct agent_data
|
|
{
|
|
uint64_t agent_count = 0;
|
|
std::vector<hsa_device_type_t> agents = {};
|
|
};
|
|
|
|
void
|
|
tool_tracing_callback(rocprofiler_callback_tracing_record_t record,
|
|
rocprofiler_user_data_t*,
|
|
void* client_data)
|
|
{
|
|
using name_map_t = std::unordered_map<rocprofiler_service_callback_tracing_kind_t,
|
|
std::unordered_map<uint32_t, const char*>>;
|
|
|
|
auto* cb_data = static_cast<callback_data*>(client_data);
|
|
|
|
static auto name_map = []() {
|
|
auto outer_cb = [](rocprofiler_service_callback_tracing_kind_t kind_v,
|
|
const char* /*kind_name*/,
|
|
void* data_v) {
|
|
auto inner_cb = [](rocprofiler_service_callback_tracing_kind_t kind,
|
|
uint32_t operation,
|
|
const char* operation_name,
|
|
void* data) {
|
|
auto& mdata = *static_cast<name_map_t*>(data);
|
|
mdata[kind][operation] = operation_name;
|
|
return 0;
|
|
};
|
|
|
|
auto& mdata_v = *static_cast<name_map_t*>(data_v);
|
|
mdata_v.emplace(kind_v, std::unordered_map<uint32_t, const char*>{});
|
|
|
|
rocprofiler_iterate_callback_tracing_kind_operation_names(kind_v, inner_cb, data_v);
|
|
return 0;
|
|
};
|
|
|
|
auto tmp = name_map_t{};
|
|
rocprofiler_iterate_callback_tracing_kind_names(outer_cb, static_cast<void*>(&tmp));
|
|
|
|
EXPECT_EQ(tmp.size(), ROCPROFILER_SERVICE_CALLBACK_TRACING_LAST - 1);
|
|
EXPECT_EQ(tmp.at(ROCPROFILER_SERVICE_CALLBACK_TRACING_HSA_API).size(),
|
|
ROCPROFILER_HSA_API_ID_LAST);
|
|
|
|
return tmp;
|
|
}();
|
|
|
|
std::cout << "[" << __FILE__ << ":" << __LINE__ << "] "
|
|
<< name_map[record.kind][record.operation] << "\n"
|
|
<< std::flush;
|
|
|
|
cb_data->client_callback_count++;
|
|
if(record.phase == ROCPROFILER_SERVICE_CALLBACK_PHASE_ENTER)
|
|
{
|
|
cb_data->current_depth++;
|
|
}
|
|
else if(record.phase == ROCPROFILER_SERVICE_CALLBACK_PHASE_EXIT)
|
|
{
|
|
cb_data->max_depth = std::max(cb_data->current_depth, cb_data->max_depth);
|
|
cb_data->current_depth--;
|
|
}
|
|
else
|
|
{
|
|
GTEST_FAIL() << "unsupported callback tracing phase " << record.phase;
|
|
}
|
|
|
|
struct info_data
|
|
{
|
|
uint64_t num_args = 0;
|
|
std::stringstream arg_ss = {};
|
|
} info_data_v;
|
|
|
|
auto info_data_cb = [](rocprofiler_service_callback_tracing_kind_t,
|
|
uint32_t,
|
|
uint32_t arg_num,
|
|
const char* arg_name,
|
|
const char* arg_value_str,
|
|
const void* const arg_value_addr,
|
|
void* data) -> int {
|
|
auto& info = *static_cast<info_data*>(data);
|
|
info.arg_ss << ((arg_num == 0) ? "(" : ", ");
|
|
info.arg_ss << arg_num << ": " << arg_name << "=" << arg_value_str;
|
|
EXPECT_NE(arg_name, nullptr);
|
|
EXPECT_NE(arg_value_str, nullptr);
|
|
EXPECT_NE(arg_value_addr, nullptr);
|
|
EXPECT_EQ(arg_num, info.num_args);
|
|
info.num_args++;
|
|
return 0;
|
|
};
|
|
|
|
ROCPROFILER_CALL(rocprofiler_iterate_callback_tracing_operation_args(
|
|
record, info_data_cb, static_cast<void*>(&info_data_v)),
|
|
"Failure iterating trace operation args");
|
|
if(record.kind == ROCPROFILER_SERVICE_CALLBACK_TRACING_HSA_API &&
|
|
!(record.operation == ROCPROFILER_HSA_API_ID_hsa_init ||
|
|
record.operation == ROCPROFILER_HSA_API_ID_hsa_shut_down))
|
|
{
|
|
EXPECT_GT(info_data_v.num_args, 0)
|
|
<< name_map[record.kind][record.operation] << info_data_v.arg_ss.str();
|
|
}
|
|
}
|
|
|
|
void
|
|
tool_tracing_buffered(rocprofiler_context_id_t context,
|
|
rocprofiler_buffer_id_t buffer_id,
|
|
rocprofiler_record_header_t** headers,
|
|
size_t num_headers,
|
|
void* buffer_data,
|
|
uint64_t drop_count)
|
|
{
|
|
using name_map_t = std::unordered_map<rocprofiler_service_buffer_tracing_kind_t,
|
|
std::unordered_map<uint32_t, const char*>>;
|
|
|
|
std::cout << __FUNCTION__ << "...\n" << std::endl;
|
|
auto* cb_data = static_cast<callback_data*>(buffer_data);
|
|
|
|
static auto name_map = []() {
|
|
auto tmp = name_map_t{};
|
|
//
|
|
static auto tracing_kind_names_cb = [](rocprofiler_service_buffer_tracing_kind_t kind,
|
|
const char* /*kind_name*/,
|
|
void* data) {
|
|
auto tracing_operation_names_cb = [](rocprofiler_service_buffer_tracing_kind_t kindv,
|
|
uint32_t operation,
|
|
const char* operation_name,
|
|
void* data_v) {
|
|
(*static_cast<name_map_t*>(data_v))[kindv][operation] = operation_name;
|
|
return 0;
|
|
};
|
|
|
|
rocprofiler_iterate_buffer_tracing_kind_operation_names(
|
|
kind, tracing_operation_names_cb, data);
|
|
|
|
return 0;
|
|
};
|
|
|
|
rocprofiler_iterate_buffer_tracing_kind_names(tracing_kind_names_cb,
|
|
static_cast<void*>(&tmp));
|
|
return tmp;
|
|
}();
|
|
|
|
auto v_records = std::vector<rocprofiler_buffer_tracing_hsa_api_record_t*>{};
|
|
v_records.reserve(num_headers);
|
|
|
|
for(size_t i = 0; i < num_headers; ++i)
|
|
{
|
|
auto* header = headers[i];
|
|
|
|
ASSERT_TRUE(header != nullptr);
|
|
auto hash = rocprofiler_record_header_compute_hash(header->category, header->kind);
|
|
EXPECT_EQ(header->hash, hash);
|
|
EXPECT_TRUE(header->category == ROCPROFILER_BUFFER_CATEGORY_TRACING &&
|
|
header->kind == ROCPROFILER_SERVICE_BUFFER_TRACING_HSA_API);
|
|
|
|
v_records.emplace_back(
|
|
static_cast<rocprofiler_buffer_tracing_hsa_api_record_t*>(header->payload));
|
|
}
|
|
|
|
std::sort(v_records.begin(), v_records.end(), [](auto lhs, auto rhs) {
|
|
return (lhs->start_timestamp == rhs->start_timestamp)
|
|
? (lhs->end_timestamp < rhs->end_timestamp)
|
|
: (lhs->start_timestamp < rhs->start_timestamp);
|
|
});
|
|
|
|
for(auto* record : v_records)
|
|
{
|
|
auto info = std::stringstream{};
|
|
info << "tid=" << record->thread_id << ", context=" << context.handle
|
|
<< ", buffer_id=" << buffer_id.handle << ", cid=" << record->correlation_id.internal
|
|
<< ", kind=" << record->kind << ", operation=" << record->operation
|
|
<< ", drop_count=" << drop_count << ", start=" << record->start_timestamp
|
|
<< ", stop=" << record->end_timestamp;
|
|
|
|
static int64_t last_corr_id = -1;
|
|
auto corr_id = static_cast<int64_t>(record->correlation_id.internal);
|
|
|
|
std::cout << info.str() << "\n" << std::flush;
|
|
EXPECT_GE(context.handle, 0) << info.str();
|
|
EXPECT_GT(record->thread_id, 0) << info.str();
|
|
EXPECT_GT(record->kind, 0) << info.str();
|
|
EXPECT_GT(corr_id, last_corr_id) << info.str();
|
|
EXPECT_GT(record->start_timestamp, 0) << info.str();
|
|
EXPECT_GT(record->end_timestamp, 0) << info.str();
|
|
EXPECT_LE(record->start_timestamp, record->end_timestamp) << info.str();
|
|
|
|
cb_data->client_callback_count++;
|
|
last_corr_id = corr_id;
|
|
}
|
|
}
|
|
|
|
void
|
|
thread_precreate(rocprofiler_internal_thread_library_t /*lib*/, void* tool_data)
|
|
{
|
|
auto* cb_data = static_cast<callback_data*>(tool_data);
|
|
cb_data->client_workflow_count++;
|
|
}
|
|
|
|
void
|
|
thread_postcreate(rocprofiler_internal_thread_library_t /*lib*/, void* tool_data)
|
|
{
|
|
auto* cb_data = static_cast<callback_data*>(tool_data);
|
|
cb_data->client_workflow_count++;
|
|
}
|
|
} // namespace
|
|
|
|
TEST(rocprofiler_lib, registration_lambda_no_result)
|
|
{
|
|
static rocprofiler_configure_func_t rocp_init =
|
|
[](uint32_t version,
|
|
const char* runtime_version,
|
|
uint32_t prio,
|
|
rocprofiler_client_id_t* client_id) -> rocprofiler_tool_configure_result_t* {
|
|
auto expected_version = ROCPROFILER_VERSION;
|
|
EXPECT_EQ(expected_version, version);
|
|
EXPECT_EQ(std::string_view{runtime_version}, std::string_view{ROCPROFILER_VERSION_STRING});
|
|
EXPECT_EQ(prio, 0);
|
|
EXPECT_EQ(client_id->name, nullptr);
|
|
return nullptr;
|
|
};
|
|
|
|
auto ctx = rocprofiler_context_id_t{};
|
|
EXPECT_NE(rocprofiler_create_context(&ctx), ROCPROFILER_STATUS_SUCCESS);
|
|
EXPECT_EQ(rocprofiler_force_configure(rocp_init), ROCPROFILER_STATUS_SUCCESS);
|
|
EXPECT_NE(rocprofiler_create_context(&ctx), ROCPROFILER_STATUS_SUCCESS);
|
|
}
|
|
|
|
TEST(rocprofiler_lib, callback_registration_lambda_with_result)
|
|
{
|
|
using init_func_t = int (*)(rocprofiler_client_finalize_t, void*);
|
|
using fini_func_t = void (*)(void*);
|
|
|
|
using hsa_iterate_agents_cb_t = hsa_status_t (*)(hsa_agent_t, void*);
|
|
|
|
auto cmd_line = rocprofiler::common::read_command_line(getpid());
|
|
ASSERT_FALSE(cmd_line.empty());
|
|
|
|
static init_func_t tool_init = [](rocprofiler_client_finalize_t fini_func,
|
|
void* client_data) -> int {
|
|
auto* cb_data = static_cast<callback_data*>(client_data);
|
|
|
|
cb_data->client_workflow_count++;
|
|
cb_data->client_fini_func = fini_func;
|
|
|
|
ROCPROFILER_CALL(rocprofiler_create_context(&cb_data->client_ctx),
|
|
"failed to create context");
|
|
|
|
ROCPROFILER_CALL(rocprofiler_configure_callback_tracing_service(
|
|
cb_data->client_ctx,
|
|
ROCPROFILER_SERVICE_CALLBACK_TRACING_HSA_API,
|
|
nullptr,
|
|
0,
|
|
tool_tracing_callback,
|
|
client_data),
|
|
"callback tracing service failed to configure");
|
|
|
|
int valid_ctx = 0;
|
|
ROCPROFILER_CALL(rocprofiler_context_is_valid(cb_data->client_ctx, &valid_ctx),
|
|
"failure checking context validity");
|
|
|
|
EXPECT_EQ(valid_ctx, 1);
|
|
|
|
ROCPROFILER_CALL(rocprofiler_start_context(cb_data->client_ctx),
|
|
"rocprofiler context start failed");
|
|
|
|
// no errors
|
|
return 0;
|
|
};
|
|
|
|
static fini_func_t tool_fini = [](void* client_data) -> void {
|
|
auto* cb_data = static_cast<callback_data*>(client_data);
|
|
ROCPROFILER_CALL(rocprofiler_stop_context(cb_data->client_ctx),
|
|
"rocprofiler context stop failed");
|
|
|
|
static_cast<callback_data*>(client_data)->client_workflow_count++;
|
|
};
|
|
|
|
static auto cb_data = callback_data{};
|
|
|
|
static auto cfg_result =
|
|
rocprofiler_tool_configure_result_t{sizeof(rocprofiler_tool_configure_result_t),
|
|
tool_init,
|
|
tool_fini,
|
|
static_cast<void*>(&cb_data)};
|
|
|
|
static rocprofiler_configure_func_t rocp_init =
|
|
[](uint32_t version,
|
|
const char* runtime_version,
|
|
uint32_t prio,
|
|
rocprofiler_client_id_t* client_id) -> rocprofiler_tool_configure_result_t* {
|
|
auto expected_version = ROCPROFILER_VERSION;
|
|
EXPECT_EQ(expected_version, version);
|
|
EXPECT_EQ(std::string_view{runtime_version}, std::string_view{ROCPROFILER_VERSION_STRING});
|
|
EXPECT_EQ(prio, 0);
|
|
EXPECT_EQ(client_id->name, nullptr);
|
|
cb_data.client_id = client_id;
|
|
cb_data.client_id->name = ::testing::UnitTest::GetInstance()->current_test_info()->name();
|
|
return &cfg_result;
|
|
};
|
|
|
|
auto ctx = rocprofiler_context_id_t{};
|
|
EXPECT_NE(rocprofiler_create_context(&ctx), ROCPROFILER_STATUS_SUCCESS);
|
|
EXPECT_EQ(rocprofiler_force_configure(rocp_init), ROCPROFILER_STATUS_SUCCESS);
|
|
EXPECT_NE(rocprofiler_create_context(&ctx), ROCPROFILER_STATUS_SUCCESS);
|
|
|
|
hsa_iterate_agents_cb_t agent_cb = [](hsa_agent_t agent, void* data) {
|
|
static_cast<agent_data*>(data)->agent_count++;
|
|
|
|
auto status = HSA_STATUS_SUCCESS;
|
|
auto agent_type = hsa_device_type_t{};
|
|
if((status = hsa_agent_get_info(agent, HSA_AGENT_INFO_DEVICE, &agent_type)) ==
|
|
HSA_STATUS_SUCCESS)
|
|
static_cast<agent_data*>(data)->agents.emplace_back(agent_type);
|
|
|
|
return status;
|
|
};
|
|
|
|
auto _agent_data = agent_data{};
|
|
hsa_init();
|
|
hsa_status_t itr_status = hsa_iterate_agents(agent_cb, static_cast<void*>(&_agent_data));
|
|
|
|
EXPECT_EQ(itr_status, HSA_STATUS_SUCCESS);
|
|
EXPECT_GT(_agent_data.agent_count, 0);
|
|
EXPECT_EQ(_agent_data.agent_count, _agent_data.agents.size());
|
|
|
|
ASSERT_NE(cb_data.client_id, nullptr);
|
|
ASSERT_NE(cb_data.client_fini_func, nullptr);
|
|
|
|
cb_data.client_fini_func(*cb_data.client_id);
|
|
|
|
// expected callback count is two for hsa_iterate_agents and two callbacks for
|
|
// hsa_agent_get_info for each agent.
|
|
uint64_t expected_cb_count = 2 + (2 * _agent_data.agent_count);
|
|
|
|
EXPECT_EQ(cb_data.client_workflow_count, 2);
|
|
EXPECT_EQ(cb_data.client_callback_count, expected_cb_count);
|
|
EXPECT_EQ(cb_data.current_depth, 0);
|
|
EXPECT_EQ(cb_data.max_depth, 2);
|
|
}
|
|
|
|
TEST(rocprofiler_lib, buffer_registration_lambda_with_result)
|
|
{
|
|
using init_func_t = int (*)(rocprofiler_client_finalize_t, void*);
|
|
using fini_func_t = void (*)(void*);
|
|
|
|
using hsa_iterate_agents_cb_t = hsa_status_t (*)(hsa_agent_t, void*);
|
|
|
|
auto cmd_line = rocprofiler::common::read_command_line(getpid());
|
|
ASSERT_FALSE(cmd_line.empty());
|
|
|
|
static init_func_t tool_init = [](rocprofiler_client_finalize_t fini_func,
|
|
void* client_data) -> int {
|
|
auto* cb_data = static_cast<callback_data*>(client_data);
|
|
|
|
cb_data->client_workflow_count++;
|
|
cb_data->client_fini_func = fini_func;
|
|
|
|
ROCPROFILER_CALL(rocprofiler_create_context(&cb_data->client_ctx),
|
|
"failed to create context");
|
|
|
|
ROCPROFILER_CALL(rocprofiler_create_buffer(cb_data->client_ctx,
|
|
4096,
|
|
2048,
|
|
ROCPROFILER_BUFFER_POLICY_LOSSLESS,
|
|
tool_tracing_buffered,
|
|
client_data,
|
|
&cb_data->client_buffer),
|
|
"buffer creation failed");
|
|
|
|
ROCPROFILER_CALL(
|
|
rocprofiler_configure_buffer_tracing_service(cb_data->client_ctx,
|
|
ROCPROFILER_SERVICE_BUFFER_TRACING_HSA_API,
|
|
nullptr,
|
|
0,
|
|
cb_data->client_buffer),
|
|
"buffer tracing service failed to configure");
|
|
|
|
ROCPROFILER_CALL(rocprofiler_create_callback_thread(&cb_data->client_thread),
|
|
"failure creating callback thread");
|
|
|
|
ROCPROFILER_CALL(
|
|
rocprofiler_assign_callback_thread(cb_data->client_buffer, cb_data->client_thread),
|
|
"failed to assign thread for buffer");
|
|
|
|
int valid_ctx = 0;
|
|
ROCPROFILER_CALL(rocprofiler_context_is_valid(cb_data->client_ctx, &valid_ctx),
|
|
"failure checking context validity");
|
|
|
|
EXPECT_EQ(valid_ctx, 1);
|
|
|
|
ROCPROFILER_CALL(rocprofiler_start_context(cb_data->client_ctx),
|
|
"rocprofiler context start failed");
|
|
|
|
// no errors
|
|
return 0;
|
|
};
|
|
|
|
static fini_func_t tool_fini = [](void* client_data) -> void {
|
|
auto* cb_data = static_cast<callback_data*>(client_data);
|
|
ROCPROFILER_CALL(rocprofiler_stop_context(cb_data->client_ctx),
|
|
"rocprofiler context stop failed");
|
|
|
|
static_cast<callback_data*>(client_data)->client_workflow_count++;
|
|
};
|
|
|
|
static auto cb_data = callback_data{};
|
|
|
|
static auto cfg_result =
|
|
rocprofiler_tool_configure_result_t{sizeof(rocprofiler_tool_configure_result_t),
|
|
tool_init,
|
|
tool_fini,
|
|
static_cast<void*>(&cb_data)};
|
|
|
|
static rocprofiler_configure_func_t rocp_init =
|
|
[](uint32_t version,
|
|
const char* runtime_version,
|
|
uint32_t prio,
|
|
rocprofiler_client_id_t* client_id) -> rocprofiler_tool_configure_result_t* {
|
|
auto expected_version = ROCPROFILER_VERSION;
|
|
EXPECT_EQ(expected_version, version);
|
|
EXPECT_EQ(std::string_view{runtime_version}, std::string_view{ROCPROFILER_VERSION_STRING});
|
|
EXPECT_EQ(prio, 0);
|
|
EXPECT_EQ(client_id->name, nullptr);
|
|
cb_data.client_id = client_id;
|
|
cb_data.client_id->name = ::testing::UnitTest::GetInstance()->current_test_info()->name();
|
|
|
|
ROCPROFILER_CALL(rocprofiler_at_internal_thread_create(thread_precreate,
|
|
thread_postcreate,
|
|
ROCPROFILER_LIBRARY,
|
|
static_cast<void*>(&cb_data)),
|
|
"failed to register for thread creation notifications");
|
|
|
|
return &cfg_result;
|
|
};
|
|
|
|
auto ctx = rocprofiler_context_id_t{};
|
|
EXPECT_NE(rocprofiler_create_context(&ctx), ROCPROFILER_STATUS_SUCCESS);
|
|
EXPECT_EQ(rocprofiler_force_configure(rocp_init), ROCPROFILER_STATUS_SUCCESS);
|
|
EXPECT_NE(rocprofiler_create_context(&ctx), ROCPROFILER_STATUS_SUCCESS);
|
|
|
|
hsa_iterate_agents_cb_t agent_cb = [](hsa_agent_t agent, void* data) {
|
|
static_cast<agent_data*>(data)->agent_count++;
|
|
|
|
auto status = HSA_STATUS_SUCCESS;
|
|
auto agent_type = hsa_device_type_t{};
|
|
if((status = hsa_agent_get_info(agent, HSA_AGENT_INFO_DEVICE, &agent_type)) ==
|
|
HSA_STATUS_SUCCESS)
|
|
static_cast<agent_data*>(data)->agents.emplace_back(agent_type);
|
|
|
|
return status;
|
|
};
|
|
|
|
auto _agent_data = agent_data{};
|
|
hsa_init();
|
|
hsa_status_t itr_status = hsa_iterate_agents(agent_cb, static_cast<void*>(&_agent_data));
|
|
|
|
EXPECT_EQ(itr_status, HSA_STATUS_SUCCESS);
|
|
EXPECT_GT(_agent_data.agent_count, 0);
|
|
EXPECT_EQ(_agent_data.agent_count, _agent_data.agents.size());
|
|
|
|
ASSERT_NE(cb_data.client_id, nullptr);
|
|
ASSERT_NE(cb_data.client_fini_func, nullptr);
|
|
|
|
EXPECT_EQ(rocprofiler_flush_buffer(cb_data.client_buffer), ROCPROFILER_STATUS_SUCCESS);
|
|
|
|
cb_data.client_fini_func(*cb_data.client_id);
|
|
|
|
// expected callback count is two for hsa_iterate_agents and two callbacks for
|
|
// hsa_agent_get_info for each agent.
|
|
uint64_t expected_cb_count = 1 + _agent_data.agent_count;
|
|
// expect the tool init, tool fini, and two calls to thread_precreate and thread_postcreate each
|
|
// (the main thread and the assigned thread for the buffer)
|
|
uint64_t expected_workflow_count = 6;
|
|
|
|
EXPECT_EQ(cb_data.client_workflow_count, expected_workflow_count);
|
|
EXPECT_EQ(cb_data.client_callback_count, expected_cb_count);
|
|
EXPECT_GT(cb_data.client_thread.handle, 0);
|
|
EXPECT_EQ(cb_data.current_depth, 0);
|
|
EXPECT_EQ(cb_data.max_depth, 0);
|
|
}
|