diff --git a/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt b/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt index c823d7e9fcd63dd7719ac1403952b03c2d2f03c0..6cfc9536f20e88571a9845a50be0341fe4d9f78b 100644 --- a/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt +++ b/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt @@ -214,7 +214,7 @@ MulOp(const std::string &type, const framework::VariableNameMap &inputs, ```cpp // if use Eigen unsupported module before include head files - #define EIGEN_USE_GPU + // #define EIGEN_USE_GPU namespace ops = paddle::operators; REGISTER_OP_GPU_KERNEL(mul, ops::MulKernel); diff --git a/develop/doc_cn/howto/dev/new_op_cn.html b/develop/doc_cn/howto/dev/new_op_cn.html index 62c591f07cfc6f3346d881c4273ab244c2bd7ba4..525030728c0c9d4d754ba63461dff9815c436ae0 100644 --- a/develop/doc_cn/howto/dev/new_op_cn.html +++ b/develop/doc_cn/howto/dev/new_op_cn.html @@ -394,7 +394,7 @@ Kernel实现 | CPU、GPU共享Kernel实现在.cu的开始请加上宏定义 #define EIGEN_USE_GPU,代码示例如下:
// if use Eigen unsupported module before include head files
-#define EIGEN_USE_GPU
+// #define EIGEN_USE_GPU
 
 namespace ops = paddle::operators;
 REGISTER_OP_GPU_KERNEL(mul, ops::MulKernel<paddle::platform::GPUPlace, float>);