Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Crayon鑫
Paddle
提交
2d2b6332
P
Paddle
项目概览
Crayon鑫
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
2d2b6332
编写于
1月 03, 2018
作者:
L
Luo Tao
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add more comments in CMakelists.txt of operator
上级
5974c1b7
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
5 addition
and
4 deletion
+5
-4
paddle/framework/op_registry.h
paddle/framework/op_registry.h
+2
-2
paddle/operators/CMakeLists.txt
paddle/operators/CMakeLists.txt
+3
-2
未找到文件。
paddle/framework/op_registry.h
浏览文件 @
2d2b6332
...
@@ -37,8 +37,8 @@ class Registrar {
...
@@ -37,8 +37,8 @@ class Registrar {
public:
public:
// In our design, various kinds of classes, e.g., operators and kernels,
// In our design, various kinds of classes, e.g., operators and kernels,
// have their corresponding registry and registrar. The action of
// have their corresponding registry and registrar. The action of
// registration is in the constructor of a global registrar variable, which
,
// registration is in the constructor of a global registrar variable, which
//
however,
are not used in the code that calls package framework, and would
// are not used in the code that calls package framework, and would
// be removed from the generated binary file by the linker. To avoid such
// be removed from the generated binary file by the linker. To avoid such
// removal, we add Touch to all registrar classes and make USE_OP macros to
// removal, we add Touch to all registrar classes and make USE_OP macros to
// call this method. So, as long as the callee code calls USE_OP, the global
// call this method. So, as long as the callee code calls USE_OP, the global
...
...
paddle/operators/CMakeLists.txt
浏览文件 @
2d2b6332
...
@@ -68,9 +68,10 @@ function(op_library TARGET)
...
@@ -68,9 +68,10 @@ function(op_library TARGET)
endif
()
endif
()
endforeach
()
endforeach
()
# The registration of USE_OP, please refer to paddle/framework/op_registry.h.
# Note that it's enough to just adding one operator to pybind in a *_op.cc file.
# And for detail pybind information, please see generated paddle/pybind/pybind.h.
file
(
READ
${
TARGET
}
.cc TARGET_CONTENT
)
file
(
READ
${
TARGET
}
.cc TARGET_CONTENT
)
# It's enough to just adding one operator to pybind.
# And for detail pybind information, please see paddle/pybind/pybind.h.
string
(
REGEX MATCH
"REGISTER_OP
\\
(.*REGISTER_OP
\\
("
multi_register
"
${
TARGET_CONTENT
}
"
)
string
(
REGEX MATCH
"REGISTER_OP
\\
(.*REGISTER_OP
\\
("
multi_register
"
${
TARGET_CONTENT
}
"
)
string
(
REGEX MATCH
"REGISTER_OP
\\
([a-z0-9_]*,"
one_register
"
${
multi_register
}
"
)
string
(
REGEX MATCH
"REGISTER_OP
\\
([a-z0-9_]*,"
one_register
"
${
multi_register
}
"
)
if
(
one_register STREQUAL
""
)
if
(
one_register STREQUAL
""
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录