mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-06-04 18:54:00 +08:00
For cuda 9.1 replace math_functions.hpp with cuda_runtime.h
This commit is contained in:
parent
06bf1047f9
commit
f9bdcea022
@ -54,9 +54,9 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define EIGEN_DEVICE_FUNC __host__ __device__
|
#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
|
// works properly on the device side
|
||||||
#include <math_functions.hpp>
|
#include <cuda_runtime.h>
|
||||||
#else
|
#else
|
||||||
#define EIGEN_DEVICE_FUNC
|
#define EIGEN_DEVICE_FUNC
|
||||||
#endif
|
#endif
|
||||||
|
Loading…
x
Reference in New Issue
Block a user