Benjamin Welton
bb69467765
* Renamed agent profiling service to device counting service
Name more aptly represents what agent profiling did (device wide
counter collection). Conversion of existing user code can be
performed by the following find/sed command:
find . -type f -exec sed -i 's/rocprofiler_agent_profile_callback_t/rocprofiler_device_counting_service_callback_t/g; s/rocprofiler_configure_agent_profile_counting_service/rocprofiler_configure_device_counting_service/g; s/agent_profile.h/device_counting_service.h/g; s/rocprofiler_sample_agent_profile_counting_service/rocprofiler_sample_device_counting_service/g' {} +
* Converted dispatch profile to dispatch counting service
* Debug for functioal counters test
* Minor changes for CI
* Minor fix
* More fixes for CI
* Update evaluate_ast.cpp
---------
Co-authored-by: Benjamin Welton <ben@amd.com>
108 lines
4.6 KiB
C++
108 lines
4.6 KiB
C++
// MIT License
|
|
//
|
|
// Copyright (c) 2023 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 <rocprofiler-sdk/agent.h>
|
|
#include <rocprofiler-sdk/dispatch_counting_service.h>
|
|
#include <rocprofiler-sdk/fwd.h>
|
|
#include <optional>
|
|
|
|
#include "lib/common/synchronized.hpp"
|
|
#include "lib/rocprofiler-sdk/counters/controller.hpp"
|
|
#include "lib/rocprofiler-sdk/hsa/agent_cache.hpp"
|
|
#include "lib/rocprofiler-sdk/hsa/rocprofiler_packet.hpp"
|
|
|
|
namespace rocprofiler
|
|
{
|
|
namespace context
|
|
{
|
|
struct context;
|
|
}
|
|
namespace counters
|
|
{
|
|
// Internal counter struct that stores the state needed to handle an intercepted
|
|
// HSA kernel packet.
|
|
struct counter_callback_info
|
|
{
|
|
// User callback
|
|
rocprofiler_dispatch_counting_service_callback_t user_cb{nullptr};
|
|
// User id
|
|
void* callback_args{nullptr};
|
|
// Link to the context this is associated with
|
|
rocprofiler_context_id_t context{.handle = 0};
|
|
// Link to the internal context this is associated with
|
|
const context::context* internal_context{nullptr};
|
|
// HSA Queue ClientID. This is an ID we get when we insert a callback into the
|
|
// HSA queue interceptor. This ID can be used to disable the callback.
|
|
rocprofiler::hsa::ClientID queue_id{-1};
|
|
// Buffer to use for storing counter data. Used if callback is not set.
|
|
std::optional<rocprofiler_buffer_id_t> buffer;
|
|
|
|
rocprofiler_profile_counting_record_callback_t record_callback;
|
|
void* record_callback_args;
|
|
|
|
// Facilitates the return of an AQL Packet to the profile config that constructed it.
|
|
rocprofiler::common::Synchronized<
|
|
std::unordered_map<rocprofiler::hsa::AQLPacket*, std::shared_ptr<profile_config>>>
|
|
packet_return_map{};
|
|
|
|
static rocprofiler_status_t setup_profile_config(std::shared_ptr<profile_config>&);
|
|
|
|
rocprofiler_status_t get_packet(std::unique_ptr<rocprofiler::hsa::AQLPacket>&,
|
|
std::shared_ptr<profile_config>&);
|
|
};
|
|
|
|
uint64_t
|
|
create_counter_profile(std::shared_ptr<rocprofiler::counters::profile_config>&& config);
|
|
|
|
void
|
|
destroy_counter_profile(uint64_t id);
|
|
|
|
rocprofiler_status_t
|
|
configure_buffered_dispatch(rocprofiler_context_id_t context_id,
|
|
rocprofiler_buffer_id_t buffer,
|
|
rocprofiler_dispatch_counting_service_callback_t callback,
|
|
void* callback_args);
|
|
|
|
rocprofiler_status_t
|
|
configure_callback_dispatch(rocprofiler_context_id_t context_id,
|
|
rocprofiler_dispatch_counting_service_callback_t callback,
|
|
void* callback_data_args,
|
|
rocprofiler_profile_counting_record_callback_t record_callback,
|
|
void* record_callback_args);
|
|
|
|
rocprofiler_status_t
|
|
configure_agent_collection(rocprofiler_context_id_t context_id,
|
|
rocprofiler_buffer_id_t buffer_id,
|
|
rocprofiler_agent_id_t agent_id,
|
|
rocprofiler_device_counting_service_callback_t cb,
|
|
void* user_data);
|
|
|
|
void
|
|
start_context(const context::context*);
|
|
|
|
void
|
|
stop_context(const context::context*);
|
|
} // namespace counters
|
|
} // namespace rocprofiler
|