Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 331
Star 4
Fork 1
P
Paddle-Lite
提交 4f8a26c4 编写于 作者: Z zhaoyang-star
浏览文件
操作

fix valid_places when using opencl. test=develop

上级 784f75fa
隐藏空白更改
内联 并排
Showing with 1 addition and 2 deletion
lite/demo/cxx/mobile_full/mobilenetv1_full_api.cc
浏览文件 @ 4f8a26c4
...@@ -45,8 +45,7 @@ void RunModel() { ...@@ -45,8 +45,7 @@ void RunModel() {
#ifdef DEMO_WITH_OPENCL #ifdef DEMO_WITH_OPENCL
std::vector<Place> valid_places{ std::vector<Place> valid_places{
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNCHW)}, Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNCHW)},
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNHWC)}, Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNHWC)}};
Place{TARGET(kARM), PRECISION(kFloat)}};
#else #else
std::vector<Place> valid_places{Place{TARGET(kARM), PRECISION(kFloat)}}; std::vector<Place> valid_places{Place{TARGET(kARM), PRECISION(kFloat)}};
#endif #endif
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或