Skip to content

Commit 279f860

Browse files
author
Zafar Takhirov
committed
Update on "Adding Scalar add/mul."
Note: This should be landed ONLY after #24259 Differential Revision: [D16846006](https://our.internmc.facebook.com/intern/diff/D16846006)
2 parents e3a6b49 + 93b9d18 commit 279f860

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/nn/quantized/modules/functional_modules.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def add_scalar(self, x, y):
118118
r"""Operation equivalent to ``torch.ops.quantized.mul(Tensor, Tensor)``"""
119119
def mul(self, x, y):
120120
# type: (Tensor, Tensor) -> Tensor
121-
return ops.quantized.add(x, y, scale=self.scale,
121+
return ops.quantized.mul(x, y, scale=self.scale,
122122
zero_point=self.zero_point)
123123

124124
r"""Operation equivalent to ``torch.ops.quantized.mul(Tensor, float)``"""

0 commit comments

Comments
 (0)