提交 78dabbed 编写于 作者: Y yejianwu

update build flow for non-gpu runtime

上级 d3ccb99b
...@@ -199,41 +199,51 @@ def build_mace_run_prod(hexagon_mode, runtime, target_abi, ...@@ -199,41 +199,51 @@ def build_mace_run_prod(hexagon_mode, runtime, target_abi,
input_shapes, output_shapes, model_name, device_type, input_shapes, output_shapes, model_name, device_type,
running_round, restart_round, tuning, running_round, restart_round, tuning,
limit_opencl_kernel_time, phone_data_dir): limit_opencl_kernel_time, phone_data_dir):
gen_opencl_and_tuning_code(target_abi, serialno, [], False)
production_or_not = False
mace_run_target = "//mace/tools/validation:mace_run" mace_run_target = "//mace/tools/validation:mace_run"
if runtime == "gpu":
gen_opencl_and_tuning_code(target_abi, serialno, [], False)
sh_commands.bazel_build( sh_commands.bazel_build(
mace_run_target, mace_run_target,
abi=target_abi, abi=target_abi,
model_tag=model_name, model_tag=model_name,
production_mode=False, production_mode=False,
hexagon_mode=hexagon_mode) hexagon_mode=hexagon_mode)
sh_commands.update_mace_run_lib(model_output_dir, target_abi, model_name, sh_commands.update_mace_run_lib(model_output_dir, target_abi,
embed_model_data) model_name, embed_model_data)
tuning_run(runtime, target_abi, serialno, vlog_level, tuning_run(runtime, target_abi, serialno, vlog_level, embed_model_data,
embed_model_data, model_output_dir, input_nodes, output_nodes, model_output_dir, input_nodes, output_nodes, input_shapes,
input_shapes, output_shapes, model_name, device_type, output_shapes, model_name, device_type, running_round=0,
running_round=0, restart_round=1, out_of_range_check=True, restart_round=1, out_of_range_check=True,
phone_data_dir=phone_data_dir, tuning=False) phone_data_dir=phone_data_dir, tuning=False)
tuning_run(runtime, target_abi, serialno, vlog_level, tuning_run(runtime, target_abi, serialno, vlog_level, embed_model_data,
embed_model_data, model_output_dir, input_nodes, output_nodes, model_output_dir, input_nodes, output_nodes, input_shapes,
input_shapes, output_shapes, model_name, device_type, output_shapes, model_name, device_type, running_round=0,
running_round=0, restart_round=1, out_of_range_check=False, restart_round=1, out_of_range_check=False,
phone_data_dir=phone_data_dir, tuning=tuning, phone_data_dir=phone_data_dir, tuning=tuning,
limit_opencl_kernel_time=limit_opencl_kernel_time) limit_opencl_kernel_time=limit_opencl_kernel_time)
gen_opencl_and_tuning_code(target_abi, serialno, [model_output_dir], True) gen_opencl_and_tuning_code(target_abi, serialno, [model_output_dir],
production_or_not = True True)
sh_commands.bazel_build( sh_commands.bazel_build(
mace_run_target, mace_run_target,
abi=target_abi, abi=target_abi,
model_tag=model_name, model_tag=model_name,
production_mode=True, production_mode=True,
hexagon_mode=hexagon_mode) hexagon_mode=hexagon_mode)
sh_commands.update_mace_run_lib(model_output_dir, target_abi, model_name, sh_commands.update_mace_run_lib(model_output_dir, target_abi,
embed_model_data) model_name, embed_model_data)
else:
gen_opencl_and_tuning_code(target_abi, serialno, [], False)
sh_commands.bazel_build(
mace_run_target,
abi=target_abi,
model_tag=model_name,
production_mode=True,
hexagon_mode=hexagon_mode)
sh_commands.update_mace_run_lib(model_output_dir, target_abi,
model_name, embed_model_data)
def merge_libs_and_tuning_results(target_soc, def merge_libs_and_tuning_results(target_soc,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册