add load datatypes filtering

app.py

@@ -16,7 +16,7 @@ OPTIMUM_TOKEN = os.environ.get("OPTIMUM_TOKEN", None)
 COLUMNS_MAPPING = {
     "model": "Model 🤗",
     "backend.name": "Backend 🏭",
-    "backend.torch_dtype": "Load Datatype 📥",
+    "backend.torch_dtype": "Datatype 📥",
     "generate.latency(s)": "Latency (s) ⬇️",
     "generate.throughput(tokens/s)": "Throughput (tokens/s) ⬆️",
 }
@@ -66,9 +66,15 @@ def search_tables(single_df, multi_df, query):
     return filtered_single, filtered_multi
 
 
-def filter_tables(single_df, multi_df, backends):
+def filter_tables(single_df, multi_df, backends, datatypes):
     filtered_single = single_df[single_df["Backend 🏭"].isin(backends)]
+    filtered_single = filtered_single[filtered_single["Datatype 📥"].isin(
+        datatypes)]
+
     filtered_multi = multi_df[multi_df["Backend 🏭"].isin(backends)]
+    filtered_multi = filtered_multi[filtered_multi["Datatype 📥"].isin(
+        datatypes)]
+
     return filtered_single, filtered_multi
 
 
@@ -93,6 +99,13 @@ with demo:
             info="Select the backends",
             elem_id="backend-checkboxes",
         )
+        datatype_checkboxes = gr.CheckboxGroup(
+            choices=["float32", "float16"],
+            value=["float32", "float16"],
+            label="Datatypes 📥",
+            info="Select the load datatypes",
+            elem_id="datatype-checkboxes",
+        )
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🖥️ A100-80GB Benchmark 🏋️", elem_id="A100-benchmark", id=0):
@@ -151,7 +164,14 @@ with demo:
             )
             backend_checkboxes.select(
                 filter_tables,
-                [single_A100_for_search, multi_A100_for_search, backend_checkboxes],
+                [single_A100_for_search, multi_A100_for_search,
+                    backend_checkboxes, datatype_checkboxes],
+                [single_A100_leaderboard, multi_A100_leaderboard],
+            )
+            datatype_checkboxes.select(
+                filter_tables,
+                [single_A100_for_search, multi_A100_for_search,
+                    backend_checkboxes, datatype_checkboxes],
                 [single_A100_leaderboard, multi_A100_leaderboard],
             )