Resolve dangerous default value [] as argument
This commit is contained in:
parent
b5cf706341
commit
697bd02c60
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user