Skip to content

Commit 6689afa

Browse files
authored
Merge pull request #567 from apbard/patch-1
[BugFix] replace view+continuous with reshape
2 parents 817bdf6 + 463630d commit 6689afa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/autograd/_functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,7 @@ def forward(ctx, A, B, out=None, bias=None, state=MatmulLtState):
323323

324324
# 1. Quantize A
325325
if len(A.shape) == 3:
326-
A = A.view(-1, A.shape[-1]).contiguous()
326+
A = A.reshape(-1, A.shape[-1])
327327
CA, CAt, SCA, SCAt, coo_tensorA = F.double_quant(A.to(torch.float16), threshold=state.threshold)
328328

329329
if state.threshold > 0.0 and coo_tensorA is not None:

0 commit comments

Comments
 (0)