Commit 697bd02c authored by Tom Aarsen's avatar Tom Aarsen
Browse files

Resolve dangerous default value [] as argument

parent b5cf7063
...@@ -50,8 +50,9 @@ class GlobalOutlierPooler: ...@@ -50,8 +50,9 @@ class GlobalOutlierPooler:
class MatMul8bit(torch.autograd.Function): class MatMul8bit(torch.autograd.Function):
@staticmethod @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: if precision[0] != 8:
with torch.no_grad(): with torch.no_grad():
output = torch.matmul(A, B) output = torch.matmul(A, B)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment