forked from jiuyuan/InfiniTensor
21 lines
813 B
C
21 lines
813 B
C
|
#pragma once
|
||
|
|
#include "core/common.h"
|
||
|
|
#include "xpu/runtime_ex.h"
|
||
|
|
#include "xpu/xdnn.h"
|
||
|
|
|
||
|
|
#define checkKUNLUNError(call) \
|
||
|
|
{ \
|
||
|
|
auto err = call; \
|
||
|
|
if (XPU_SUCCESS != err) { \
|
||
|
|
fprintf(stderr, "KUNLUN error in %s:%i : %s.\n", __FILE__, \
|
||
|
|
__LINE__, xpu_strerror(err)); \
|
||
|
|
exit(EXIT_FAILURE); \
|
||
|
|
} \
|
||
|
|
}
|
||
|
|
|
||
|
|
namespace infini {
|
||
|
|
|
||
|
|
using KUNLUNPtr = void *;
|
||
|
|
|
||
|
|
} // namespace infini
|