#include <c10/util/Exception.h>

#include "torch_npu/csrc/framework/interface/LibAscendHal.h"
#include "torch_npu/csrc/core/npu/register/FunctionLoader.h"
#include "torch_npu/csrc/core/npu/NPUException.h"

namespace at_npu {
namespace native {
#undef TORCH_NPU_LOAD_FUNC
#define TORCH_NPU_LOAD_FUNC(funcName) \
  TORCH_NPU_REGISTER_FUNCTION(libascend_hal, funcName)

#undef TORCH_NPU_GET_FUNC
#define TORCH_NPU_GET_FUNC(funcName)              \
  TORCH_NPU_GET_FUNCTION(libascend_hal, funcName)

TORCH_NPU_REGISTER_LIBRARY(libascend_hal)
TORCH_NPU_LOAD_FUNC(halGetDeviceInfo)
TORCH_NPU_LOAD_FUNC(halGetAPIVersion)
constexpr uint32_t KHZTOMHZ = 1000U;
constexpr uint32_t DRV_ERROR_NONE = 0;
constexpr uint32_t ERR_FREQ = 0;
constexpr uint32_t ERR_VER = 0;
constexpr uint32_t FREQ_CONFIG = 24;

int64_t getFreq()
{
    using getReqFun = int32_t (*)(uint32_t, int32_t, int32_t, int64_t*);
    static getReqFun getFreqInfo = nullptr;
    if (getFreqInfo == nullptr) {
        getFreqInfo = (getReqFun)TORCH_NPU_GET_FUNC(halGetDeviceInfo);
        if (getFreqInfo == nullptr) {
            TORCH_NPU_WARN("Failed to find function halGetDeviceInfo.");
            return ERR_FREQ;
        }
    }
    int64_t freq = ERR_FREQ;
    if (getFreqInfo(0, 0, FREQ_CONFIG, &freq) == DRV_ERROR_NONE && freq > 0) {
        return freq / KHZTOMHZ;
    }
    return ERR_FREQ;
}

int64_t getVer()
{
    using getReqFun = int32_t (*)(int32_t*);
    static getReqFun getVerInfo = nullptr;
    if (getVerInfo == nullptr) {
        getVerInfo = (getReqFun)TORCH_NPU_GET_FUNC(halGetAPIVersion);
        if (getVerInfo == nullptr) {
            TORCH_NPU_WARN("Failed to find function halGetAPIVersion.");
            return ERR_VER;
        }
    }
    int32_t ver = ERR_VER;
    if (getVerInfo(&ver) != DRV_ERROR_NONE) {
        TORCH_NPU_WARN("Failed to find version.");
        return ERR_VER;
    }
    return ver;
}

bool isSyscntEnable()
{
    constexpr int32_t supportVersion = 0x071905;
    return getVer() >= supportVersion && getFreq() != ERR_FREQ;
}

} // namespace native
} // namespace at_npu