diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2019-04-25 11:19:17 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-04-25 11:19:17 -0700 |
| commit | 01d7e6ee9b4bb390842e6f1e1afd4e77a7e26f78 (patch) | |
| tree | 3e1e6cc7e16ed77dd3b3028207e2fcef7abcb62b /unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h | |
| parent | 07355d47c6007bdbc13a18e758f932446c2b2ba4 (diff) | |
Check if gpu_assert was overridden in TensorGpuHipCudaDefines
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h b/unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h index 40f58f628..c03096363 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorGpuHipCudaDefines.h @@ -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 |