diff options
author | Kenneth Heafield <github@kheafield.com> | 2020-03-03 18:26:04 +0300 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2020-03-03 18:26:04 +0300 |
commit | aa174e619583552dcbd9690af961e11ab75c24e2 (patch) | |
tree | c09a94364b9db673d0a680e535cc79f1b3e0977a | |
parent | 65393a40af704dca15ea5d283fb32dbee2580ef6 (diff) |
Fix output address for quantizer
-rw-r--r-- | multiply.h | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -85,7 +85,7 @@ target static void Quantize(const float *const input, int8_t *const output, floa inputs[i] = &input[fast_end]; \ } \ Register result = q.Tile(inputs[0], inputs[1], inputs[2], inputs[3]); \ - std::memcpy(output, &result, overhang); \ + std::memcpy(output + (size & ~(kBatch - 1)), &result, overhang); \ } /* Take 4 registers with 32-bit values to be horizontally added. Reduce them |