Fix bug for Expr16

This commit is contained in:
xiaying 2022-06-07 11:27:47 +08:00
parent 007a6e867f
commit 2a0427775e
1 changed files with 1 additions and 1 deletions

View File

@ -35,7 +35,7 @@ void Arm82MNNExp(FLOAT16* dst, const FLOAT16* src, size_t dataSize) {
int16_t srcTemp[16];
int16_t dstTemp[16];
::memcpy(srcTemp, src + blockCount * 16, remainSize * sizeof(int16_t));
MNNExpFP16(dstTemp, srcTemp, params, blockCount);
MNNExpFP16(dstTemp, srcTemp, params, 1);
::memcpy(dst + blockCount * 16, dstTemp, remainSize * sizeof(int16_t));
}
}