repri kernels

Author: yf225

benchmarks/run.py

@@ -61,28 +61,28 @@
     #     "examples.jagged_mean",
     #     "jagged_mean_tritonbench",
     # ),
+    # Multiple kernel mappings: (<tritonbench_module_path>, [(<helion_module>, <helion_func>), ...])
+    "gemm": (
+        "tritonbench.operators.gemm.operator",
+        [
+            ("examples.matmul", "matmul"),
+            ("examples.matmul_split_k", "matmul_split_k"),
+        ],
+    ),
     "flash_attention": (
         "tritonbench.operators.flash_attention.operator",
         "examples.attention",
         "attention",
     ),
-    "fp8_attention": (
-        "tritonbench.operators.fp8_attention.operator",
-        "examples.fp8_attention",
-        "fp8_attention_tritonbench",
-    ),
     "fp8_gemm": (
         "tritonbench.operators.fp8_gemm.fp8_gemm",
         "examples.fp8_gemm",
         "fp8_gemm_tritonbench",
     ),
-    # Multiple kernel mappings: (<tritonbench_module_path>, [(<helion_module>, <helion_func>), ...])
-    "gemm": (
-        "tritonbench.operators.gemm.operator",
-        [
-            ("examples.matmul", "matmul"),
-            ("examples.matmul_split_k", "matmul_split_k"),
-        ],
+    "fp8_attention": (
+        "tritonbench.operators.fp8_attention.operator",
+        "examples.fp8_attention",
+        "fp8_attention_tritonbench",
     ),
 }