diff options
Diffstat (limited to 'tests/validation/reference/Floor.cpp')
-rw-r--r-- | tests/validation/reference/Floor.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/validation/reference/Floor.cpp b/tests/validation/reference/Floor.cpp index b011a16974..21fa1c9932 100644 --- a/tests/validation/reference/Floor.cpp +++ b/tests/validation/reference/Floor.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2020 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -42,6 +42,9 @@ SimpleTensor<T> floor_layer(const SimpleTensor<T> &src) SimpleTensor<T> dst{ src.shape(), src.data_type() }; // Compute reference +#if defined(_OPENMP) + #pragma omp parallel for +#endif /* _OPENMP */ for(int i = 0; i < src.num_elements(); ++i) { dst[i] = std::floor(src[i]); |