/* SPDX-FileCopyrightText: 2011-2022 Blender Foundation
 *
 * SPDX-License-Identifier: Apache-2.0 */

#ifdef WITH_HIP

#  include "device/hip/util.h"
#  include "device/hip/device_impl.h"

CCL_NAMESPACE_BEGIN

HIPContextScope::HIPContextScope(HIPDevice *device) : device(device)
{
  hip_device_assert(device, hipCtxPushCurrent(device->hipContext));
}

HIPContextScope::~HIPContextScope()
{
  hip_device_assert(device, hipCtxPopCurrent(nullptr));
}

#  ifndef WITH_HIP_DYNLOAD
const char *hipewErrorString(hipError_t result)
{
  /* We can only give error code here without major code duplication, that
   * should be enough since dynamic loading is only being disabled by folks
   * who knows what they're doing anyway.
   *
   * NOTE: Avoid call from several threads.
   */
  static string error;
  error = string_printf("%d", result);
  return error.c_str();
}

const char *hipewCompilerPath()
{
  return CYCLES_HIP_HIPCC_EXECUTABLE;
}

int hipewCompilerVersion()
{
  return (HIP_VERSION / 100) + (HIP_VERSION % 100 / 10);
}
#  endif /* !WITH_HIP_DYNLOAD */

bool hipSupportsDriver()
{
#  ifdef _WIN32
  /* This check is only necessary if we're using HIP SDK 6 or newer. */
  int hip_driver_version = 0;
  hipError_t result = hipDriverGetVersion(&hip_driver_version);
  if (result != hipSuccess) {
    LOG_WARNING << "Error getting driver version: " << hipewErrorString(result);
    return false;
  }

  LOG_TRACE << "Detected HIP driver version: " << hip_driver_version;

  if (hip_driver_version < 60140252) {
    /* Cycles crashes during rendering due to issues in older GPU drivers.
     * 60140252 corresponds to Adrenalin 24.6.1. */
    return false;
  }
#  endif

  return true;
}

CCL_NAMESPACE_END

#endif /* WITH_HIP */