未验证 提交 443b2e68 编写于 作者: T tensor-tang 提交者: GitHub

[NPU] fix npu compile of publish_inference (#1911)

上级 57ee8714
...@@ -59,6 +59,9 @@ if (LITE_WITH_LIGHT_WEIGHT_FRAMEWORK AND LITE_WITH_ARM) ...@@ -59,6 +59,9 @@ if (LITE_WITH_LIGHT_WEIGHT_FRAMEWORK AND LITE_WITH_ARM)
if (LITE_WITH_OPENCL) if (LITE_WITH_OPENCL)
set(INFER_LITE_PUBLISH_ROOT "${INFER_LITE_PUBLISH_ROOT}.opencl") set(INFER_LITE_PUBLISH_ROOT "${INFER_LITE_PUBLISH_ROOT}.opencl")
endif(LITE_WITH_OPENCL) endif(LITE_WITH_OPENCL)
if (LITE_WITH_NPU)
set(INFER_LITE_PUBLISH_ROOT "${INFER_LITE_PUBLISH_ROOT}.npu")
endif(LITE_WITH_NPU)
message(STATUS "publish inference lib to ${INFER_LITE_PUBLISH_ROOT}") message(STATUS "publish inference lib to ${INFER_LITE_PUBLISH_ROOT}")
# The final target for publish lite lib # The final target for publish lite lib
......
...@@ -28,6 +28,7 @@ lite_cc_library(model_parser SRCS model_parser.cc DEPS ...@@ -28,6 +28,7 @@ lite_cc_library(model_parser SRCS model_parser.cc DEPS
target_wrapper_host target_wrapper_host
compatible_pb compatible_pb
memory memory
CUDA_DEPS target_wrapper_cuda) CUDA_DEPS target_wrapper_cuda
NPU_DEPS npu_helper)
lite_cc_test(test_compatible_pb SRCS compatible_pb_test.cc DEPS compatible_pb) lite_cc_test(test_compatible_pb SRCS compatible_pb_test.cc DEPS compatible_pb)
...@@ -107,7 +107,14 @@ function build_npu { ...@@ -107,7 +107,14 @@ function build_npu {
test_name=$6 test_name=$6
fi fi
build_dir=$cur_dir/build.lite.npu.${os}.${abi}.${lang}.${stl} # the c++ symbol is not recognized by the bundled script
if [[ "${stl}" == "c++_shared" ]]; then
stl_dir="cxx_shared"
fi
if [[ "${stl}" == "c++_static" ]]; then
stl_dir="cxx_static"
fi
build_dir=$cur_dir/build.lite.npu.${os}.${abi}.${lang}.${stl_dir}
mkdir -p $build_dir mkdir -p $build_dir
cd $build_dir cd $build_dir
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册