mila-iqia · Delaunay · Sep 16, 2024
diff --git a/benchmarks/accelerate_opt/requirements.rocm.txt b/benchmarks/accelerate_opt/requirements.rocm.txt
diff --git a/benchmarks/huggingface/requirements.rocm.txt b/benchmarks/huggingface/requirements.rocm.txt
diff --git a/benchmarks/timm/requirements.rocm.txt b/benchmarks/timm/requirements.rocm.txt
diff --git a/benchmarks/torchvision/requirements.rocm.txt b/benchmarks/torchvision/requirements.rocm.txt
diff --git a/milabench/report.py b/milabench/report.py
@@ -272,6 +272,29 @@ def sort_by(key):
             for key in all_keys
         }
     ).transpose()
+
+    return df
+
+
+@error_guard({})
+def make_report(
+    summary,
+    compare=None,
+    html=None,
+    compare_gpus=False,
+    price=None,
+    title=None,
+    sources=None,
+    errdata=None,
+    weights=None,
+):
+    if weights is None:
+        weights = dict()
+
+    df = make_dataframe(summary, compare, weights)
+
+    # Reorder columns
+    df = df[sorted(df.columns, key=lambda k: columns_order.get(k, 0))]
 
     # Reorder columns
     df = df[sorted(df.columns, key=lambda k: columns_order.get(k, 2000))]