Add path to generate profile graph image (#2451)

* Add path to generate prfile graph image * Fix pre-commit --------- Co-authored-by: Ankith Gunapal <[email protected]>
pytorch · Jul 13, 2023 · 2f5a784 · 2f5a784
1 parent d31b6c3
commit 2f5a784
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/benchmarks/benchmark-ab.py b/benchmarks/benchmark-ab.py
@@ -659,7 +659,10 @@ def plot_line(fig, data, color="blue", title=None):
         title="Combined Graph",
     )
     fig5.grid()
-    plt.savefig("api-profile1.png", bbox_inches="tight")
+    plt.savefig(
+        f"{execution_params['report_location']}/benchmark/api-profile1.png",
+        bbox_inches="tight",
+    )
 
 
 def stop_torchserve():