@@ -1253,14 +1253,14 @@ def unified_flops_metrics(
12531253 print (
12541254 f"{ dtype_prefix } "
12551255 f"Total floating-point ops: { total_flops } , Step Time (median): "
1256- f"{ average_time_ms_statistics .statistics [' p50' ]:.2f} , "
1256+ f"{ average_time_ms_statistics .statistics [" p50" ]:.2f} , "
12571257 f"Throughput (median): "
1258- f"{ tflops_per_sec_statistics .statistics [' p50' ]:.2f} "
1258+ f"{ tflops_per_sec_statistics .statistics [" p50" ]:.2f} "
12591259 f" TFLOP / second / device, "
12601260 f"TotalThroughput (median): "
1261- f"{ tflops_per_sec_all_devices_statistics .statistics [' p50' ]:.2f} "
1261+ f"{ tflops_per_sec_all_devices_statistics .statistics [" p50" ]:.2f} "
12621262 f"TFLOP / second, "
1263- f"MFU: { mfu_statistics .statistics [' p50' ]:.2%} "
1263+ f"MFU: { mfu_statistics .statistics [" p50" ]:.2%} "
12641264 )
12651265 # print()
12661266 # time_ms_list =
@@ -1340,12 +1340,12 @@ def unified_bytes_metrics(
13401340 print (
13411341 f"{ type_prefix } "
13421342 f"Total bytes: { total_bytes } , Step Time (median): "
1343- f"{ average_time_ms_statistics .statistics [' p50' ]:.2f} , "
1343+ f"{ average_time_ms_statistics .statistics [" p50" ]:.2f} , "
13441344 f"Throughput (median):"
1345- f"{ gigabytes_per_sec_statistics .statistics [' p50' ]:.2f} "
1345+ f"{ gigabytes_per_sec_statistics .statistics [" p50" ]:.2f} "
13461346 f"GBytes / second / device, "
13471347 f"TotalThroughput (median): "
1348- f"{ gigabytes_per_sec_all_devices_statistics .statistics [' p50' ]:.2f} "
1348+ f"{ gigabytes_per_sec_all_devices_statistics .statistics [" p50" ]:.2f} "
13491349 f"GBytes / second"
13501350 )
13511351 print ()
@@ -1405,5 +1405,5 @@ def get_peak_flops_multiplier(in_dtype_str: str) -> float:
14051405 return 0.25
14061406 else :
14071407 raise RuntimeError (
1408- f"{ in_dtype_lower } is not supported for setting " " peak_flops_multiplier."
1408+ f"{ in_dtype_lower } is not supported for setting peak_flops_multiplier."
14091409 )
0 commit comments