diff --git a/mace/python/tools/binary_codegen.py b/mace/python/tools/binary_codegen.py index 21c6a84929e3527dd2b1e97feb4e9473d7a03ffa..3be2b086461dd70e7fa6066f086d91442601dec5 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 18829635bce120a5dc9d912d4f66f76ef8e08239..96f4c6a6d11b27c60f01b83e35ecd47bca863e05 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 d739bc41f873646adb3f8cb7a572134a763f9cb1..002e430e744d3bac46100643bee30f1f3c7e2ff9 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)