Standardize plot_name values across scripts

Author: voltjia

add.py

@@ -41,7 +41,7 @@
             line_names=["NineToothed", "PyTorch", "Triton"],
             styles=[("blue", "-"), ("green", "-"), ("orange", "-")],
             ylabel="ms",
-            plot_name="vector-addition-performance",
+            plot_name="add-performance",
             args={},
         )
     )

bmm.py

@@ -41,7 +41,7 @@
             line_names=["NineToothed", "PyTorch", "Triton"],
             styles=[("blue", "-"), ("green", "-"), ("orange", "-")],
             ylabel="ms",
-            plot_name="batched-matrix-multiplication-performance",
+            plot_name="bmm-performance",
             args={"b": 4},
         )
     )

conv2d.py

@@ -43,7 +43,7 @@
             line_names=["NineToothed", "PyTorch", "Triton"],
             styles=[("blue", "-"), ("green", "-"), ("orange", "-")],
             ylabel="ms",
-            plot_name="2d-convolution-performance",
+            plot_name="conv2d-performance",
             args={},
         )
     )

max_pool2d.py

@@ -78,7 +78,7 @@ def max_pool2d(input, window_shape):
             line_names=["NineToothed", "PyTorch"],
             styles=[("blue", "-"), ("green", "-")],
             ylabel="ms",
-            plot_name="2d-max-pooling-performance",
+            plot_name="max-pool2d-performance",
             args={},
         )
     )

mm.py

@@ -41,7 +41,7 @@
             line_names=["NineToothed", "PyTorch", "Triton"],
             styles=[("blue", "-"), ("green", "-"), ("orange", "-")],
             ylabel="ms",
-            plot_name="matrix-multiplication-performance",
+            plot_name="mm-performance",
             args={},
         )
     )

rotary_position_embedding.py

@@ -87,7 +87,7 @@ def _generate_sin_and_cos_tables(
             line_names=["NineToothed", "PyTorch", "Triton"],
             styles=[("blue", "-"), ("green", "-"), ("orange", "-")],
             ylabel="ms",
-            plot_name="rotary_position_embedding-performance",
+            plot_name="rotary-position-embedding-performance",
             args={},
         )
     )

scaled_dot_product_attention.py

@@ -157,7 +157,7 @@ def rotary_position_embedding_backend(backend_name):
             line_names=["NineToothed", "PyTorch", "Triton"],
             styles=[("blue", "-"), ("green", "-"), ("orange", "-")],
             ylabel="ms",
-            plot_name="attention-performance",
+            plot_name="scaled-dot-product-attention-performance",
             args={},
         )
     )