Skip to content

Commit 4c849bb

Browse files
cleanup
1 parent 5388877 commit 4c849bb

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

bitsandbytes/research/autograd/_functions.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,6 @@ def forward(ctx, A, B, out=None, bias=None, state: Optional[MatmulLtState] = Non
223223
if state.has_fp16_weights:
224224
idx = outlier_cols
225225
CA[:, idx] = 0
226-
# CAt[:, idx] = 0
227226
subA = A[:, idx]
228227
state.subB = B[:, idx].t().contiguous()
229228
state.idx = idx
@@ -264,7 +263,7 @@ def forward(ctx, A, B, out=None, bias=None, state: Optional[MatmulLtState] = Non
264263
outliers = state.CB[:, state.idx.long()].clone()
265264
state.subB = (outliers * state.SCB.view(-1, 1) / 127.0).t().contiguous().to(A.dtype)
266265
CA[:, state.idx.long()] = 0
267-
# CAt[:, state.idx.long()] = 0
266+
268267
subA = A[:, state.idx.long()]
269268

270269
shapeB = state.SB[0]

0 commit comments

Comments
 (0)