Device side assert support
This commit is contained in:
@@ -99,7 +99,7 @@ extern int HIP_TRACE_API;
|
|||||||
// TODO-HCC add a dummy implementation of assert, need to replace with a proper kernel exit call.
|
// TODO-HCC add a dummy implementation of assert, need to replace with a proper kernel exit call.
|
||||||
#if __HIP_DEVICE_COMPILE__ == 1
|
#if __HIP_DEVICE_COMPILE__ == 1
|
||||||
#undef assert
|
#undef assert
|
||||||
#define assert(COND) { if (COND) {} }
|
#define assert(COND) { if (!COND) {abort();} }
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -111,6 +111,8 @@ kernelName<<<numblocks,numthreads,memperblock,streamId>>>(__VA_ARGS__);\
|
|||||||
|
|
||||||
#ifdef __HIP_DEVICE_COMPILE__
|
#ifdef __HIP_DEVICE_COMPILE__
|
||||||
#define abort() {asm("trap;");}
|
#define abort() {asm("trap;");}
|
||||||
|
#undef assert
|
||||||
|
#define assert(COND) { if (!COND) {abort();} }
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
Reference in New Issue
Block a user