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

github.com/marian-nmt/intgemm.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKenneth Heafield <github@kheafield.com>2020-03-03 18:26:04 +0300
committerKenneth Heafield <github@kheafield.com>2020-03-03 18:26:04 +0300
commitaa174e619583552dcbd9690af961e11ab75c24e2 (patch)
treec09a94364b9db673d0a680e535cc79f1b3e0977a
parent65393a40af704dca15ea5d283fb32dbee2580ef6 (diff)
Fix output address for quantizer
-rw-r--r--multiply.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/multiply.h b/multiply.h
index e2aa0fb..a9766d3 100644
--- a/multiply.h
+++ b/multiply.h
@@ -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