Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/FBGEMM.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/GroupwiseConvAcc32Avx2.cc')
-rw-r--r--src/GroupwiseConvAcc32Avx2.cc20
1 files changed, 11 insertions, 9 deletions
diff --git a/src/GroupwiseConvAcc32Avx2.cc b/src/GroupwiseConvAcc32Avx2.cc
index ef4ba7b..40f3fba 100644
--- a/src/GroupwiseConvAcc32Avx2.cc
+++ b/src/GroupwiseConvAcc32Avx2.cc
@@ -1872,15 +1872,17 @@ void fbgemmGroupwiseConv(
outProcess.getBZeroPoint()[0] == 0) ||
rowOffsetBuf == nullptr;
- requantizationParams_t r = {a_zero_point,
- outProcess.getBZeroPoint(),
- outProcess.getCZeroPoint(),
- outProcess.getCMultiplier(),
- rowOffsetBuf,
- outProcess.getColOffsets(),
- outProcess.getBias(),
- outProcess.getNCols(),
- G};
+ requantizationParams_t<typename processOutputType::BIAS_T> r = {
+ a_zero_point,
+ outProcess.getBZeroPoint(),
+ outProcess.getCZeroPoint(),
+ outProcess.getCMultiplier(),
+ rowOffsetBuf,
+ outProcess.getColOffsets(),
+ outProcess.getBias(),
+ outProcess.getNCols(),
+ G,
+ outProcess.getActWScale()};
const std::int32_t* inp = outBuffer;
block_type_t block{i * oh_ow, oh_ow, gOuter * K_per_G, 8 * K_per_G};