#ifndef TRITON_TVM_FFI_EXCEPTION_H_
#define TRITON_TVM_FFI_EXCEPTION_H_

#include "type.h"
#include <cuda.h>
#include <exception>

namespace triton_tvm_ffi {

class CUDAException : public std::exception {
public:
  CUDAException(CUresult code);
  const char *what() const noexcept override;

private:
  const CUresult code_;
};

class UnknownTypeException : public std::exception {
public:
  UnknownTypeException(Type type);
  UnknownTypeException(std::string_view type);
  const char *what() const noexcept override;

private:
  const std::string message_;
};

} // namespace triton_tvm_ffi

#endif