Skip to content

Commit e62598b

Browse files
authored
[cxx demo] fix arm linux full demo bug (#6058) (#6068)
1 parent 81ecf2d commit e62598b

File tree

3 files changed

+12
-3
lines changed

3 files changed

+12
-3
lines changed

lite/demo/cxx/armlinux_mobilenetv1_full_demo/CMakeLists.txt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,14 @@ set(ARMLINUX_ARCH_ABI armv8)
77

88
# 1. path to Paddle-Lite lib
99
set(LITE_DIR "${PROJECT_SOURCE_DIR}/../../../cxx")
10+
set(GFLAGS_DIR "${PROJECT_SOURCE_DIR}/../../../../third_party/install/gflags")
1011

1112
# 2. link Paddle-Lite directory
1213
link_directories(${LITE_DIR}/lib)
14+
link_directories(${GFLAGS_DIR}/lib)
1315

1416
include_directories(${LITE_DIR}/include)
17+
include_directories(${GFLAGS_DIR}/include)
1518

1619
# 3. compile options
1720
if(ARMLINUX_ARCH_ABI STREQUAL "armv8")
@@ -36,7 +39,9 @@ set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR})
3639
# 4.add executable output
3740
add_executable(${TARGET} ${TARGET}.cc)
3841
target_link_libraries(${TARGET} -lpaddle_full_api_shared)
39-
42+
target_link_libraries(${TARGET} libgflags.a)
43+
target_link_libraries(${TARGET} -ldl)
44+
target_link_libraries(${TARGET} -lpthread)
4045
###############################################################
4146
# How to use one of static libaray: #
4247
# `libpaddle_api_full_bundled.a` #

lite/demo/cxx/armlinux_mobilenetv1_full_demo/mobilenet_full_api.cc

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,12 @@ void RunModel() {
4545
config.set_model_dir(FLAGS_model_dir);
4646
#ifdef DEMO_WITH_OPENCL
4747
std::vector<Place> valid_places{
48+
Place{TARGET(kOpenCL), PRECISION(kFP16), DATALAYOUT(kImageDefault)},
4849
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNCHW)},
49-
Place{TARGET(kOpenCL), PRECISION(kFloat), DATALAYOUT(kNHWC)},
50-
Place{TARGET(kARM), PRECISION(kFloat)}};
50+
Place{TARGET(kOpenCL), PRECISION(kAny), DATALAYOUT(kImageDefault)},
51+
Place{TARGET(kOpenCL), PRECISION(kAny), DATALAYOUT(kNCHW)},
52+
Place{TARGET(kOpenCL), PRECISION(kInt32), DATALAYOUT(kNCHW)},
53+
Place{TARGET(kARM)}};
5154
#else
5255
std::vector<Place> valid_places{Place{TARGET(kARM), PRECISION(kFloat)}};
5356
#endif

lite/demo/cxx/armlinux_mobilenetv1_light_demo/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR})
3636
# 4.add executable output
3737
add_executable(${TARGET} ${TARGET}.cc)
3838
target_link_libraries(${TARGET} -lpaddle_light_api_shared)
39+
target_link_libraries(${TARGET} -ldl)
3940

4041
###############################################################
4142
# How to use one of static libaray: #

0 commit comments

Comments
 (0)