danielhn commited on
Commit
e71c5f8
1 Parent(s): f141a8b

update graphs

Browse files
Files changed (1) hide show
  1. graphs.py +13 -4
graphs.py CHANGED
@@ -490,16 +490,21 @@ def speedup_bar_chart(df: pd.DataFrame, baseline) -> None:
490
  )
491
 
492
 
493
- def kpi_to_markdown(compute_ratio, device, is_baseline=False, color="blue"):
 
 
494
 
495
- title = f"""<br><br>
496
- <p style="font-family:sans-serif; font-size: 20px;text-align: center;">Median {device} Acceleration ({len(compute_ratio)} models):</p>"""
497
  if is_baseline:
 
 
498
  return (
499
  title
500
  + f"""<p style="font-family:sans-serif; color:{colors[color]}; font-size: 26px;text-align: center;"> {1}x (Baseline)</p>"""
501
  )
502
 
 
 
 
503
  if len(compute_ratio) > 0:
504
  kpi_min, kpi_median, kpi_max = (
505
  round(compute_ratio.min(), 2),
@@ -534,21 +539,25 @@ def speedup_text_summary(df: pd.DataFrame, baseline) -> None:
534
  nvidia_compute_ratio = nvidia_compute_ratio[nvidia_compute_ratio != 0]
535
  groq_compute_ratio = groq_compute_ratio[groq_compute_ratio != 0]
536
 
 
537
  x86_text = kpi_to_markdown(
538
  x86_compute_ratio,
539
  device="Intel(R) Xeon(R) X40 CPU @ 2.00GHz",
 
540
  color="blue",
541
  is_baseline=baseline == "x86",
542
  )
543
  groq_text = kpi_to_markdown(
544
  groq_compute_ratio,
545
- device="GroqChip 1",
 
546
  color="orange",
547
  is_baseline=baseline == "groq",
548
  )
549
  nvidia_text = kpi_to_markdown(
550
  nvidia_compute_ratio,
551
  device="NVIDIA A100-PCIE-40GB",
 
552
  color="green",
553
  is_baseline=baseline == "nvidia",
554
  )
 
490
  )
491
 
492
 
493
+ def kpi_to_markdown(
494
+ compute_ratio, device, num_baseline_models, is_baseline=False, color="blue"
495
+ ):
496
 
 
 
497
  if is_baseline:
498
+ title = f"""<br><br>
499
+ <p style="font-family:sans-serif; font-size: 20px;text-align: center;">Median {device} Acceleration ({len(compute_ratio)} models):</p>"""
500
  return (
501
  title
502
  + f"""<p style="font-family:sans-serif; color:{colors[color]}; font-size: 26px;text-align: center;"> {1}x (Baseline)</p>"""
503
  )
504
 
505
+ title = f"""<br><br>
506
+ <p style="font-family:sans-serif; font-size: 20px;text-align: center;">Median {device} Acceleration ({len(compute_ratio)}/{num_baseline_models} models):</p>"""
507
+
508
  if len(compute_ratio) > 0:
509
  kpi_min, kpi_median, kpi_max = (
510
  round(compute_ratio.min(), 2),
 
539
  nvidia_compute_ratio = nvidia_compute_ratio[nvidia_compute_ratio != 0]
540
  groq_compute_ratio = groq_compute_ratio[groq_compute_ratio != 0]
541
 
542
+ num_baseline_models = len(df[f"{baseline}_compute_ratio"])
543
  x86_text = kpi_to_markdown(
544
  x86_compute_ratio,
545
  device="Intel(R) Xeon(R) X40 CPU @ 2.00GHz",
546
+ num_baseline_models=num_baseline_models,
547
  color="blue",
548
  is_baseline=baseline == "x86",
549
  )
550
  groq_text = kpi_to_markdown(
551
  groq_compute_ratio,
552
+ device="GroqChip 1 Estimated",
553
+ num_baseline_models=num_baseline_models,
554
  color="orange",
555
  is_baseline=baseline == "groq",
556
  )
557
  nvidia_text = kpi_to_markdown(
558
  nvidia_compute_ratio,
559
  device="NVIDIA A100-PCIE-40GB",
560
+ num_baseline_models=num_baseline_models,
561
  color="green",
562
  is_baseline=baseline == "nvidia",
563
  )