Resolve dangerous default value [] as argument

697bd02c · Tom Aarsen · b5cf7063 · 697bd02c
Commit 697bd02c authored Oct 27, 2022 by Tom Aarsen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

bitsandbytes/autograd/_functions.py bitsandbytes/autograd/_functions.py +3 -2

No files found.
--- a/bitsandbytes/autograd/_functions.py
+++ b/bitsandbytes/autograd/_functions.py
@@ -50,8 +50,9 @@ class GlobalOutlierPooler:

 class MatMul8bit(torch.autograd.Function):
    @staticmethod
-    def forward(ctx, A, B, out=None, quant_type="vector", precision=[8, 8, 8]):
-
+    def forward(ctx, A, B, out=None, quant_type="vector", precision=None):
+        if precision is None:
+            precision = [8, 8, 8]
        if precision[0] != 8:
            with torch.no_grad():
                output = torch.matmul(A, B)