|
#pragma once |
|
|
|
#define CHECK_CUDA(call) \ |
|
do { \ |
|
cudaError_t status_ = call; \ |
|
if (status_ != cudaSuccess) { \ |
|
fprintf(stderr, "CUDA error (%s:%d): %s\n", __FILE__, __LINE__, cudaGetErrorString(status_)); \ |
|
exit(1); \ |
|
} \ |
|
} while(0) |
|
|
|
#define CHECK_CUDA_KERNEL_LAUNCH() CHECK_CUDA(cudaGetLastError()) |
|
|
|
|
|
#define FLASH_ASSERT(cond) \ |
|
do { \ |
|
if (not (cond)) { \ |
|
fprintf(stderr, "Assertion failed (%s:%d): %s\n", __FILE__, __LINE__, #cond); \ |
|
exit(1); \ |
|
} \ |
|
} while(0) |
|
|
|
|
|
#define FLASH_DEVICE_ASSERT(cond) \ |
|
do { \ |
|
if (not (cond)) { \ |
|
printf("Assertion failed (%s:%d): %s\n", __FILE__, __LINE__, #cond); \ |
|
asm("trap;"); \ |
|
} \ |
|
} while(0) |
|
|
|
|
|
#define BOOL_SWITCH(COND, CONST_NAME, ...) \ |
|
[&] { \ |
|
if (COND) { \ |
|
constexpr static bool CONST_NAME = true; \ |
|
return __VA_ARGS__(); \ |
|
} else { \ |
|
constexpr static bool CONST_NAME = false; \ |
|
return __VA_ARGS__(); \ |
|
} \ |
|
}() |
|
|
|
|
|
#define MLA_NUM_SPLITS_SWITCH(NUM_SPLITS, NAME, ...) \ |
|
[&] { \ |
|
if (NUM_SPLITS <= 32) { \ |
|
constexpr static int NAME = 32; \ |
|
return __VA_ARGS__(); \ |
|
} else if (NUM_SPLITS <= 64) { \ |
|
constexpr static int NAME = 64; \ |
|
return __VA_ARGS__(); \ |
|
} else if (NUM_SPLITS <= 96) { \ |
|
constexpr static int NAME = 96; \ |
|
return __VA_ARGS__(); \ |
|
} else if (NUM_SPLITS <= 128) { \ |
|
constexpr static int NAME = 128; \ |
|
return __VA_ARGS__(); \ |
|
} else if (NUM_SPLITS <= 160) { \ |
|
constexpr static int NAME = 160; \ |
|
return __VA_ARGS__(); \ |
|
} else { \ |
|
FLASH_ASSERT(false); \ |
|
} \ |
|
}() |
|
|
