Skip to content

Commit 37c4f97

Browse files
fixed quantized_matmul_out
Differential Revision: D60810654 Pull Request resolved: #4553
1 parent d3a7c71 commit 37c4f97

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

backends/cadence/aot/functions.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@
150150
- arg_meta: null
151151
kernel_name: impl::reference::quantized_relu_out
152152

153-
func: cadence::quantized_matmul.out(Tensor X, int X_zero_point, Tensor Y, int Y_zero_point, Tensor? bias, int out_multiplier, int out_shift, int out_zero_point, bool transposed, *, Tensor(a!) out) -> Tensor(a!)
153+
- func: cadence::quantized_matmul.out(Tensor X, int X_zero_point, Tensor Y, int Y_zero_point, Tensor? bias, int out_multiplier, int out_shift, int out_zero_point, bool transposed, *, Tensor(a!) out) -> Tensor(a!)
154154
kernels:
155155
- arg_meta: null
156156
kernel_name: impl::reference::quantized_matmul_out

backends/cadence/reference/operators/quantized_matmul_out.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,6 @@ void inline _typed_quantized_matmul(
105105
out_dim);
106106
}
107107
}
108-
break;
109108
}
110109

111110
void quantized_matmul_out(
@@ -120,7 +119,7 @@ void quantized_matmul_out(
120119
int64_t out_zero_point,
121120
bool transposed,
122121
Tensor& out) {
123-
if (out.scalar_type() == at::ScalarType::Byte) {
122+
if (out.scalar_type() == exec_aten::ScalarType::Byte) {
124123
_typed_quantized_matmul<uint8_t>(
125124
X,
126125
X_zero_point,
@@ -132,7 +131,7 @@ void quantized_matmul_out(
132131
out_zero_point,
133132
transposed,
134133
out);
135-
} else if (out.scalar_type() == at::ScalarType::Char) {
134+
} else if (out.scalar_type() == exec_aten::ScalarType::Char) {
136135
_typed_quantized_matmul<int8_t>(
137136
X,
138137
X_zero_point,

0 commit comments

Comments
 (0)