Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
SummerGao.
Paddle
提交
740c0d58
P
Paddle
项目概览
SummerGao.
/
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,发现更多精彩内容 >>
未验证
提交
740c0d58
编写于
12月 11, 2020
作者:
W
Wilber
提交者:
GitHub
12月 11, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update for xpu ci. (#29568)
上级
d33d468f
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
54 addition
and
5 deletion
+54
-5
cmake/external/lite.cmake
cmake/external/lite.cmake
+5
-1
paddle/fluid/inference/tests/api/lite_resnet50_test.cc
paddle/fluid/inference/tests/api/lite_resnet50_test.cc
+48
-4
paddle/scripts/paddle_build.sh
paddle/scripts/paddle_build.sh
+1
-0
未找到文件。
cmake/external/lite.cmake
浏览文件 @
740c0d58
...
...
@@ -132,7 +132,11 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR)
endif
()
if
(
WITH_ARM
)
if
(
LITE_WITH_XPU
)
set
(
LITE_OUTPUT_BIN_DIR inference_lite_lib.armlinux.armv8.xpu
)
else
()
set
(
LITE_OUTPUT_BIN_DIR inference_lite_lib.armlinux.armv8
)
endif
()
else
()
set
(
LITE_OUTPUT_BIN_DIR inference_lite_lib
)
endif
()
...
...
paddle/fluid/inference/tests/api/lite_resnet50_test.cc
浏览文件 @
740c0d58
...
...
@@ -26,11 +26,7 @@ namespace inference {
TEST
(
AnalysisPredictor
,
use_gpu
)
{
std
::
string
model_dir
=
FLAGS_infer_model
+
"/"
+
"model"
;
AnalysisConfig
config
;
#if defined(PADDLE_WITH_CUDA)
config
.
EnableUseGpu
(
100
,
0
);
#elif defined(LITE_SUBGRAPH_WITH_XPU)
config
.
EnableXpu
(
100
);
#endif
config
.
SetModel
(
model_dir
+
"/model"
,
model_dir
+
"/params"
);
config
.
EnableLiteEngine
(
paddle
::
AnalysisConfig
::
Precision
::
kFloat32
,
true
);
...
...
@@ -73,6 +69,54 @@ TEST(AnalysisPredictor, use_gpu) {
}
}
#ifdef LITE_SUBGRAPH_WITH_XPU
TEST
(
AnalysisPredictor
,
use_xpu
)
{
std
::
string
model_dir
=
FLAGS_infer_model
+
"/"
+
"model"
;
AnalysisConfig
config
;
config
.
EnableLiteEngine
(
paddle
::
AnalysisConfig
::
Precision
::
kFloat32
,
true
);
config
.
EnableXpu
(
100
);
config
.
SetModel
(
model_dir
+
"/model"
,
model_dir
+
"/params"
);
std
::
vector
<
PaddleTensor
>
inputs
;
auto
predictor
=
CreatePaddlePredictor
(
config
);
const
int
batch
=
1
;
const
int
channel
=
3
;
const
int
height
=
318
;
const
int
width
=
318
;
const
int
input_num
=
batch
*
channel
*
height
*
width
;
std
::
vector
<
float
>
input
(
input_num
,
1
);
PaddleTensor
in
;
in
.
shape
=
{
batch
,
channel
,
height
,
width
};
in
.
data
=
PaddleBuf
(
static_cast
<
void
*>
(
input
.
data
()),
input_num
*
sizeof
(
float
));
in
.
dtype
=
PaddleDType
::
FLOAT32
;
inputs
.
emplace_back
(
in
);
std
::
vector
<
PaddleTensor
>
outputs
;
ASSERT_TRUE
(
predictor
->
Run
(
inputs
,
&
outputs
));
const
std
::
vector
<
float
>
truth_values
=
{
127.84
,
738.088
,
1013.22
,
-
438.055
,
366.451
,
927.585
,
736.341
,
-
633.776
,
-
329.904
,
-
430.149
,
-
633.082
,
-
146.597
,
-
1324.19
,
-
1349.29
,
-
242.68
,
117.541
,
-
801.704
,
-
391.428
,
-
404.756
,
453.995
,
515.373
,
-
133.003
,
69.3941
,
590.056
,
-
1434.66
,
-
1070.81
,
307.093
,
400.463
,
-
316.094
,
-
587.089
,
-
161.033
,
800.357
,
-
96.4212
,
748.706
,
868.226
,
-
447.936
,
112.782
,
1127.24
,
47.4587
,
677.698
,
593.126
,
-
336.462
,
551.328
,
397.816
,
78.3572
,
-
715.269
,
406.002
,
404.149
,
246.067
,
-
8.4649
,
131.345
,
-
647.951
,
};
const
size_t
expected_size
=
1
;
EXPECT_EQ
(
outputs
.
size
(),
expected_size
);
float
*
data_o
=
static_cast
<
float
*>
(
outputs
[
0
].
data
.
data
());
for
(
size_t
j
=
0
;
j
<
outputs
[
0
].
data
.
length
()
/
sizeof
(
float
);
j
+=
10
)
{
EXPECT_NEAR
((
data_o
[
j
]
-
truth_values
[
j
/
10
])
/
truth_values
[
j
/
10
],
0.
,
10e-5
);
}
}
#endif
}
// namespace inference
}
// namespace paddle
...
...
paddle/scripts/paddle_build.sh
浏览文件 @
740c0d58
...
...
@@ -288,6 +288,7 @@ EOF
-DWITH_GLOO
=
${
gloo_flag
}
\
-DLITE_GIT_TAG
=
develop
\
-DWITH_XPU
=
${
WITH_XPU
:-
OFF
}
\
-DXPU_SDK_ROOT
=
${
XPU_SDK_ROOT
:-
""
}
\
-DWITH_LITE
=
${
WITH_LITE
:-
OFF
}
\
-DWITH_UNITY_BUILD
=
${
WITH_UNITY_BUILD
:-
OFF
}
;
build_error
=
$?
if
[
"
$build_error
"
!=
0
]
;
then
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录