From 91f7b0936f0d4de71448d1a9408a015dc760731e Mon Sep 17 00:00:00 2001 From: Satya Nikhil Kodukula Date: Mon, 10 Nov 2025 09:48:10 -0500 Subject: [PATCH] Bug fix --- aiter/ops/triton/batched_gemm_afp4wfp4_pre_quant.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aiter/ops/triton/batched_gemm_afp4wfp4_pre_quant.py b/aiter/ops/triton/batched_gemm_afp4wfp4_pre_quant.py index 8679344856..b61dc52c4d 100755 --- a/aiter/ops/triton/batched_gemm_afp4wfp4_pre_quant.py +++ b/aiter/ops/triton/batched_gemm_afp4wfp4_pre_quant.py @@ -60,6 +60,7 @@ def batched_gemm_afp4wfp4_pre_quant( By, _, _ = y.shape assert Bx == Bw == By Batch = Bx + w = w.transpose(1, 2) if config is None: config = _get_config(M, N, K)