We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 78cc292 commit 99d32ebCopy full SHA for 99d32eb
paddle/fluid/operators/fake_quantize_op.cu.h
@@ -305,7 +305,7 @@ __global__ void ChannelClipAndQuantKernelQuantAxisN(
305
int64_t idx = blockDim.x * blockIdx.x + threadIdx.x;
306
for (int64_t i = idx; i < n; i += blockDim.x * gridDim.x) {
307
T s = scale[(i / quant_stride) % nScale];
308
- T inv_s = 1.0 / s;
+ T inv_s = inverse(s);
309
T x = in[i];
310
T v = x > s ? s : x;
311
v = v < -s ? -s : v;
0 commit comments