From 4da2227fcbc3803d680dff113403aecac1827bc3 Mon Sep 17 00:00:00 2001 From: justheuristic Date: Sun, 18 Sep 2022 01:03:21 +0300 Subject: debug --- bitsandbytes/autograd/_functions.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'bitsandbytes') diff --git a/bitsandbytes/autograd/_functions.py b/bitsandbytes/autograd/_functions.py index 9928fbd..407f14b 100644 --- a/bitsandbytes/autograd/_functions.py +++ b/bitsandbytes/autograd/_functions.py @@ -370,6 +370,8 @@ class MatMul8bitLt(torch.autograd.Function): if state.threshold > 0.0 and subA is not None: grad_B[:, idx] += torch.matmul(grad_output.t(), subA) + raise NotImplementedError("!!") + if req_gradA: if state.CBt is not None: C32grad, Sgrad = F.transform(Cgrad, "col32") -- cgit v1.2.3