Compare commits

...

1 Commits

Author SHA1 Message Date
099216fbc1 Test nn.linear bias
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
2024-05-15 15:01:59 -07:00

View File

@ -1764,9 +1764,10 @@ class TestQuantizeFx(QuantizationTestCase):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.linear = torch.nn.Linear(5, 10) self.linear = torch.nn.Linear(5, 10)
self.bias = torch.tensor((5,))
def forward(self, x): def forward(self, x):
return self.linear(x) return self.linear(xi, bias=self.bias)
linear_module_input = torch.rand(8, 5) linear_module_input = torch.rand(8, 5)