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

small fix

上级 ff517375
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册