Fix sdpa backend legend

bar_plot.py

@@ -23,11 +23,20 @@ def reorder_data(per_scenario_data: dict) -> dict:
 
 def infer_bar_label(config: dict) -> str:
     """Format legend labels to be more readable."""
-    attn_implementation = {
-        "flash_attention_2": "Flash attention",
-        "sdpa": "SDPA",
-        "eager": "Eager",
-    }[config["attn_implementation"]]
+    if config["attn_implementation"] == "eager":
+        attn_implementation = "Eager"
+    elif config["attn_implementation"] == "flash_attention_2":
+        attn_implementation = "Flash attention"
+    elif config["attn_implementation"] == "sdpa":
+        attn_implementation = {
+            "flash_attention": "SDPA (flash attention)",
+            "efficient_attention": "SDPA (efficient_attention)",
+            "cudnn_attention": "SDPA (cudnn)",
+            "math": "SDPA (math)",
+        }.get(config["sdpa_backend"], "SDPA (unknown backend)")
+    else:
+        attn_implementation = "Unknown"
+
     compile = "compiled" if config["compilation"] else "no compile"
     kernels = "kernelized" if config["kernelize"] else "no kernels"
     return f"{attn_implementation}, {compile}, {kernels}"