diff options
Diffstat (limited to 'test/multiply_test.cc')
| -rw-r--r-- | test/multiply_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/multiply_test.cc b/test/multiply_test.cc index 0fd8231..a054753 100644 --- a/test/multiply_test.cc +++ b/test/multiply_test.cc @@ -278,7 +278,7 @@ template <class Routine> void TestMultiply(Index A_rows, Index width, Index B_co Routine::PrepareB(B.begin(), B_prep.begin(), quant_mult, width, B_cols); AlignedVector<float> test_C(A_rows * B_cols); - Routine::Multiply(A_prep.begin(), B_prep.begin(), A_rows, width, B_cols, callbacks::UnquantizeAndWrite(unquant_mult, test_C.begin())); + OMPParallelWrap<callbacks::UnquantizeAndWrite, Routine>(A_prep.begin(), B_prep.begin(), A_rows, width, B_cols, callbacks::UnquantizeAndWrite(unquant_mult, test_C.begin())); // Routine::Multiply(A_prep.begin(), B_prep.begin(), A_rows, width, B_cols, callbacks::Sequence( // callbacks::Unquantize(unquant_mult), // callbacks::Write<float>(test_C.begin()) |