support GetDeviceProperties

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
2026-01-28 02:16:10 +08:00
parent ab83dded12
commit 3a485166b4
8 changed files with 95 additions and 9 deletions
+20
View File
@@ -0,0 +1,20 @@
#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