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

github.com/marian-nmt/intgemm/intgemm.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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