def main():
config_dict = create_config("tnn")
tnn = Engine(config_dict)
model_dict = tnn.prepare_models()
device_dict = tnn.prepare_devices()
tnn.set_config("model_dict", model_dict)
tnn.set_config("device_dict", device_dict)
tnn.prepare_models_for_devices()
tnn.prepare_benchmark_assets_for_devices()
bench_dict = tnn.run_bench()
bench_str_list = tnn.generate_benchmark_summary(bench_dict)
tnn.write_list_to_file(bench_str_list)
def bench(name):
if name is None:
return
config_dict = create_config(name)
engine = Engine(config_dict)
model_dict = engine.prepare_models()
device_dict = engine.prepare_devices()
engine.set_config("model_dict", model_dict)
engine.set_config("device_dict", device_dict)
engine.prepare_models_for_devices()
engine.prepare_benchmark_assets_for_devices()
bench_dict = engine.run_bench()
bench_str_list = engine.generate_benchmark_summary(bench_dict)
engine.write_list_to_file(bench_str_list)