Skip to content

Commit 45ead7c

Browse files
committed
change title and add tree
1 parent 01404d1 commit 45ead7c

File tree

3 files changed

+19
-18
lines changed

3 files changed

+19
-18
lines changed

docs/dev_guides/api_contributing_guides/api_contributing_guides_cn.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,5 +107,6 @@ API设计文档的目的是为了社区开发者更容易的参与开源项目
107107
api_design_guidelines_standard_cn.md
108108
new_python_api_cn.md
109109
new_cpp_op_cn.md
110+
new_cpp_op_notes_cn.md
110111
api_docs_guidelines_cn.md
111112
api_accpetance_criteria_cn.md

docs/dev_guides/api_contributing_guides/new_cpp_op_cn.md

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# C++ OP 开发(新增原生算子)
1+
# C++ OP 开发
22

33
> 注:飞桨原生算子的开发范式正在进行重构与升级,升级后算子开发方式会大幅简化,我们会及时更新本文档内容,升级后的算子开发范式预计会在2.3版本正式上线。
44
@@ -590,28 +590,28 @@ void TraceKernel(const Context& dev_ctx,
590590
仍然以trace op为例,首先在`paddle/phi/ops/compat`目录下新建`trace_sig.cc`文件,用于放置这里的映射函数。
591591
592592
- 由于函数式kernel的一个最重要的特别就是参数顺序和类型(顺序和类型是关键,变量名称不影响),我们需要定义一个函数来做一个从OpMaker中如何获取信息,并且按照顺序传递给新的kernel函数; 这个模块就是OpArgumentMapping, trace反向op的OpArgumentMapping定义如下, KernelSignature共包含4个内容
593-
1. kernel名称,这个是我们给kernel注册的时候的名称
594-
2. input list: 这个要和OpMaker(或者GradOpMaker)中定义的Key要完全一致
595-
3. attribute list: 这个要和OpMaker(或者GradOpMaker)中定义的Key要完全一致
596-
4. output list: 这个要和OpMaker(或者GradOpMaker)中定义的Key要完全一致
593+
1. kernel名称,这个是我们给kernel注册的时候的名称
594+
2. input list: 这个要和OpMaker(或者GradOpMaker)中定义的Key要完全一致
595+
3. attribute list: 这个要和OpMaker(或者GradOpMaker)中定义的Key要完全一致
596+
4. output list: 这个要和OpMaker(或者GradOpMaker)中定义的Key要完全一致
597597
598598
599-
```cpp
600-
#include "paddle/phi/core/compat/op_utils.h"
599+
```cpp
600+
#include "paddle/phi/core/compat/op_utils.h"
601601
602-
namespace phi {
602+
namespace phi {
603603
604-
KernelSignature TraceGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
605-
return KernelSignature("trace_grad",
606-
{GradVarName("Out"), "Input"},
607-
{"offset", "axis1", "axis2"},
608-
{GradVarName("Input")});
609-
}
604+
KernelSignature TraceGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
605+
return KernelSignature("trace_grad",
606+
{GradVarName("Out"), "Input"},
607+
{"offset", "axis1", "axis2"},
608+
{GradVarName("Input")});
609+
}
610610
611-
} // namespace phi
611+
} // namespace phi
612612
613-
PD_REGISTER_ARG_MAPPING_FN(trace_grad, phi::TraceGradOpArgumentMapping);
614-
```
613+
PD_REGISTER_ARG_MAPPING_FN(trace_grad, phi::TraceGradOpArgumentMapping);
614+
```
615615
616616
>注:没有input list或attribute list的,相应花括号内留空,不能省略花括号
617617

docs/dev_guides/api_contributing_guides/new_cpp_op_notes_cn.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# C++ OP 开发(新增原生算子)注意事项
1+
# C++ OP 开发注意事项
22

33
## Paddle中Op的构建逻辑
44
### 1.Paddle中Op的构建逻辑

0 commit comments

Comments
 (0)