提交 4f8a26c4 编写于 作者: Z zhaoyang-star

fix valid_places when using opencl. test=develop

上级 784f75fa
......@@ -45,8 +45,7 @@ void RunModel() {
#ifdef DEMO_WITH_OPENCL
std::vector<Place> valid_places{
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNCHW)},
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNHWC)},
Place{TARGET(kARM), PRECISION(kFloat)}};
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNHWC)}};
#else
std::vector<Place> valid_places{Place{TARGET(kARM), PRECISION(kFloat)}};
#endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册