implement launch with cpp

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-07-01 08:51:56 +08:00 · 2026-01-30 15:05:05 +08:00
parent 37a8f4a5be
commit 524cf83708
10 changed files with 250 additions and 243 deletions
@@ -1,8 +1,19 @@
 #ifndef TRITON_TVM_FFI_MACRO_H_
 #define TRITON_TVM_FFI_MACRO_H_

+#include "exception.h"
+
 #if defined(__GNUC__) || defined(__clang__)
 #define TRITON_TVM_FFI_INLINE __attribute__((always_inline)) inline
 #endif

+#define UNLIKELY(cond) __builtin_expect((cond), 0)
+
+#define CUDA_CHECK(code)                                                       \
+  do {                                                                         \
+    if (UNLIKELY((code) != CUDA_SUCCESS)) {                                    \
+      throw triton_tvm_ffi::CUDAException(code);                               \
+    }                                                                          \
+  } while (false)
+
 #endif