mirror of
https://github.com/deepseek-ai/DeepEP
synced 2025-04-30 10:43:21 +00:00
52 lines
1.2 KiB
Plaintext
52 lines
1.2 KiB
Plaintext
#pragma once
|
|
|
|
#include <string>
|
|
#include <exception>
|
|
|
|
#include "configs.cuh"
|
|
|
|
#ifndef EP_STATIC_ASSERT
|
|
#define EP_STATIC_ASSERT(cond, reason) static_assert(cond, reason)
|
|
#endif
|
|
|
|
class EPException: public std::exception {
|
|
private:
|
|
std::string message = {};
|
|
|
|
public:
|
|
explicit EPException(const char *name, const char* file, const int line, const std::string& error) {
|
|
message = std::string("Failed: ") + name + " error " + file + ":" + std::to_string(line) + " '" + error + "'";
|
|
}
|
|
|
|
const char *what() const noexcept override { return message.c_str(); }
|
|
};
|
|
|
|
#ifndef CUDA_CHECK
|
|
#define CUDA_CHECK(cmd) \
|
|
do { \
|
|
cudaError_t e = (cmd); \
|
|
if (e != cudaSuccess) { \
|
|
throw EPException("CUDA", __FILE__, __LINE__, cudaGetErrorString(e)); \
|
|
} \
|
|
} while (0)
|
|
#endif
|
|
|
|
#ifndef EP_HOST_ASSERT
|
|
#define EP_HOST_ASSERT(cond) \
|
|
do { \
|
|
if (not (cond)) { \
|
|
throw EPException("Assertion", __FILE__, __LINE__, #cond); \
|
|
} \
|
|
} while (0)
|
|
#endif
|
|
|
|
#ifndef EP_DEVICE_ASSERT
|
|
#define EP_DEVICE_ASSERT(cond) \
|
|
do { \
|
|
if (not (cond)) { \
|
|
printf("Assertion failed: %s:%d, condition: %s\n", __FILE__, __LINE__, #cond); \
|
|
asm("trap;"); \
|
|
} \
|
|
} while (0)
|
|
#endif
|