未验证 提交 4fec182d 编写于 作者: W Wilber 提交者: GitHub

[Lite-Subgraph] Fix compile error for lite subgraph. (#29146)

上级 f23665e5
...@@ -82,10 +82,8 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR) ...@@ -82,10 +82,8 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR)
${EXTERNAL_OPTIONAL_ARGS} ${EXTERNAL_OPTIONAL_ARGS}
${LITE_OPTIONAL_ARGS} ${LITE_OPTIONAL_ARGS}
) )
set(LITE_OUTPUT_BIN_DIR inference_lite_lib.armlinux.armv8)
else() else()
set(LITE_BUILD_COMMAND $(MAKE) publish_inference -j) set(LITE_BUILD_COMMAND $(MAKE) publish_inference -j)
set(LITE_OUTPUT_BIN_DIR inference_lite_lib)
set(LITE_OPTIONAL_ARGS -DWITH_MKL=ON set(LITE_OPTIONAL_ARGS -DWITH_MKL=ON
-DLITE_WITH_CUDA=${WITH_GPU} -DLITE_WITH_CUDA=${WITH_GPU}
-DWITH_MKLDNN=OFF -DWITH_MKLDNN=OFF
...@@ -133,6 +131,12 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR) ...@@ -133,6 +131,12 @@ if (NOT LITE_SOURCE_DIR OR NOT LITE_BINARY_DIR)
endif() endif()
if (WITH_ARM)
set(LITE_OUTPUT_BIN_DIR inference_lite_lib.armlinux.armv8)
else()
set(LITE_OUTPUT_BIN_DIR inference_lite_lib)
endif()
message(STATUS "Paddle-lite BINARY_DIR: ${LITE_BINARY_DIR}") message(STATUS "Paddle-lite BINARY_DIR: ${LITE_BINARY_DIR}")
message(STATUS "Paddle-lite SOURCE_DIR: ${LITE_SOURCE_DIR}") message(STATUS "Paddle-lite SOURCE_DIR: ${LITE_SOURCE_DIR}")
include_directories(${LITE_SOURCE_DIR}) include_directories(${LITE_SOURCE_DIR})
......
...@@ -55,8 +55,7 @@ paddle::lite_api::PaddlePredictor* EngineManager::Create( ...@@ -55,8 +55,7 @@ paddle::lite_api::PaddlePredictor* EngineManager::Create(
#ifdef PADDLE_WITH_ARM #ifdef PADDLE_WITH_ARM
lite_cxx_config.set_threads(cfg.cpu_math_library_num_threads); lite_cxx_config.set_threads(cfg.cpu_math_library_num_threads);
#else #else
lite_cxx_config.set_x86_math_library_num_threads( lite_cxx_config.set_x86_math_num_threads(cfg.cpu_math_library_num_threads);
cfg.cpu_math_library_num_threads);
#endif #endif
#ifdef LITE_SUBGRAPH_WITH_XPU #ifdef LITE_SUBGRAPH_WITH_XPU
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册