diff options
Diffstat (limited to 'tests/scripts/merge_result_of_benchmark_nnpkg.py')
-rwxr-xr-x | tests/scripts/merge_result_of_benchmark_nnpkg.py | 32 |
1 files changed, 16 insertions, 16 deletions
diff --git a/tests/scripts/merge_result_of_benchmark_nnpkg.py b/tests/scripts/merge_result_of_benchmark_nnpkg.py index 79c167ce4..d87e751b9 100755 --- a/tests/scripts/merge_result_of_benchmark_nnpkg.py +++ b/tests/scripts/merge_result_of_benchmark_nnpkg.py @@ -17,7 +17,7 @@ import sys, glob, os import csv import copy -from optparse import OptionParser +import argparse g_header = [] g_new_header = [] @@ -53,9 +53,9 @@ def global_init(): 'Execute_Time_Mean_Diff_Vs_Tflite_Cpu', 'Execute_Time_Mean_Ratio_Vs_Tflite_Cpu', 'Peak_RSS_Diff_Vs_Tflite_Cpu', - 'Peak_RSS_Time_Mean_Ratio_Vs_Tflite_Cpu', + 'Peak_RSS_Ratio_Vs_Tflite_Cpu', 'Peak_HWM_Diff_Vs_Tflite_Cpu', - 'Peak_HWM_Time_Mean_Ratio_Vs_Tflite_Cpu', + 'Peak_HWM_Ratio_Vs_Tflite_Cpu', ] # if new backend comes from csv, it will be stored in g_backends @@ -145,9 +145,9 @@ class Data(object): row.append(0.0) # 'Execute_Time_Mean_Diff_Vs_Tflite_Cpu' row.append(val) # 'Execute_Time_Mean_Ratio_Vs_Tflite_Cpu' row.append(0) # 'Peak_RSS_Diff_Vs_Tflite_Cpu' - row.append(val) # 'Peak_RSS_Time_Mean_Ratio_Vs_Tflite_Cpu' + row.append(val) # 'Peak_RSS_Ratio_Vs_Tflite_Cpu' row.append(0) # 'Peak_HWM_Diff_Vs_Tflite_Cpu' - row.append(val) # 'Peak_HWM_Time_Mean_Ratio_Vs_Tflite_Cpu' + row.append(val) # 'Peak_HWM_Ratio_Vs_Tflite_Cpu' return row def RowVs(self, vs_data): @@ -187,9 +187,9 @@ class Data(object): row[g_new_header.index('Execute_Time_Mean_Diff_Vs_Tflite_Cpu')] = exec_diff row[g_new_header.index('Execute_Time_Mean_Ratio_Vs_Tflite_Cpu')] = exec_ratio row[g_new_header.index('Peak_RSS_Diff_Vs_Tflite_Cpu')] = rss_diff - row[g_new_header.index('Peak_RSS_Time_Mean_Ratio_Vs_Tflite_Cpu')] = rss_ratio + row[g_new_header.index('Peak_RSS_Ratio_Vs_Tflite_Cpu')] = rss_ratio row[g_new_header.index('Peak_HWM_Diff_Vs_Tflite_Cpu')] = hwm_diff - row[g_new_header.index('Peak_HWM_Time_Mean_Ratio_Vs_Tflite_Cpu')] = hwm_ratio + row[g_new_header.index('Peak_HWM_Ratio_Vs_Tflite_Cpu')] = hwm_ratio return row @@ -215,19 +215,19 @@ class Model(object): def main(): # Option use = "Usage: %prog [options] filename" - parser = OptionParser(usage=use) - parser.add_option( + parser = argparse.ArgumentParser(usage=use) + parser.add_argument( "-i", "--input_dir", dest="input_dir", default=".", help="dir to have csv files") - parser.add_option( + parser.add_argument( "-o", "--output_dir", dest="output_dir", default=".", help="dir to be moved csv files into") - parser.add_option( + parser.add_argument( "-l", "--model_list", dest="model_list", help="file to have model list") - options, args = parser.parse_args() + options = parser.parse_args() # args check input_dir = options.input_dir @@ -255,7 +255,7 @@ def main(): if (os.path.exists(new_csv_file)): os.remove(new_csv_file) print("new csv file: {}".format(new_csv_file)) - print + print() # decl for using global vars global g_header @@ -275,7 +275,7 @@ def main(): print("* " + m) model_to_csvs[m] = [] model_data_list.append(None) - print + print() for f in glob.glob(os.path.join(input_dir, "*.csv")): # TODO handle if file name doesn't come as we follow @@ -288,7 +288,7 @@ def main(): print("Current csv file list") for model, csvs in model_to_csvs.items(): print("* {}: {}".format(model, csvs)) - print + print() for model, csvs in model_to_csvs.items(): assert (model in model_list) @@ -297,7 +297,7 @@ def main(): for model_data in model_data_list: print("{}: {}".format(model_data.model_name, model_data.backends)) - print + print() def getEmptyData(model_name, backend): d = Data(None, True) |