Enable ZP Support for Machete (#20268)

Signed-off-by: czhu-cohere <conway.zhu@cohere.com>
2025-07-01 00:12:20 -07:00
parent 22e9d42040
commit 9909726d2a
3 changed files with 19 additions and 5 deletions
--- a/benchmarks/kernels/benchmark_machete.py
+++ b/benchmarks/kernels/benchmark_machete.py
@@ -234,8 +234,10 @@ def marlin_create_bench_fn(bt: BenchmarkTensors) -> Callable:

        fn = lambda: ops.gptq_marlin_gemm(
            a=bt.a,
+            c=None,
            b_q_weight=w_q,
            b_scales=w_s,
+            global_scale=None,
            b_zeros=w_zp,
            g_idx=g_idx,
            perm=sort_indices,