rocm-systems/source/include/rocprofiler-sdk/internal_threading.h

// 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.

#pragma once

#include <rocprofiler-sdk/defines.h>
#include <rocprofiler-sdk/fwd.h>

ROCPROFILER_EXTERN_C_INIT

/**
 * @defgroup INTERNAL_THREADING Internal Thread Handling
 * @brief Callbacks before and after threads created internally by libraries
 *
 * @{
 * @example api_buffered_tracing/client.cpp
 * Example demonstrating @ref BUFFER_TRACING_SERVICE that includes usage of
 * ::rocprofiler_at_internal_thread_create, ::rocprofiler_create_callback_thread, and
 * ::rocprofiler_assign_callback_thread.
 */

/**
 * @brief (experimental) Callback type before and after internal thread creation. @see
 * rocprofiler_at_internal_thread_create
 *
 */
ROCPROFILER_SDK_EXPERIMENTAL
typedef void (*rocprofiler_internal_thread_library_cb_t)(rocprofiler_runtime_library_t, void*);

/**
 * @brief (experimental) Invoke this function to receive callbacks before and after the creation of
 * an internal thread by a library which as invoked on the thread which is creating the internal
 * thread(s).
 *
 * Use the ::rocprofiler_runtime_library_t enumeration for specifying which libraries you want
 * callbacks before and after the library creates an internal thread. These callbacks will be
 * invoked on the thread that is about to create the new thread (not on the newly created thread).
 * In thread-aware tools that wrap pthread_create, this can be used to disable the wrapper before
 * the pthread_create invocation and re-enable the wrapper afterwards. In many cases, tools will
 * want to ignore the thread(s) created by rocprofiler since these threads do not exist in the
 * normal application execution, whereas the internal threads for HSA, HIP, etc. are created in
 * normal application execution; however, the HIP, HSA, etc. internal threads are typically
 * background threads which just monitor kernel completion and are unlikely to contribute to any
 * performance issues. Please note that the postcreate callback is guaranteed to be invoked after
 * the underlying system call to create a new thread but it does not guarantee that the new thread
 * has been started. Please note, that once these callbacks are registered, they cannot be removed
 * so the caller is responsible for ignoring these callbacks if they want to ignore them beyond a
 * certain point in the application.
 *
 * @param [in] precreate Callback invoked immediately before a new internal thread is created
 * @param [in] postcreate Callback invoked immediately after a new internal thread is created
 * @param [in] libs Bitwise-or of libraries, e.g. `ROCPROFILER_LIBRARY | ROCPROFILER_MARKER_LIBRARY`
 * means the callbacks will be invoked whenever rocprofiler and/or the marker library create
 * internal threads but not when the HSA or HIP libraries create internal threads.
 * @param [in] data Data shared between callbacks
 * @return ::rocprofiler_status_t
 * @retval ::ROCPROFILER_STATUS_SUCCESS There are currently no conditions which result in any other
 * value, even if internal threads have already been created
 */
ROCPROFILER_SDK_EXPERIMENTAL
rocprofiler_status_t
rocprofiler_at_internal_thread_create(rocprofiler_internal_thread_library_cb_t precreate,
                                      rocprofiler_internal_thread_library_cb_t postcreate,
                                      int                                      libs,
                                      void* data) ROCPROFILER_API;

/**
 * @brief (experimental) opaque handle to an internal thread identifier which delivers callbacks for
 * buffers
 * @see rocprofiler_create_callback_thread
 */
typedef struct ROCPROFILER_SDK_EXPERIMENTAL rocprofiler_callback_thread_t
{
    uint64_t handle;
} rocprofiler_callback_thread_t;

/**
 * @brief (experimental) Create a handle to a unique thread (created by rocprofiler) which, when
 * associated with a particular buffer, will guarantee those buffered results always get delivered
 * on the same thread. This is useful to prevent/control thread-safety issues and/or enable
 * multithreaded processing of buffers with non-overlapping data
 *
 * @param [in] cb_thread_id User-provided pointer to a ::rocprofiler_callback_thread_t
 * @return ::rocprofiler_status_t
 * @retval ::ROCPROFILER_STATUS_SUCCESS Successful thread creation
 * @retval ::ROCPROFILER_STATUS_ERROR_CONFIGURATION_LOCKED Thread creation is no longer available
 * post-initialization
 * @retval ::ROCPROFILER_STATUS_ERROR Failed to create thread
 */
ROCPROFILER_SDK_EXPERIMENTAL
rocprofiler_status_t
rocprofiler_create_callback_thread(rocprofiler_callback_thread_t* cb_thread_id) ROCPROFILER_API
    ROCPROFILER_NONNULL(1);

/**
 * @brief (experimental) By default, all buffered results are delivered on the same thread. Using
 * ::rocprofiler_create_callback_thread, one or more buffers can be assigned to deliever their
 * results on a unique, dedicated thread.
 *
 * @param [in] buffer_id Buffer identifier
 * @param [in] cb_thread_id Callback thread identifier via ::rocprofiler_create_callback_thread
 * @return ::rocprofiler_status_t
 * @retval ::ROCPROFILER_STATUS_SUCCESS Successful assignment of the delivery thread for the given
 * buffer
 * @retval ::ROCPROFILER_STATUS_ERROR_CONFIGURATION_LOCKED Thread assignment is no longer available
 * post-initialization
 * @retval ::ROCPROFILER_STATUS_ERROR_THREAD_NOT_FOUND Thread identifier did not match any of the
 * threads created by rocprofiler
 * @retval ::ROCPROFILER_STATUS_ERROR_BUFFER_NOT_FOUND Buffer identifier did not match any of the
 * buffers registered with rocprofiler
 */
ROCPROFILER_SDK_EXPERIMENTAL
rocprofiler_status_t
rocprofiler_assign_callback_thread(rocprofiler_buffer_id_t       buffer_id,
                                   rocprofiler_callback_thread_t cb_thread_id) ROCPROFILER_API;

/** @} */

ROCPROFILER_EXTERN_C_FINI