Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
Paddle
提交
b5766fbf
P
Paddle
项目概览
BaiXuePrincess
/
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看板
未验证
提交
b5766fbf
编写于
5月 26, 2022
作者:
C
Chen Weihang
提交者:
GitHub
5月 26, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
polish kernel type str (#42791) (#42931)
上级
d5b6eec2
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
7 addition
and
8 deletion
+7
-8
paddle/fluid/framework/op_kernel_type.h
paddle/fluid/framework/op_kernel_type.h
+3
-3
paddle/fluid/framework/op_kernel_type_test.cc
paddle/fluid/framework/op_kernel_type_test.cc
+4
-5
未找到文件。
paddle/fluid/framework/op_kernel_type.h
浏览文件 @
b5766fbf
...
...
@@ -82,9 +82,9 @@ class OpKernelType {
inline
std
::
ostream
&
operator
<<
(
std
::
ostream
&
os
,
const
OpKernelType
&
kernel_key
)
{
os
<<
"
data_type["
<<
kernel_key
.
data_type_
<<
"]:
data_layout["
<<
kernel_key
.
data_layout_
<<
"]
:
place["
<<
kernel_key
.
place_
<<
"]
:library_type["
<<
kernel_key
.
library_type_
<<
"]
"
;
os
<<
"
{data_type["
<<
kernel_key
.
data_type_
<<
"];
data_layout["
<<
kernel_key
.
data_layout_
<<
"]
;
place["
<<
kernel_key
.
place_
<<
"]
; library_type["
<<
kernel_key
.
library_type_
<<
"]}
"
;
return
os
;
}
...
...
paddle/fluid/framework/op_kernel_type_test.cc
浏览文件 @
b5766fbf
...
...
@@ -27,16 +27,15 @@ TEST(OpKernelType, ToString) {
LibraryType
::
kCUDNN
);
ASSERT_EQ
(
paddle
::
framework
::
KernelTypeToString
(
op_kernel_type
),
"
data_type[float]:data_layout[NCHW]:place[Place(cpu)]:library_type[
"
"
CUDNN]
"
);
"
{data_type[float]; data_layout[NCHW]; place[Place(cpu)];
"
"
library_type[CUDNN]}
"
);
using
CUDAPlace
=
paddle
::
platform
::
CUDAPlace
;
OpKernelType
op_kernel_type2
(
DataType
::
FP16
,
CUDAPlace
(
0
),
DataLayout
::
kNCHW
,
LibraryType
::
kCUDNN
);
ASSERT_EQ
(
paddle
::
framework
::
KernelTypeToString
(
op_kernel_type2
),
"data_type[::paddle::platform::float16]:data_layout[NCHW]:place["
"Place(gpu:0)]:library_"
"type[CUDNN]"
);
"{data_type[::paddle::platform::float16]; data_layout[NCHW]; "
"place[Place(gpu:0)]; library_type[CUDNN]}"
);
}
TEST
(
OpKernelType
,
Hash
)
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录