Check if gpu_assert was overridden in TensorGpuHipCudaDefines

This commit is contained in:
Eugene Zhulenev 2019-04-25 11:19:17 -07:00
parent 07355d47c6
commit 01d7e6ee9b

View File

@ -78,6 +78,9 @@
#endif
// gpu_assert can be overridden
#ifndef gpu_assert
#if defined(EIGEN_HIP_DEVICE_COMPILE) || (defined(EIGEN_CUDACC) && (EIGEN_CUDACC_VER==0))
// clang-cuda and HIPCC do not support the use of assert on the GPU side.
#define gpu_assert(COND)
@ -85,4 +88,6 @@
#define gpu_assert(COND) assert(COND)
#endif
#endif // gpu_assert
#endif // EIGEN_CXX11_TENSOR_GPU_HIP_CUDA_DEFINES_H