@@ -22,12 +22,6 @@ def ProcessSummary(
2222 else :
2323 launched_queries = len (df )
2424
25- print (
26- f"Launched queries: { launched_queries } , "
27- f"pending queries: { pending_queries } , "
28- f"finished queries: { len (df )} "
29- )
30-
3125 if qps is None :
3226 qps = 0.0
3327
@@ -55,6 +49,12 @@ def ProcessSummary(
5549 average_ratio = df ['ratio' ].mean ()
5650
5751 print ("\n ==================== Performance summary ======================" )
52+ print (
53+ f" Launched queries: { launched_queries } , "
54+ f"pending queries: { pending_queries } , "
55+ f"finished queries: { len (df )} "
56+ )
57+
5858 print (f" Processing speed: { finished_qps :.4f} reqs/s" )
5959 print (f" Input tokens per second: { average_prefill_speed :.4f} tokens/s" )
6060 print (f" Output tokens per second: { average_generation_speed :.4f} tokens/s" )
@@ -103,13 +103,15 @@ def process_output(filename: str, **kwargs):
103103 print (f"Performance summary saved to { results_path } " )
104104
105105 # Save a copy of the results file to ~/srv/runner-db/
106+ print (f"Saving results to ~/srv/runner-db/{ filename_without_parent_or_ext } -{ timestamp } .results" )
106107 runner_db_path = os .path .expanduser ("~/srv/runner-db/" )
107108 os .makedirs (runner_db_path , exist_ok = True )
108109 runner_db_file = os .path .join (runner_db_path , f"{ filename_without_parent_or_ext } -{ timestamp } .results" )
109110
110111 # Copy the contents to the new location
111112 with open (results_path , "r" ) as src , open (runner_db_file , "w" ) as dst :
112113 dst .write (src .read ())
114+ print (f"Results saved to ~/srv/runner-db/{ filename_without_parent_or_ext } -{ timestamp } .results" )
113115
114116if __name__ == "__main__" :
115117 if len (sys .argv ) < 2 :
0 commit comments