From 6c33224aa0a55294c9151913216a43cd0056d3e0 Mon Sep 17 00:00:00 2001 From: liuqi Date: Thu, 5 Apr 2018 17:31:51 +0800 Subject: [PATCH] Remove unused code. --- mace/python/tools/binary_codegen.py | 2 +- mace/python/tools/opencl_codegen.py | 3 +-- tools/mace_tools.py | 4 ++-- 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/mace/python/tools/binary_codegen.py b/mace/python/tools/binary_codegen.py index 21c6a849..3be2b086 100644 --- a/mace/python/tools/binary_codegen.py +++ b/mace/python/tools/binary_codegen.py @@ -25,7 +25,7 @@ def generate_cpp_source(): with open(binary_path, "rb") as f: binary_array = np.fromfile(f, dtype=np.uint8) - print "Generate binary", binary_path + print "Generate binary from", binary_path idx = 0 size, = struct.unpack("Q", binary_array[idx:idx+8]) idx += 8 diff --git a/mace/python/tools/opencl_codegen.py b/mace/python/tools/opencl_codegen.py index 18829635..96f4c6a6 100644 --- a/mace/python/tools/opencl_codegen.py +++ b/mace/python/tools/opencl_codegen.py @@ -19,11 +19,10 @@ def generate_cpp_source(): binary_dirs = FLAGS.cl_binary_dirs.strip().split(",") for binary_dir in binary_dirs: binary_path = os.path.join(binary_dir, FLAGS.built_kernel_file_name) - print 'Before generate opencl code:', binary_path if not os.path.exists(binary_path): continue - print 'generate opencl code:', binary_path + print 'generate opencl code from', binary_path with open(binary_path, "rb") as f: binary_array = np.fromfile(f, dtype=np.uint8) diff --git a/tools/mace_tools.py b/tools/mace_tools.py index d739bc41..002e430e 100644 --- a/tools/mace_tools.py +++ b/tools/mace_tools.py @@ -156,7 +156,7 @@ def tuning_run(model_name, stdout_buff = [] process_output = sh_commands.make_output_processor(stdout_buff) p = sh.bash("tools/tuning_run.sh", target_soc, model_output_dir, - running_round, int(tuning), int(production_mode), + running_round, int(tuning), restart_round, option_args, _out=process_output, _bg=True, _err_to_out=True) p.wait() @@ -188,7 +188,7 @@ def benchmark_model(target_soc, model_output_dir, option_args=''): def run_model(model_name, target_runtime, target_abi, target_soc, model_output_dir, running_round, restart_round, option_args): tuning_run(model_name, target_runtime, target_abi, target_soc, - model_output_dir, running_round, False, False, + model_output_dir, running_round, False, restart_round, option_args) -- GitLab