// Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
// All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
//
// SPDX-License-Identifier: BSD-2-Clause
//
// This file is part of CEED:  http://github.com/ceed

#ifndef _ceed_common_hip_h
#define _ceed_common_hip_h

#include <ceed/ceed.h>
#include <ceed/backend.h>
#include <ceed/jit-source/hip/hip-types.h>
#include <hip/hip_runtime.h>
#if (HIP_VERSION >= 50200000)
#include <hipblas/hipblas.h>
#else
#include <hipblas.h>
#endif

#define QUOTE(...) #__VA_ARGS__

#define CeedChk_Hip(ceed, x) \
do { \
  hipError_t hip_result = x; \
  if (hip_result != hipSuccess) { \
    const char *msg = hipGetErrorName(hip_result); \
    return CeedError((ceed), CEED_ERROR_BACKEND, msg); \
  } \
} while (0)

#define CeedChk_Hipblas(ceed, x) \
do { \
  hipblasStatus_t hipblas_result = x; \
  if (hipblas_result != HIPBLAS_STATUS_SUCCESS) { \
    const char *msg = hipblasGetErrorName(hipblas_result); \
    return CeedError((ceed), CEED_ERROR_BACKEND, msg); \
   } \
} while (0)

#define CASE(name) case name: return #name
// LCOV_EXCL_START
CEED_UNUSED static const char *hipblasGetErrorName(hipblasStatus_t error) {
  switch (error) {
    CASE(HIPBLAS_STATUS_SUCCESS);
    CASE(HIPBLAS_STATUS_NOT_INITIALIZED);
    CASE(HIPBLAS_STATUS_ALLOC_FAILED);
    CASE(HIPBLAS_STATUS_INVALID_VALUE);
    CASE(HIPBLAS_STATUS_ARCH_MISMATCH);
    CASE(HIPBLAS_STATUS_MAPPING_ERROR);
    CASE(HIPBLAS_STATUS_EXECUTION_FAILED);
    CASE(HIPBLAS_STATUS_INTERNAL_ERROR);
  default: return "HIPBLAS_STATUS_UNKNOWN_ERROR";
  }
}
// LCOV_EXCL_STOP

typedef struct {
  int opt_block_size;
  int device_id;
  hipblasHandle_t hipblas_handle;
} Ceed_Hip;

CEED_INTERN int CeedHipGetResourceRoot(Ceed ceed, const char *resource,
                                       char **resource_root);

CEED_INTERN int CeedHipInit(Ceed ceed, const char *resource);

CEED_INTERN int CeedDestroy_Hip(Ceed ceed);

#endif // _ceed_hip_common_h