提交 9d8b3059 编写于 作者: K Kexin Zhao

small fix

上级 ff517375
...@@ -18,7 +18,7 @@ limitations under the License. */ ...@@ -18,7 +18,7 @@ limitations under the License. */
#include <istream> #include <istream>
#include <ostream> #include <ostream>
#include <cuda.h> // seems need to delete it #include <cuda.h>
#ifdef USE_EIGEN // delete this #if macro #ifdef USE_EIGEN // delete this #if macro
#include "Eigen/src/Core/arch/CUDA/Half.h" #include "Eigen/src/Core/arch/CUDA/Half.h"
...@@ -32,7 +32,7 @@ limitations under the License. */ ...@@ -32,7 +32,7 @@ limitations under the License. */
#endif // CUDA_VERSION >= 7050 #endif // CUDA_VERSION >= 7050
#else #else
#define PADDLE_HOSTDEVICE #define PADDLE_HOSTDEVICE
#endif // __CUDA_ARCH__ #endif // __CUDACC__
#ifdef __arm__ #ifdef __arm__
#define PADDLE_ARM_32 #define PADDLE_ARM_32
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册