From f9bdcea022e24bac4a66a937c37de92f7f22b9da Mon Sep 17 00:00:00 2001
From: nluehr <nluehr@nvidia.com>
Date: Mon, 18 Dec 2017 16:51:15 -0800
Subject: For cuda 9.1 replace math_functions.hpp with cuda_runtime.h

---
 Eigen/Core | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'Eigen/Core')

diff --git a/Eigen/Core b/Eigen/Core
index c66359b79..5a6dec8cc 100644
--- a/Eigen/Core
+++ b/Eigen/Core
@@ -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
-- 
cgit v1.2.3