debug assert

2022-09-18 00:00:45 +03:00 · 2022-09-18 00:00:45 +03:00 · 210b9ed9ce
commit 210b9ed9ce
parent 647c976a74
1 changed files with 1 additions and 0 deletions
--- a/bitsandbytes/autograd/_functions.py
+++ b/bitsandbytes/autograd/_functions.py
@ -323,6 +323,7 @@ class MatMul8bitLt(torch.autograd.Function):

        # 4. Mixed-precision decomposition matmul
        if coo_tensorA is not None and subA is not None:
+            assert subA.dtype == state.subB.dtype == output.dtype
            output.addmm_(subA, state.subB)

        # 5. Save state