format

scxiao · scxiao · commit abaa3f261dbf · 2026-01-21T11:32:17.000-06:00
diff --git a/tritonbench/operators/grouped_gemm/kernels.py b/tritonbench/operators/grouped_gemm/kernels.py
@@ -72,7 +72,7 @@ def torch_dtype_to_triton_dtype(dtype):
                 "BLOCK_SIZE_K": BLOCK_K,
                 "NUM_SMS": num_sms(),
             },
-            num_stages = 3 if torch.version.hip is None else 2,
+            num_stages=3 if torch.version.hip is None else 2,
         )
         for BLOCK_M, BLOCK_N, BLOCK_K in itertools.product([128, 256], repeat=3)
     ],

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ def torch_dtype_to_triton_dtype(dtype):`
`72`	`72`	`"BLOCK_SIZE_K": BLOCK_K,`
`73`	`73`	`"NUM_SMS": num_sms(),`
`74`	`74`	`},`
`75`		`- num_stages = 3 if torch.version.hip is None else 2,`
	`75`	`+ num_stages=3 if torch.version.hip is None else 2,`
`76`	`76`	`)`
`77`	`77`	`for BLOCK_M, BLOCK_N, BLOCK_K in itertools.product([128, 256], repeat=3)`
`78`	`78`	`],`