precommit

Signed-off-by: czhu-cohere <conway.zhu@cohere.com>

Author: czhu-cohere

benchmarks/kernels/weight_shapes.py

@@ -96,9 +96,9 @@
         ([1408, 2048], 0),
     ],
     "CohereLabs/c4ai-command-a-03-2025": [
-        ([12288, 73728], 1),    # gate_up_proj
-        ([36864, 12288], 0),    # down_proj
-        ([12288,  14336], 1),   # qkv_proj
-        ([12288, 12288], 0),    # o_proj
+        ([12288, 73728], 1),  # gate_up_proj
+        ([36864, 12288], 0),  # down_proj
+        ([12288, 14336], 1),  # qkv_proj
+        ([12288, 12288], 0),  # o_proj
     ],
 }

csrc/quantization/machete/generate.py

@@ -191,7 +191,6 @@
   {{DataTypeTag[t.b_group_zeropoint]}}, // GroupZeroT
   {{DataTypeTag[t.b_channel_scale]}}, // ChannelScaleT
   {{DataTypeTag[t.a_token_scale]}}, // TokenScaleT
-  // cutlass::gemm::KernelTmaWarpSpecializedCooperative, // moved to schedule config
   Sch>;
 
 {% for sch in schs %}
@@ -520,11 +519,12 @@ def generate():
     # Heuristic is currently tuned for H100s
     default_heuristic = [
         (
-            cond, ScheduleConfig(
+            cond,
+            ScheduleConfig(
                 *tile_config,
                 **sch_config_overrides.get(cond,
                                            sch_common_params)  # type: ignore
-        )) for cond, tile_config in default_tile_heuristic_config.items()
+            )) for cond, tile_config in default_tile_heuristic_config.items()
     ]
 
     def get_unique_schedules(heuristic: dict[str, ScheduleConfig]):