From 9d8b30596491930c6137e56d7883370bff24d2c8 Mon Sep 17 00:00:00 2001
From: Kexin Zhao <zhaokexin01@baidu.com>
Date: Tue, 7 Nov 2017 13:19:59 -0800
Subject: [PATCH] small fix

---
 paddle/math/float16.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/math/float16.h b/paddle/math/float16.h
index 84fe613d5..5fe285496 100644
--- a/paddle/math/float16.h
+++ b/paddle/math/float16.h
@@ -18,7 +18,7 @@ limitations under the License. */
 #include <istream>
 #include <ostream>
 
-#include <cuda.h>  // seems need to delete it
+#include <cuda.h>
 
 #ifdef USE_EIGEN  // delete this #if macro
 #include "Eigen/src/Core/arch/CUDA/Half.h"
@@ -32,7 +32,7 @@ limitations under the License. */
 #endif  // CUDA_VERSION >= 7050
 #else
 #define PADDLE_HOSTDEVICE
-#endif  // __CUDA_ARCH__
+#endif  // __CUDACC__
 
 #ifdef __arm__
 #define PADDLE_ARM_32
-- 
GitLab