Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
飞不过沧海
Paddle
提交
f51a5791
P
Paddle
项目概览
飞不过沧海
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
f51a5791
编写于
3月 24, 2022
作者:
H
huzhiqiang
提交者:
GitHub
3月 24, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[Infrt] add method for automatically scanning pass and kernel info (#40822)
上级
68c9e3e4
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
121 addition
and
0 deletion
+121
-0
tools/infrt/print_kernel_pass_info.py
tools/infrt/print_kernel_pass_info.py
+121
-0
未找到文件。
tools/infrt/print_kernel_pass_info.py
0 → 100644
浏览文件 @
f51a5791
# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import
os
import
re
import
json
skip_list
=
[]
def
remove_grad_kernel
(
kernels
):
clean_kernels
=
[]
for
kernel_
in
kernels
:
if
(
not
"_grad"
in
kernel_
):
clean_kernels
.
append
(
kernel_
)
return
clean_kernels
CPU_KERNEL_REGISTER
=
"REGISTER_OP_CPU_KERNEL("
GPU_KERNEL_REGISTER
=
"REGISTER_OP_CUDA_KERNEL("
XPU_KERNEL_REGISTER
=
"REGISTER_OP_XPU_KERNEL("
def
get_compat_kernels_info
(
register
):
kernels_info
=
{}
kernel_names
=
[]
for
dirpath
,
dirnames
,
filenames
in
os
.
walk
(
"../../paddle/fluid/operators"
):
for
file_name
in
filenames
:
if
not
".cc"
in
file_name
:
continue
with
open
(
os
.
path
.
join
(
dirpath
,
file_name
))
as
f
:
txt
=
f
.
readlines
()
content
=
""
registry
=
False
is_macro_defination
=
False
for
line
in
txt
:
if
line
.
strip
().
startswith
(
"#define"
)
and
line
.
strip
(
).
endswith
(
"
\\
"
):
is_macro_defination
=
True
continue
if
is_macro_defination
:
if
not
line
.
strip
().
endswith
(
"
\\
"
):
is_macro_defination
=
False
continue
if
(
register
in
line
):
content
=
""
registry
=
True
if
(
registry
):
content
+=
line
if
(
registry
and
";"
in
line
):
kernel_name
=
content
.
replace
(
"
\n
"
,
""
).
replace
(
" "
,
""
).
strip
(
register
).
split
(
","
)
registry
=
False
kernel_names
.
append
(
kernel_name
[
0
])
return
remove_grad_kernel
(
kernel_names
)
def
show_kernel_statistics
(
backend
,
kernels
):
print
(
"=== kernels statistics === "
)
print
(
"the number of "
+
backend
+
" kernels is: "
+
str
(
len
(
kernels
))
+
"
\n
"
)
print
(
kernels
)
print
(
"
\n
"
)
def
show_pass_statistics
(
backend
,
passes
):
print
(
"=== Passes Statistics === "
)
print
(
"The number of "
+
backend
+
" passes is: "
+
str
(
len
(
passes
))
+
"
\n
"
)
print
(
passes
)
print
(
"
\n
"
)
def
get_passes_info
(
register
):
pass_registry_func
=
""
with
open
(
"../../paddle/fluid/inference/api/paddle_pass_builder.cc"
)
as
f
:
txt
=
f
.
readlines
()
stack
=
[]
registry_fun_found
=
False
for
line
in
txt
:
if
line
.
strip
().
startswith
(
"//"
):
continue
if
register
in
line
:
registry_fun_found
=
True
if
(
registry_fun_found
):
pass_registry_func
+=
line
if
registry_fun_found
:
for
char
in
line
:
if
char
==
"{"
:
stack
.
append
(
char
)
if
char
==
"}"
:
stack
.
pop
()
if
len
(
stack
)
==
0
:
registry_fun_found
=
False
pass_list
=
re
.
findall
(
"
\"
(.+?)_pass
\"
"
,
pass_registry_func
)
return
pass_list
if
__name__
==
"__main__"
:
cpu_kernels
=
get_compat_kernels_info
(
CPU_KERNEL_REGISTER
)
gpu_kernels
=
get_compat_kernels_info
(
GPU_KERNEL_REGISTER
)
xpu_kernels
=
get_compat_kernels_info
(
XPU_KERNEL_REGISTER
)
show_kernel_statistics
(
"CPU"
,
cpu_kernels
)
show_kernel_statistics
(
"GPU"
,
gpu_kernels
)
show_kernel_statistics
(
"XPU"
,
xpu_kernels
)
cpu_passes
=
get_passes_info
(
"CpuPassStrategy::CpuPassStrategy()"
)
gpu_passes
=
get_passes_info
(
"GpuPassStrategy::GpuPassStrategy()"
)
show_pass_statistics
(
"CPU"
,
cpu_passes
)
show_pass_statistics
(
"GPU"
,
gpu_passes
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录