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:
authorMateusz Chudyk <mateuszchudyk@gmail.com>2019-07-09 22:56:08 +0300
committerMateusz Chudyk <mateuszchudyk@gmail.com>2019-07-09 22:56:08 +0300
commit916c1d1d38abc4d480c1382292898637ba99b7b0 (patch)
treecbbdb4452166674dcc1a94e8bc2105683d0b653d /README.md
parent6de7be50a5ba393512206a41b58634e418492bba (diff)
Update README.md
Diffstat (limited to 'README.md')
-rw-r--r--README.md6
1 files changed, 3 insertions, 3 deletions
diff --git a/README.md b/README.md
index 7742cbb..0a6bfc5 100644
--- a/README.md
+++ b/README.md
@@ -28,11 +28,11 @@ Both A and B should be prepared before multiplication.
* B is width x B_cols.
*/
/* Prepare A for multiplication. This might be offline or on the fly. */
-intgemm::Int16::PrepareA(A, A_prepared, quant_mult, A_rows, width);
+intgemm::Int16::PrepareA(A.begin(), A_prepared.begin(), quant_mult, A_rows, width);
/* Prepare B for multiplication. This is typically done offline. */
-intgemm::Int16::PrepareB(B, B_prepared, quant_mult, width, B_cols);
+intgemm::Int16::PrepareB(B.begin(), B_prepared.begin(), quant_mult, width, B_cols);
/* Multiply and produce results in C */
-intgemm::Int16::Multiply(A_prepared.begin(), B_prepared.begin(), C.begin(), intgemm::CreatePostprocessPipeline(intgemm::Unquantize(1.0 / (quant_mult * quant_mult))), A_rows, width, B_cols);
+intgemm::Int16::Multiply(A_prepared.begin(), B_prepared.begin(), A_rows, width, B_cols, intgemm::callbacks::UnquantizeAndWrite(1.0 / (quant_mult * quant_mult), C.begin()));
```
For 8-bit, use `Int8` instead of `Int16`.