Update app.py

app.py

@@ -1,6 +1,3 @@
-import signal
-import os
-
 import gradio as gr
 
 import src.dependency  # noqa
@@ -8,17 +5,17 @@ from src.assets import custom_css
 
 # from src.attention import create_attn_plots
 from src.content import ABOUT, CITATION_BUTTON, CITATION_BUTTON_LABEL, LOGO, TITLE
-from src.hardware import load_hardware_configs
-from src.leaderboard import create_leaderboard_table
-from src.llm_perf import get_llm_perf_df
-from src.map import create_lat_score_mem_plot
-from src.panel import (
-    create_control_callback,
-    create_control_panel,
-    create_select_callback,
-)
+# from src.hardware import load_hardware_configs
+# from src.leaderboard import create_leaderboard_table
+# from src.llm_perf import get_llm_perf_df
+# from src.map import create_lat_score_mem_plot
+# from src.panel import (
+#     create_control_callback,
+#     create_control_panel,
+#     create_select_callback,
+# )
 
-configs = load_hardware_configs("hardware.yaml")
+# configs = load_hardware_configs("hardware.yaml")
 
 
 demo = gr.Blocks(
@@ -28,122 +25,115 @@ demo = gr.Blocks(
 with demo:
     gr.HTML(LOGO, elem_classes="logo")
     gr.HTML(TITLE, elem_classes="title")
-    ####################### HARDWARE TABS #######################
-    with gr.Tabs(elem_classes="tabs"):
-        for id, config in enumerate(configs):
-            with gr.TabItem(config.description, id=id):
-                ####################### HARDWARE DETAILS #######################
-                if config.detail:
-                    gr.Markdown(config.detail, elem_classes="descriptive-text")
-
-                # ####################### CONTROL PANEL #######################
-                (
-                    filter_button,
-                    machine_value,
-                    subsets_value,
-                    backends_value,
-                    hardware_type_value,
-                    score_slider,
-                    memory_slider,
-                    backend_checkboxes,
-                    datatype_checkboxes,
-                    optimization_checkboxes,
-                    quantization_checkboxes,
-                    kernels_checkboxes,
-                ) = create_control_panel(
-                    machine=config.machine,
-                    subsets=config.subsets,
-                    backends=config.backends,
-                    hardware_type=config.hardware_type,
-                    hardware_provider=config.hardware_provider,
-                )
-                ####################### HARDWARE SUBTABS #######################
-                with gr.Tabs(elem_classes="subtabs"):
-                    open_llm_perf_df = get_llm_perf_df(
-                        machine=config.machine,
-                        subsets=config.subsets,
-                        backends=config.backends,
-                        hardware_type=config.hardware_type,
-                    )
-                    ####################### LEADERBOARD TAB #######################
-                    with gr.TabItem("Leaderboard 🏅", id=0):
-                        search_bar, columns_checkboxes, leaderboard_table = (
-                            create_leaderboard_table(open_llm_perf_df)
-                        )
-                    with gr.TabItem("Find Your Best Model 🧭", id=1):
-                        lat_score_mem_plot = create_lat_score_mem_plot(
-                            open_llm_perf_df
-                        )
-                    ###################### ATTENTIONS SPEEDUP TAB #######################
-                    # with gr.TabItem("Attention 📈", id=2):
-                    #     attn_prefill_plot, attn_decode_plot = create_attn_plots(
-                    #         open_llm_perf_df
-                    #     )
-                    # ####################### KERNELS SPEEDUP TAB #######################
-                    # with gr.TabItem("Kernels 📈", id=4):
-                    #     quant_krnl_prefill_plot, quant_krnl_decode_plot = (
-                    #         create_quant_krnl_plots(llm_perf_df)
-                    #     )
+    # ####################### HARDWARE TABS #######################
+    # with gr.Tabs(elem_classes="tabs"):
+    #     for id, config in enumerate(configs):
+    #         with gr.TabItem(config.description, id=id):
+    #             ####################### HARDWARE DETAILS #######################
+    #             if config.detail:
+    #                 gr.Markdown(config.detail, elem_classes="descriptive-text")
 
-                ####################### CONTROL CALLBACK #######################
-                create_control_callback(
-                    filter_button,
-                    # inputs
-                    machine_value,
-                    subsets_value,
-                    backends_value,
-                    hardware_type_value,
-                    score_slider,
-                    memory_slider,
-                    backend_checkboxes,
-                    datatype_checkboxes,
-                    optimization_checkboxes,
-                    quantization_checkboxes,
-                    kernels_checkboxes,
-                    # interactive
-                    columns_checkboxes,
-                    search_bar,
-                    # outputs
-                    leaderboard_table,
-                    lat_score_mem_plot,
-                    # attn_prefill_plot,
-                    # attn_decode_plot,
-                    # quant_krnl_prefill_plot,
-                    # quant_krnl_decode_plot,
-                )
+    #             # ####################### CONTROL PANEL #######################
+    #             (
+    #                 filter_button,
+    #                 machine_value,
+    #                 subsets_value,
+    #                 backends_value,
+    #                 hardware_type_value,
+    #                 score_slider,
+    #                 memory_slider,
+    #                 backend_checkboxes,
+    #                 datatype_checkboxes,
+    #                 optimization_checkboxes,
+    #                 quantization_checkboxes,
+    #                 kernels_checkboxes,
+    #             ) = create_control_panel(
+    #                 machine=config.machine,
+    #                 subsets=config.subsets,
+    #                 backends=config.backends,
+    #                 hardware_type=config.hardware_type,
+    #                 hardware_provider=config.hardware_provider,
+    #             )
+    #             ####################### HARDWARE SUBTABS #######################
+    #             with gr.Tabs(elem_classes="subtabs"):
+    #                 open_llm_perf_df = get_llm_perf_df(
+    #                     machine=config.machine,
+    #                     subsets=config.subsets,
+    #                     backends=config.backends,
+    #                     hardware_type=config.hardware_type,
+    #                 )
+    #                 ####################### LEADERBOARD TAB #######################
+    #                 with gr.TabItem("Leaderboard 🏅", id=0):
+    #                     search_bar, columns_checkboxes, leaderboard_table = (
+    #                         create_leaderboard_table(open_llm_perf_df)
+    #                     )
+    #                 with gr.TabItem("Find Your Best Model 🧭", id=1):
+    #                     lat_score_mem_plot = create_lat_score_mem_plot(
+    #                         open_llm_perf_df
+    #                     )
+    #                 ###################### ATTENTIONS SPEEDUP TAB #######################
+    #                 # with gr.TabItem("Attention 📈", id=2):
+    #                 #     attn_prefill_plot, attn_decode_plot = create_attn_plots(
+    #                 #         open_llm_perf_df
+    #                 #     )
+    #                 # ####################### KERNELS SPEEDUP TAB #######################
+    #                 # with gr.TabItem("Kernels 📈", id=4):
+    #                 #     quant_krnl_prefill_plot, quant_krnl_decode_plot = (
+    #                 #         create_quant_krnl_plots(llm_perf_df)
+    #                 #     )
 
-                create_select_callback(
-                    # inputs
-                    machine_value,
-                    subsets_value,
-                    backends_value,
-                    hardware_type_value,
-                    # interactive
-                    columns_checkboxes,
-                    search_bar,
-                    # outputs
-                    leaderboard_table,
-                )
+    #             ####################### CONTROL CALLBACK #######################
+    #             create_control_callback(
+    #                 filter_button,
+    #                 # inputs
+    #                 machine_value,
+    #                 subsets_value,
+    #                 backends_value,
+    #                 hardware_type_value,
+    #                 score_slider,
+    #                 memory_slider,
+    #                 backend_checkboxes,
+    #                 datatype_checkboxes,
+    #                 optimization_checkboxes,
+    #                 quantization_checkboxes,
+    #                 kernels_checkboxes,
+    #                 # interactive
+    #                 columns_checkboxes,
+    #                 search_bar,
+    #                 # outputs
+    #                 leaderboard_table,
+    #                 lat_score_mem_plot,
+    #                 # attn_prefill_plot,
+    #                 # attn_decode_plot,
+    #                 # quant_krnl_prefill_plot,
+    #                 # quant_krnl_decode_plot,
+    #             )
 
-        ####################### ABOUT TAB #######################
-        with gr.TabItem("About 📖", id=len(configs)):
-            gr.Markdown(ABOUT, elem_classes="descriptive-text")
-    ####################### CITATION
-    with gr.Row():
-        with gr.Accordion("📙 Citation", open=False):
-            citation_button = gr.Textbox(
-                value=CITATION_BUTTON,
-                label=CITATION_BUTTON_LABEL,
-                elem_id="citation-button",
-                show_copy_button=True,
-            )
+    #             create_select_callback(
+    #                 # inputs
+    #                 machine_value,
+    #                 subsets_value,
+    #                 backends_value,
+    #                 hardware_type_value,
+    #                 # interactive
+    #                 columns_checkboxes,
+    #                 search_bar,
+    #                 # outputs
+    #                 leaderboard_table,
+    #             )
 
-def tracefunc(frame, event, arg):
-    if event == 'call':
-        print(f"Calling function: {frame.f_code.co_name}")
-    return tracefunc
+    #     ####################### ABOUT TAB #######################
+    #     with gr.TabItem("About 📖", id=len(configs)):
+    #         gr.Markdown(ABOUT, elem_classes="descriptive-text")
+    # ####################### CITATION
+    # with gr.Row():
+    #     with gr.Accordion("📙 Citation", open=False):
+    #         citation_button = gr.Textbox(
+    #             value=CITATION_BUTTON,
+    #             label=CITATION_BUTTON_LABEL,
+    #             elem_id="citation-button",
+    #             show_copy_button=True,
+    #         )
 
 if __name__ == "__main__":
-    import sys
-    sys.settrace(tracefunc)
     demo.queue().launch(ssr_mode=False)