Add CUTLASS_CHECK macro
This commit is contained in:
@@ -22,6 +22,17 @@
|
||||
#include <cutlass/util/packed_stride.hpp>
|
||||
#include <cutlass/util/device_memory.h>
|
||||
|
||||
// CUTLASS_CHECK macro (not in headers, define our own)
|
||||
#define CUTLASS_CHECK(status) \
|
||||
do { \
|
||||
cutlass::Status _s = status; \
|
||||
if (_s != cutlass::Status::kSuccess) { \
|
||||
fprintf(stderr, "CUTLASS error at %s:%d: %s\n", \
|
||||
__FILE__, __LINE__, cutlassGetStatusString(_s)); \
|
||||
return -1; \
|
||||
} \
|
||||
} while (0)
|
||||
|
||||
#if defined(CUTLASS_ARCH_MMA_SM100_SUPPORTED)
|
||||
|
||||
using namespace cute;
|
||||
|
||||
Reference in New Issue
Block a user