diff --git a/mace/libmace/mace_version_script.lds b/mace/libmace/mace_version_script.lds index 4bdc33db9f6162924285c52ecc9bbe76435d487d..424cef31b2ed2def8b00acfe29eee6ceeb1a4b53 100644 --- a/mace/libmace/mace_version_script.lds +++ b/mace/libmace/mace_version_script.lds @@ -11,6 +11,8 @@ mace { # api for static library of models *mace*logging*LogMessage*; *mace*NetDef*; + *mace*MemoryType*; + *mace*DataType*; *mace*MemoryArena*; *mace*InputInfo*; *mace*OutputInfo*; diff --git a/tools/converter.py b/tools/converter.py index d10861055df7cb81a26e35460390025b91fcd0e5..c2469d7b10ed908e6559e0323cbd2137f06496b8 100644 --- a/tools/converter.py +++ b/tools/converter.py @@ -977,7 +977,8 @@ def build_quantize_stat(configs): build_tmp_binary_dir) -def build_example(configs, target_abi, enable_openmp, mace_lib_type): +def build_example(configs, target_abi, enable_openmp, address_sanitizer, + mace_lib_type): library_name = configs[YAMLKeyword.library_name] hexagon_mode = get_hexagon_mode(configs) @@ -997,6 +998,7 @@ def build_example(configs, target_abi, enable_openmp, mace_lib_type): enable_openmp=enable_openmp, enable_opencl=get_opencl_mode(configs), hexagon_mode=hexagon_mode, + address_sanitizer=address_sanitizer, symbol_hidden=symbol_hidden) if os.path.exists(LIB_CODEGEN_DIR): @@ -1025,6 +1027,7 @@ def build_example(configs, target_abi, enable_openmp, mace_lib_type): enable_openmp=enable_openmp, enable_opencl=get_opencl_mode(configs), hexagon_mode=hexagon_mode, + address_sanitizer=address_sanitizer, extra_args=build_arg) target_bin = "/".join(sh_commands.bazel_target_to_bin(example_target)) @@ -1395,6 +1398,7 @@ def run_mace(flags): if flags.example: build_example(configs, target_abi, not flags.disable_openmp, + flags.address_sanitizer, flags.mace_lib_type) else: build_mace_run(configs, target_abi,