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:
authorMateusz Chudyk <mateuszchudyk@gmail.com>2020-02-05 19:33:05 +0300
committerMateusz Chudyk <mateuszchudyk@gmail.com>2020-02-05 19:38:25 +0300
commitfc816e0b6b2eeb94cc14f80c38ea3fb75a5518d7 (patch)
treea55a00e3fb210f614ea112bb3eb180c11c7c8b87
parent4587fea73375c6dac775308c650082e42eee6578 (diff)
Change order of size args in PrepareBTransposed
-rw-r--r--interleave.h2
-rw-r--r--test/prepare_b_transposed.cc2
2 files changed, 2 insertions, 2 deletions
diff --git a/interleave.h b/interleave.h
index 728d19d..30e0dea 100644
--- a/interleave.h
+++ b/interleave.h
@@ -263,7 +263,7 @@ target static inline void PrepareBQuantizedTransposed(const Integer* input, Inte
* Cols has to be a multiple of sizeof(Register) / sizeof(float).
*/
#define INTGEMM_PREPARE_B_TRANSPOSED(target, Quantizer, integer) \
-target static inline void PrepareBTransposed(const float* input, integer* output, float quant_mult, Index rows, Index cols) { \
+target static inline void PrepareBTransposed(const float* input, integer* output, float quant_mult, Index cols, Index rows) { \
using Register = typename Quantizer::Integer; \
const Index RegisterElemsInt = sizeof(Register) / sizeof(integer); \
const Index RegisterElemsFloat = sizeof(Register) / sizeof(float); \
diff --git a/test/prepare_b_transposed.cc b/test/prepare_b_transposed.cc
index 1edd5c1..45730a7 100644
--- a/test/prepare_b_transposed.cc
+++ b/test/prepare_b_transposed.cc
@@ -31,7 +31,7 @@ bool Test(const AlignedVector<float>& input, Index rows, Index cols, float quant
bool success = true;
AlignedVector<typename Backend::Integer> output(input.size());
- Backend::PrepareBTransposed(input.begin(), output.begin(), quant_mult, rows, cols);
+ Backend::PrepareBTransposed(input.begin(), output.begin(), quant_mult, cols, rows);
AlignedVector<typename Backend::Integer> reference(input.size());
PrepareBTransposedRef<Backend>(input.begin(), reference.begin(), quant_mult, rows, cols);