TheRock/rocm-systems/projects/rocprofiler-systems/source/lib/rocprof-sys/library/rocprofiler-sdk/rccl.hpp

// Copyright (c) Advanced Micro Devices, Inc.
// SPDX-License-Identifier: MIT

#pragma once

// Defines RCCL_API_TRACE_VERSION_PATCH. Must precede callback_tracing.h, which
// pulls in api_id.h where this macro gates the newer RCCL API ids (e.g.
// ncclAlltoAll). api_id.h from the SDK headers does not define it itself,
// so without this include the macro defaults to 0 and those members silently disappear.
#include <rocprofiler-sdk/rccl/details/api_trace.h>

#include <cstdint>
#include <rocprofiler-sdk/buffer_tracing.h>
#include <rocprofiler-sdk/callback_tracing.h>
#include <rocprofiler-sdk/fwd.h>
#include <rocprofiler-sdk/registration.h>

namespace rocprofsys
{
namespace rocprofiler_sdk
{
void
rccl_comm_data_initialize();

void
tool_tracing_callback_rccl(std::uint32_t                                 operation,
                           rocprofiler_callback_tracing_rccl_api_data_t* payload,
                           std::uint64_t begin_ts, std::uint64_t end_ts);

}  // namespace rocprofiler_sdk

}  // namespace rocprofsys