Files
candle/candle-flash-attn/kernels/error.h
2024-07-15 20:37:36 +02:00

9 lines
313 B
C

#pragma once
#define C10_CUDA_CHECK(EXPR) \
do { \
const cudaError_t __err = EXPR; \
} while (0)
#define C10_CUDA_KERNEL_LAUNCH_CHECK() C10_CUDA_CHECK(cudaGetLastError())