提交 13007ebd 编写于 作者: Y yejianwu

fix quantize model in android example and add TODO comment for memory leak

上级 0b878466
......@@ -56,7 +56,7 @@ python tools/converter.py convert --config=mace/examples/android/mobilenet.yml -
cp -rf builds/mobilenet/include/mace/public/*.h $INCLUDE_DIR
cp -rf builds/mobilenet/model $LIBRARY_DIR
bazel build --config android --config optimization $BAZEL_LIBMACE_TARGET --define neon=true --define openmp=true --define opencl=true --cpu=$TARGET_ABI
bazel build --config android --config optimization $BAZEL_LIBMACE_TARGET --define neon=true --define openmp=true --define opencl=true --define quantize=true --cpu=$TARGET_ABI
cp -rf mace/public/*.h $INCLUDE_DIR
cp -rf $BAZEL_GEN_LIBMACE_PATH $LIBMACE_DIR
......
......@@ -68,6 +68,7 @@ mace::DeviceType ParseDeviceType(const std::string &device) {
}
MaceContext& GetMaceContext() {
// TODO(yejianwu): In multi-dlopen process, this step may cause memory leak.
static auto *mace_context = new MaceContext;
return *mace_context;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册