未验证 提交 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)
if (LITE_WITH_OPENCL)
set(INFER_LITE_PUBLISH_ROOT "${INFER_LITE_PUBLISH_ROOT}.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}")
# The final target for publish lite lib
......
......@@ -28,6 +28,7 @@ lite_cc_library(model_parser SRCS model_parser.cc DEPS
target_wrapper_host
compatible_pb
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)
......@@ -107,7 +107,14 @@ function build_npu {
test_name=$6
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
cd $build_dir
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册