From a031afb2f6d2b7cd16188b4f18ddd8555cb46cbf Mon Sep 17 00:00:00 2001 From: Akshay Kashyap Date: Sun, 26 Nov 2023 19:07:36 -0500 Subject: [PATCH] Update display_name in resnet50 example (#2454) --- examples/handcode_resnet50_opt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/handcode_resnet50_opt.py b/examples/handcode_resnet50_opt.py index 43c2caaf22..e67749af21 100644 --- a/examples/handcode_resnet50_opt.py +++ b/examples/handcode_resnet50_opt.py @@ -64,9 +64,9 @@ if __name__ == "__main__": choices.append((tm, gflops, lin.linearize())) # print all kernels - if DEBUG >= 1: print(f" kernel {i:2d} {lin.display_name+' '*(37-ansilen(lin.display_name))} {str(lin.global_size):18s} {str(lin.local_size):12s} takes {tm*1000:7.2f} ms, {gflops:6.0f} GFLOPS") + if DEBUG >= 1: print(f" kernel {i:2d} {lin.name+' '*(37-ansilen(lin.name))} {str(lin.global_size):18s} {str(lin.local_size):12s} takes {tm*1000:7.2f} ms, {gflops:6.0f} GFLOPS") tm, gflops, lin = sorted(choices, key=lambda x: x[0])[0] - print(f"*** {total_tm*1000:7.2f} ms : kernel {i:2d} {lin.display_name+' '*(37-ansilen(lin.display_name))} {str(lin.global_size):18s} {str(lin.local_size):12s} takes {tm*1000:7.2f} ms, {gflops:6.0f} GFLOPS") + print(f"*** {total_tm*1000:7.2f} ms : kernel {i:2d} {lin.name+' '*(37-ansilen(lin.name))} {str(lin.global_size):18s} {str(lin.local_size):12s} takes {tm*1000:7.2f} ms, {gflops:6.0f} GFLOPS") total_tm += tm running_gflops += gflops * tm print(f"******* total {total_tm*1000:.2f} ms, {running_gflops/total_tm:6.0f} GFLOPS")