Check all flags by default, full-width outputs, simpler examples

app.py

@@ -17,7 +17,6 @@ FLAGS = {
     "cuobjdump": ["-sass", "-ptx", "-elf", "-res-usage", "-symbols", "-lelf", "-lptx"],
     "nvdisasm": ["-c", "-g", "-gi", "-hex", "-print-life-ranges", "-print-raw", "-print-line-info", "-cfg", "-bbcfg"],
 }
-DEFAULT_FLAGS = {"cuobjdump": ["-sass"], "nvdisasm": ["-c"]}
 EXAMPLES = sorted(str(p) for p in Path(__file__).with_name("examples").glob("*.cubin"))
 
 
@@ -30,7 +29,7 @@ def run_one(tool, args, file):
         r = subprocess.run([TOOLS[tool], *args, file], capture_output=True, text=True, timeout=60)
     except subprocess.TimeoutExpired:
         return "Timed out after 60s."
-    return r.stdout + r.stderr + (f"\n[exit code {r.returncode}]" if r.returncode else "")
+    return (r.stdout + r.stderr or "(no output)") + (f"\n[exit code {r.returncode}]" if r.returncode else "")
 
 
 def run(file, tool, flags, extra):
@@ -43,24 +42,26 @@ def run(file, tool, flags, extra):
 
 with gr.Blocks(title="cuobjdump / nvdisasm") as demo:
     gr.Markdown("# cuobjdump / nvdisasm\n"
-                "Disassemble CUDA cubins, fatbins and ELF objects with NVIDIA's tools from PyPI.\n\n"
-                + "\n".join(f"- `{t}`: {version(t)}" for t in TOOLS))
-    with gr.Row():
-        with gr.Column():
+                "Disassemble CUDA cubins, fatbins and ELF objects with NVIDIA's tools from PyPI ("
+                + ", ".join(f"`{t}` {version(t)}" for t in TOOLS) + ").")
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=1):
             file = gr.File(label="cubin / fatbin / CUDA ELF", type="filepath")
-            tool = gr.Radio(list(TOOLS), value="cuobjdump", label="Tool")
-            flags = gr.CheckboxGroup(FLAGS["cuobjdump"], value=DEFAULT_FLAGS["cuobjdump"], label="Flags (each runs separately with its own output)")
-            extra = gr.Textbox(label="Extra arguments", placeholder="-fun vecadd")
-            btn = gr.Button("Run", variant="primary")
         with gr.Column(scale=2):
-            results = gr.State({})
-
-            @gr.render(inputs=results)
-            def show(results):
-                for label, output in results.items():
-                    gr.Code(output, label=label, max_lines=60)
-    gr.Examples([[e, t] for e in EXAMPLES for t in TOOLS], [file, tool])
-    tool.change(lambda t: gr.CheckboxGroup(FLAGS[t], value=DEFAULT_FLAGS[t]), tool, flags)
+            tool = gr.Radio(list(TOOLS), value="cuobjdump", label="Tool")
+            flags = gr.CheckboxGroup(FLAGS["cuobjdump"], value=FLAGS["cuobjdump"], label="Flags (each runs separately with its own output)")
+            with gr.Row(equal_height=True):
+                extra = gr.Textbox(label="Extra arguments", placeholder="-fun vecadd", scale=4)
+                btn = gr.Button("Run", variant="primary", scale=1)
+    gr.Examples(EXAMPLES, file)
+    results = gr.State({})
+
+    @gr.render(inputs=results)
+    def show(results):
+        for label, output in results.items():
+            gr.Code(output, label=label, max_lines=60)
+
+    tool.change(lambda t: gr.CheckboxGroup(FLAGS[t], value=FLAGS[t]), tool, flags)
     btn.click(run, [file, tool, flags, extra], results)
 
 if __name__ == "__main__":