TheRock/rocm-libraries/dnn-providers/hip-kernel-provider/include/hip_kernel_provider_common/HipDeviceUtils.hpp

// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier:  MIT

#pragma once

#include <hip/hip_runtime.h>
#include <stdexcept>
#include <string>

namespace hip_kernel_provider_common
{

/**
 * @brief Gets the device properties from a HIP stream.
 *
 * @param stream The HIP stream to query the device from
 * @return The device properties
 * @throws std::runtime_error if the device query fails
 */
inline hipDeviceProp_t getDeviceProperties(hipStream_t stream)
{
    hipDevice_t deviceId = -1;
    hipDeviceProp_t props;
    auto status = hipStreamGetDevice(stream, &deviceId);
    if(status != hipSuccess)
    {
        throw std::runtime_error("hipStreamGetDevice failed with error code: "
                                 + std::to_string(status));
    }

    status = hipGetDeviceProperties(&props, deviceId);
    if(status != hipSuccess)
    {
        throw std::runtime_error("hipGetDeviceProperties failed with error code: "
                                 + std::to_string(status));
    }

    return props;
}

/**
 * @brief Gets the device architecture string (e.g., "gfx942") from a HIP stream.
 *
 * @param stream The HIP stream to query the device from
 * @return The device architecture string without the colon suffix
 * @throws std::runtime_error if the device query fails
 */
inline std::string getDeviceString(hipStream_t stream)
{
    auto props = getDeviceProperties(stream);

    const std::string archStr(props.gcnArchName);

    return archStr.substr(0, archStr.find(':'));
}

} // namespace hip_kernel_provider_common