Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wmsofts
Paddle
提交
8aebcabf
P
Paddle
项目概览
wmsofts
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
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,发现更多精彩内容 >>
未验证
提交
8aebcabf
编写于
9月 04, 2020
作者:
A
arlesniak
提交者:
GitHub
9月 04, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Test for passing global env FLAGS_use_mkldnn (#26938)
上级
21565e8d
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
107 addition
and
1 deletion
+107
-1
python/paddle/fluid/tests/unittests/mkldnn/check_flags_use_mkldnn.py
...le/fluid/tests/unittests/mkldnn/check_flags_use_mkldnn.py
+48
-0
python/paddle/fluid/tests/unittests/mkldnn/test_flags_use_mkldnn.py
...dle/fluid/tests/unittests/mkldnn/test_flags_use_mkldnn.py
+58
-0
python/paddle/fluid/tests/unittests/test_imperative_basic.py
python/paddle/fluid/tests/unittests/test_imperative_basic.py
+1
-1
未找到文件。
python/paddle/fluid/tests/unittests/mkldnn/check_flags_use_mkldnn.py
0 → 100644
浏览文件 @
8aebcabf
# Copyright (c) 2020 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.
from
__future__
import
unicode_literals
from
__future__
import
print_function
import
numpy
as
np
import
paddle.fluid
as
fluid
import
os
from
paddle.fluid.layer_helper
import
LayerHelper
def
check
():
print
(
"check: fluid.core.globals()['FLAGS_use_mkldnn']="
,
fluid
.
core
.
globals
()[
"FLAGS_use_mkldnn"
])
print
(
"check: fluid.get_flags('FLAGS_use_mkldnn')="
,
fluid
.
get_flags
([
'FLAGS_use_mkldnn'
]))
print
(
"check: DNNL_VERBOSE="
,
os
.
environ
[
'DNNL_VERBOSE'
])
a_np
=
np
.
random
.
uniform
(
-
2
,
2
,
(
10
,
20
,
30
)).
astype
(
np
.
float32
)
helper
=
LayerHelper
(
fluid
.
unique_name
.
generate
(
str
(
"test"
)),
act
=
"relu"
)
func
=
helper
.
append_activation
with
fluid
.
dygraph
.
guard
(
fluid
.
core
.
CPUPlace
()):
a
=
fluid
.
dygraph
.
to_variable
(
a_np
)
res1
=
func
(
a
)
res2
=
np
.
maximum
(
a_np
,
0
)
assert
(
np
.
array_equal
(
res1
.
numpy
(),
res2
))
if
__name__
==
'__main__'
:
try
:
check
()
for
k
,
v
in
sorted
(
os
.
environ
.
items
()):
print
(
k
+
':'
,
v
)
print
(
'
\n
'
)
except
Exception
as
e
:
print
(
e
)
print
(
type
(
e
))
python/paddle/fluid/tests/unittests/mkldnn/test_flags_use_mkldnn.py
0 → 100644
浏览文件 @
8aebcabf
# Copyright (c) 2020 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.
from
__future__
import
unicode_literals
from
__future__
import
print_function
import
unittest
import
os
import
sys
import
subprocess
class
TestFlagsUseMkldnn
(
unittest
.
TestCase
):
def
setUp
(
self
):
self
.
_python_interp
=
sys
.
executable
self
.
_python_interp
+=
" check_flags_use_mkldnn.py"
self
.
env
=
os
.
environ
.
copy
()
self
.
env
[
str
(
"GLOG_v"
)]
=
str
(
"3"
)
self
.
env
[
str
(
"DNNL_VERBOSE"
)]
=
str
(
"1"
)
self
.
env
[
str
(
"FLAGS_use_mkldnn"
)]
=
str
(
"1"
)
def
test_flags_use_mkl_dnn
(
self
):
cmd
=
self
.
_python_interp
proc
=
subprocess
.
Popen
(
cmd
.
split
(
" "
),
stdout
=
subprocess
.
PIPE
,
stderr
=
subprocess
.
PIPE
,
env
=
self
.
env
)
out
,
err
=
proc
.
communicate
()
returncode
=
proc
.
returncode
print
(
'out'
,
out
)
print
(
'err'
,
err
)
assert
returncode
==
0
# in python3, type(out) is 'bytes', need use encode
assert
out
.
find
(
"dnnl_verbose,exec,cpu,eltwise,jit:avx512_common,forward_training,"
"data_f32::blocked:abc:f0 diff_undef::undef::f0,,alg:eltwise_relu"
.
encode
())
!=
-
1
if
__name__
==
'__main__'
:
unittest
.
main
()
python/paddle/fluid/tests/unittests/test_imperative_basic.py
浏览文件 @
8aebcabf
...
...
@@ -652,7 +652,7 @@ class TestDygraphUtils(unittest.TestCase):
a_np
=
np
.
random
.
uniform
(
-
2
,
2
,
(
10
,
20
,
30
)).
astype
(
np
.
float32
)
helper
=
LayerHelper
(
fluid
.
unique_name
.
generate
(
"test"
),
act
=
"relu"
)
func
=
helper
.
append_activation
with
fluid
.
dygraph
.
guard
():
with
fluid
.
dygraph
.
guard
(
fluid
.
core
.
CPUPlace
()
):
a
=
fluid
.
dygraph
.
to_variable
(
a_np
)
fluid
.
set_flags
({
'FLAGS_use_mkldnn'
:
True
})
try
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录