Jonathan R. Madsen
cf5e4b4b1b
* Add external/cereal submodule - used for integration testing * Update lib/common/container/small_vector.hpp - documentation notes * Update tests/apps - update transpose app (fix build) - add reproducible-runtime app * Update include/rocprofiler/fwd.h - rocprofiler_service_callback_phase_t -> rocprofiler_callback_phase_t * Update PTL submodule - fix for task group: submitting tasks from different thread * Update lib/rocprofiler/hsa/queue.cpp - CHECK_NOTNULL(_buffer) * Update lib/rocprofiler/hsa/hsa.cpp - use buffer::get_buffer instead of manually looking for buffer * Update lib/rocprofiler/internal_threading.cpp - use buffer::get_buffer instead of manually looking for buffer * Update lib/rocprofiler/buffer.cpp - offset the buffer id - properly handle rocprofiler_create_buffer reusing rocprofiler_buffer_id_t on a different context * Update tests - kernel tracing library for integration testing * Add cereal submodule * Update lib/rocprofiler/registration.* - OnUnload - Support ROCP_TOOL_LIBRARIES for python usage - improve finalize function - remove calling hsa_shut_down in finalize function * Update lib/rocprofiler/buffer.* - allocate_buffer sets the buffer id value - expose (internally) is_valid_buffer_id - update test * Update tests/kernel-tracing - installation - better organization of JSON groups - improved messaging * Update lib/rocprofiler/registration.cpp - add workaround for hsa-runtime supporting rocprofiler-register * Update tests/kernel-tracing/kernel-tracing.cpp - fix memory leaks * cereal support for minimal JSON - update cereal submodule to rocprofiler branch - change REPO_BRANCH in rocprofiler_checkout_git_submodule for cereal - update tests/kernel-tracing/kernel-tracing.cpp - use minimal json - slight tweak putting giving contexts name in storing name + context pointer pair in map * Update tests/kernel-tracing/kernel-tracing.cpp - support runtime selection of contexts via KERNEL_TRACING_CONTEXTS environment variable * Update tests - tests/CMakeLists.txt - find_package(Python3 REQUIRED) - tests/kernel-tracing - pytest validation * Update CI workflow - install pytest - add checks for test labels * Update scripts/run-ci.py - change --coverage options - replace 'unittests' with 'tests' - replace test label regex '-L unittests' with '-L tests' * Update requirements.txt - this is now an empty file since none of the packages are required for this repo
79 satır
3.0 KiB
C++
79 satır
3.0 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.
|
|
|
|
#include "lib/rocprofiler/buffer.hpp"
|
|
#include "lib/common/units.hpp"
|
|
|
|
#include <rocprofiler/buffer.h>
|
|
#include <rocprofiler/fwd.h>
|
|
#include <rocprofiler/registration.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <pthread.h>
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
#include <random>
|
|
#include <typeinfo>
|
|
|
|
TEST(rocprofiler_lib, buffer)
|
|
{
|
|
namespace buffer = ::rocprofiler::buffer;
|
|
namespace common = ::rocprofiler::common;
|
|
|
|
ASSERT_EQ(buffer::get_buffers().size(), 0)
|
|
<< "no buffers should have been created at this point";
|
|
auto buffer_id = buffer::allocate_buffer();
|
|
|
|
EXPECT_TRUE(buffer_id) << "failed to allocate buffer";
|
|
EXPECT_GT(buffer_id->handle, 0);
|
|
EXPECT_TRUE(buffer::is_valid_buffer_id(*buffer_id)) << "id=" << buffer_id->handle;
|
|
ASSERT_EQ(buffer::get_buffers().size(), 1) << "incorrect number of buffers created";
|
|
|
|
// get pointer to buffer
|
|
auto* buffer_v = buffer::get_buffer(*buffer_id);
|
|
ASSERT_NE(buffer_v, nullptr) << "get_buffer returned a nullptr. id=" << buffer_id->handle;
|
|
EXPECT_EQ(buffer_v->buffer_id, buffer_id->handle);
|
|
|
|
buffer_v->watermark = common::units::get_page_size();
|
|
{
|
|
auto records = buffer_v->get_internal_buffer().get_record_headers();
|
|
EXPECT_EQ(records.size(), 0);
|
|
}
|
|
|
|
EXPECT_TRUE(buffer_v->get_internal_buffer().allocate(sizeof(rocprofiler_buffer_id_t)));
|
|
|
|
EXPECT_EQ(buffer_v->get_internal_buffer().capacity(), common::units::get_page_size());
|
|
|
|
auto data = *buffer_id;
|
|
buffer_v->emplace(1, 1, data);
|
|
|
|
auto records = buffer_v->get_internal_buffer().get_record_headers();
|
|
EXPECT_EQ(records.size(), 1);
|
|
|
|
auto flush_status = buffer::flush(*buffer_id, true);
|
|
EXPECT_EQ(flush_status, ROCPROFILER_STATUS_SUCCESS);
|
|
|
|
auto destroy_status = rocprofiler_destroy_buffer(*buffer_id);
|
|
EXPECT_EQ(destroy_status, ROCPROFILER_STATUS_SUCCESS);
|
|
}
|