24 lines
1.4 KiB
C
24 lines
1.4 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#include "cuda_runtime.h"
|
||
|
|
|
||
|
|
#define CUTLASS_CHECK(status) \
|
||
|
|
{ \
|
||
|
|
cutlass::Status error = status; \
|
||
|
|
if (error != cutlass::Status::kSuccess) { \
|
||
|
|
std::cerr << "Got cutlass error: " << cutlassGetStatusString(error) << " at: " << __LINE__ \
|
||
|
|
<< std::endl; \
|
||
|
|
exit(EXIT_FAILURE); \
|
||
|
|
} \
|
||
|
|
}
|
||
|
|
|
||
|
|
#define CUDA_CHECK(status) \
|
||
|
|
{ \
|
||
|
|
cudaError_t error = status; \
|
||
|
|
if (error != cudaSuccess) { \
|
||
|
|
std::cerr << "Got bad cuda status: " << cudaGetErrorString(error) \
|
||
|
|
<< " at line: " << __LINE__ << std::endl; \
|
||
|
|
exit(EXIT_FAILURE); \
|
||
|
|
} \
|
||
|
|
}
|