diff --git a/convbench/conv_bench.py b/convbench/conv_bench.py index 3adac44..4ad0216 100644 --- a/convbench/conv_bench.py +++ b/convbench/conv_bench.py @@ -125,6 +125,7 @@ def compile_conv(tag, config, kernel_dir, vmfb_dir, extra_compiler_args): tflops_per_second = (flops / 1e12) / (benchmark_conv_mean_time_us / 1e6) # Compute percentage of the roofline. + # TODO: Make this target specific and move to common utils. tflops_map = { "f32": 653.7, "f16": 1307.4,