For cuda 9.1 replace math_functions.hpp with cuda_runtime.h

This commit is contained in:
nluehr 2017-12-18 16:51:15 -08:00
parent 06bf1047f9
commit f9bdcea022

View File

@ -54,9 +54,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