forked from jiuyuan/InfiniTensor
Add: IT_ASSERT in checkCudaError
This commit is contained in:
parent
b13b799fbe
commit
f877eca517
|
@ -13,6 +13,7 @@
|
|||
if (cudaSuccess != err) { \
|
||||
fprintf(stderr, "Cuda error in %s:%i : %s.\n", __FILE__, __LINE__, \
|
||||
cudaGetErrorString(err)); \
|
||||
IT_ASSERT(false); \
|
||||
exit(EXIT_FAILURE); \
|
||||
} \
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue