| 25 | namespace bc = boost::compute; |
| 26 | |
| 27 | BOOST_AUTO_TEST_CASE(partition_float_vector) |
| 28 | { |
| 29 | bc::vector<float> vector(context); |
| 30 | vector.push_back(1.0f, queue); |
| 31 | vector.push_back(2.0f, queue); |
| 32 | vector.push_back(-1.0f, queue); |
| 33 | vector.push_back(-2.0f, queue); |
| 34 | vector.push_back(3.0f, queue); |
| 35 | vector.push_back(4.0f, queue); |
| 36 | vector.push_back(-3.0f, queue); |
| 37 | vector.push_back(-4.0f, queue); |
| 38 | |
| 39 | // verify is_partitioned() |
| 40 | BOOST_VERIFY(bc::is_partitioned(vector.begin(), |
| 41 | vector.end(), |
| 42 | bc::signbit_<float>(), |
| 43 | queue) == false); |
| 44 | |
| 45 | // partition by signbit |
| 46 | bc::vector<float>::iterator iter = bc::partition(vector.begin(), |
| 47 | vector.end(), |
| 48 | bc::signbit_<float>(), |
| 49 | queue); |
| 50 | queue.finish(); |
| 51 | BOOST_VERIFY(iter == vector.begin() + 4); |
| 52 | BOOST_CHECK_LT(vector[0], 0.0f); |
| 53 | BOOST_CHECK_LT(vector[1], 0.0f); |
| 54 | BOOST_CHECK_LT(vector[2], 0.0f); |
| 55 | BOOST_CHECK_LT(vector[3], 0.0f); |
| 56 | BOOST_CHECK_GT(vector[4], 0.0f); |
| 57 | BOOST_CHECK_GT(vector[5], 0.0f); |
| 58 | BOOST_CHECK_GT(vector[6], 0.0f); |
| 59 | BOOST_CHECK_GT(vector[7], 0.0f); |
| 60 | |
| 61 | // verify is_partitioned() |
| 62 | BOOST_VERIFY(bc::is_partitioned(vector.begin(), |
| 63 | vector.end(), |
| 64 | bc::signbit_<float>(), |
| 65 | queue) == true); |
| 66 | } |
| 67 | |
| 68 | BOOST_AUTO_TEST_CASE(partition_small_vector) |
| 69 | { |