Skip to content

Commit 99d32eb

Browse files
Fix fake quant cuda kernel (#41305) (#41588)
1 parent 78cc292 commit 99d32eb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/fake_quantize_op.cu.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ __global__ void ChannelClipAndQuantKernelQuantAxisN(
305305
int64_t idx = blockDim.x * blockIdx.x + threadIdx.x;
306306
for (int64_t i = idx; i < n; i += blockDim.x * gridDim.x) {
307307
T s = scale[(i / quant_stride) % nScale];
308-
T inv_s = 1.0 / s;
308+
T inv_s = inverse(s);
309309
T x = in[i];
310310
T v = x > s ? s : x;
311311
v = v < -s ? -s : v;

0 commit comments

Comments
 (0)