backport some nvcc 9 fixes

This commit is contained in:
Gael Guennebaud
2018-07-30 14:45:08 +02:00
parent b89d81b2a8
commit 6eb4ce5f8e
3 changed files with 102 additions and 78 deletions

View File

@@ -53,9 +53,9 @@
#endif
#define EIGEN_DEVICE_FUNC __host__ __device__
// We need math_functions.hpp to ensure that that EIGEN_USING_STD_MATH macro
// We need cuda_runtime.h to ensure that that EIGEN_USING_STD_MATH macro
// works properly on the device side
#include <math_functions.hpp>
#include <cuda_runtime.h>
#else
#define EIGEN_DEVICE_FUNC
#endif