Log the results at the end too

bench-TriLMs.py

@@ -208,5 +208,8 @@ if __name__ == "__main__":
         "mulmat_perf": mulmat_perf,
         "results": results,
     }
+    logger.info("Writing output to: %s", args.out)
+    logger.debug("Final results: %s", json.dumps(final_result, indent=4))
     with open(args.out, "w") as f:
-        json.dump(results, f, indent=4)
+        json.dump(final_result, f, indent=4)
+        f.flush()