Skip to content

Commit 9e27d1e

Browse files
author
Yang Yang(Tony)
authored
Update new_op_cn.md
1 parent 208dd50 commit 9e27d1e

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

doc/fluid/dev/new_op_cn.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -224,8 +224,7 @@ MulOp(const std::string &type, const framework::VariableNameMap &inputs,
224224

225225
在上面的代码中:
226226

227-
- `REGISTER_OP` : 注册`ops::MulOp`类,类型名为`mul`,该类的`ProtoMaker`为`ops::MulOpMaker`,注册`ops::MulOpGrad`,类型名为`mul_grad`。
228-
- `REGISTER_OP_WITHOUT_GRADIENT` : 用于注册没有反向的Op。
227+
- `REGISTER_OPERATOR` : 注册`ops::MulOp`类,类型名为`mul`,该类的`ProtoMaker`为`ops::MulOpMaker`,注册`ops::MulOpGrad`,类型名为`mul_grad`。
229228
- `REGISTER_OP_CPU_KERNEL` :注册`ops::MulKernel`类,并特化模板参数为`paddle::platform::CPUPlace`和`float`类型,同理,注册`ops::MulGradKernel`类。
230229

231230

@@ -316,7 +315,7 @@ Op单元测试继承自`OpTest`。各项更加具体的单元测试在`TestMulOp
316315

317316
### 编译和执行
318317

319-
`python/paddle/v2/framework/tests` 目录下新增的 `test_*.py` 单元测试会被自动加入工程进行编译。
318+
`python/paddle/fluid/tests/unittests/` 目录下新增的 `test_*.py` 单元测试会被自动加入工程进行编译。
320319

321320
请注意,**不同于Op的编译测试,运行单元测试测时需要编译整个工程**,并且编译时需要打开`WITH_TESTING`, 即`cmake paddle_dir -DWITH_TESTING=ON`。编译成功后,执行下面的命令来运行单元测试:
322321

0 commit comments

Comments
 (0)