Add CUTLASS_CHECK macro

This commit is contained in:
2026-05-13 23:28:03 +00:00
parent b7c5cba407
commit 33e5d67326

View File

@@ -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;