#ifndef TRITON_TVM_FFI_EXCEPTION_H_
#define TRITON_TVM_FFI_EXCEPTION_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;
};

} // namespace triton_tvm_ffi

#endif