| 1 | // -*- C++ -*- |
| 2 | //===-- find.pass.cpp -----------------------------------------------------===// |
| 3 | // |
| 4 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 5 | // See https://llvm.org/LICENSE.txt for license information. |
| 6 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | // UNSUPPORTED: c++03, c++11, c++14 |
| 11 | |
| 12 | // Tests for find |
| 13 | #include "support/pstl_test_config.h" |
| 14 | |
| 15 | #include <execution> |
| 16 | #include <algorithm> |
| 17 | |
| 18 | #include "support/utils.h" |
| 19 | |
| 20 | using namespace TestUtils; |
| 21 | |
| 22 | struct test_find |
| 23 | { |
| 24 | #if defined(_PSTL_ICC_17_VC141_TEST_SIMD_LAMBDA_DEBUG_32_BROKEN) || \ |
| 25 | defined(_PSTL_ICC_16_VC14_TEST_SIMD_LAMBDA_DEBUG_32_BROKEN) //dummy specialization by policy type, in case of broken configuration |
| 26 | template <typename Iterator, typename Value> |
| 27 | void |
| 28 | operator()(pstl::execution::unsequenced_policy, Iterator first, Iterator last, Value value) |
| 29 | { |
| 30 | } |
| 31 | template <typename Iterator, typename Value> |
| 32 | void |
| 33 | operator()(pstl::execution::parallel_unsequenced_policy, Iterator first, Iterator last, Value value) |
| 34 | { |
| 35 | } |
| 36 | #endif |
| 37 | |
| 38 | template <typename Policy, typename Iterator, typename Value> |
| 39 | void |
| 40 | operator()(Policy&& exec, Iterator first, Iterator last, Value value) |
| 41 | { |
| 42 | auto i = std::find(first, last, value); |
| 43 | auto j = find(exec, first, last, value); |
| 44 | EXPECT_TRUE(i == j, "wrong return value from find" ); |
| 45 | } |
| 46 | }; |
| 47 | |
| 48 | template <typename T, typename Value, typename Hit, typename Miss> |
| 49 | void |
| 50 | test(Value value, Hit hit, Miss miss) |
| 51 | { |
| 52 | // Try sequences of various lengths. |
| 53 | for (size_t n = 0; n <= 100000; n = n <= 16 ? n + 1 : size_t(3.1415 * n)) |
| 54 | { |
| 55 | Sequence<T> in(n, [&](size_t k) -> T { return miss(n ^ k); }); |
| 56 | // Try different find positions, including not found. |
| 57 | // By going backwards, we can add extra matches that are *not* supposed to be found. |
| 58 | // The decreasing exponential gives us O(n) total work for the loop since each find takes O(m) time. |
| 59 | for (size_t m = n; m > 0; m *= 0.6) |
| 60 | { |
| 61 | if (m < n) |
| 62 | in[m] = hit(n ^ m); |
| 63 | invoke_on_all_policies(test_find(), in.begin(), in.end(), value); |
| 64 | invoke_on_all_policies(test_find(), in.cbegin(), in.cend(), value); |
| 65 | } |
| 66 | } |
| 67 | } |
| 68 | |
| 69 | // Type defined for sake of checking that std::find works with asymmetric ==. |
| 70 | class Weird |
| 71 | { |
| 72 | Number value; |
| 73 | |
| 74 | public: |
| 75 | friend bool |
| 76 | operator==(Number x, Weird y) |
| 77 | { |
| 78 | return x == y.value; |
| 79 | } |
| 80 | Weird(int32_t val, OddTag) : value(val, OddTag()) {} |
| 81 | }; |
| 82 | |
| 83 | int |
| 84 | main() |
| 85 | { |
| 86 | // Note that the "hit" and "miss" functions here avoid overflow issues. |
| 87 | test<Number>(value: Weird(42, OddTag()), hit: [](int32_t) { return Number(42, OddTag()); }, // hit |
| 88 | miss: [](int32_t j) { return Number(j == 42 ? 0 : j, OddTag()); }); // miss |
| 89 | |
| 90 | // Test with value that is equal to two different bit patterns (-0.0 and 0.0) |
| 91 | test<float32_t>(value: -0.0, hit: [](int32_t j) { return j & 1 ? 0.0 : -0.0; }, // hit |
| 92 | miss: [](int32_t j) { return j == 0 ? ~j : j; }); // miss |
| 93 | |
| 94 | std::cout << done() << std::endl; |
| 95 | return 0; |
| 96 | } |
| 97 | |