devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle min/max/inf/etc issue in cuda_fp16.h directly in test/main.h

Browse Source
This commit is contained in:
Gael Guennebaud
2017-08-24 11:26:41 +02:00
parent 21633e585b
commit 304ef29571
13 changed files with 13 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/cuda_basic.cu
View File

@@ -20,9 +20,6 @@
#include <math_constants.h>
#include <cuda.h>
#if EIGEN_CUDACC_VER >= 70500
#include <cuda_fp16.h>
#endif
#include "main.h"
#include "cuda_common.h"

13
test/main.h
View File

@@ -50,6 +50,19 @@
#endif
#endif
// Same for cuda_fp16.h
#if defined(__CUDACC_VER_MAJOR__) && (__CUDACC_VER_MAJOR__ >= 9)
#define EIGEN_TEST_CUDACC_VER ((__CUDACC_VER_MAJOR__ * 10000) + (__CUDACC_VER_MINOR__ * 100))
#elif defined(__CUDACC_VER__)
#define EIGEN_TEST_CUDACC_VER __CUDACC_VER__
#else
#define EIGEN_TEST_CUDACC_VER 0
#endif
#if EIGEN_TEST_CUDACC_VER >= 70500
#include <cuda_fp16.h>
#endif
// To test that all calls from Eigen code to std::min() and std::max() are
// protected by parenthesis against macro expansion, the min()/max() macros
// are defined here and any not-parenthesized min/max call will cause a