68 lines · cpp
1//===----------------------------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9// UNSUPPORTED: c++03, c++11, c++14, c++1710 11#include <algorithm>12#include <cstddef>13#include <deque>14#include <iterator>15#include <list>16#include <string>17#include <vector>18 19#include "benchmark/benchmark.h"20#include "../../GenerateInput.h"21 22int main(int argc, char** argv) {23 auto std_transform = [](auto first, auto last, auto out, auto f) { return std::transform(first, last, out, f); };24 25 // {std,ranges}::transform(normal container)26 {27 auto bm = []<class Container>(std::string name, auto transform) {28 benchmark::RegisterBenchmark(29 name,30 [transform](auto& st) {31 std::size_t const size = st.range(0);32 using ValueType = typename Container::value_type;33 Container c;34 std::generate_n(std::back_inserter(c), size, [] { return Generate<ValueType>::random(); });35 36 std::vector<ValueType> out(size);37 38 auto f = [](auto element) {39 benchmark::DoNotOptimize(element);40 return element;41 };42 43 for ([[maybe_unused]] auto _ : st) {44 benchmark::DoNotOptimize(c);45 benchmark::DoNotOptimize(out);46 auto result = transform(c.begin(), c.end(), out.begin(), f);47 benchmark::DoNotOptimize(result);48 }49 })50 ->Arg(32)51 ->Arg(50) // non power-of-two52 ->Arg(1024)53 ->Arg(8192);54 };55 bm.operator()<std::vector<int>>("std::transform(vector<int>) (identity transform)", std_transform);56 bm.operator()<std::deque<int>>("std::transform(deque<int>) (identity transform)", std_transform);57 bm.operator()<std::list<int>>("std::transform(list<int>) (identity transform)", std_transform);58 bm.operator()<std::vector<int>>("rng::transform(vector<int>) (identity transform)", std::ranges::transform);59 bm.operator()<std::deque<int>>("rng::transform(deque<int>) (identity transform)", std::ranges::transform);60 bm.operator()<std::list<int>>("rng::transform(list<int>) (identity transform)", std::ranges::transform);61 }62 63 benchmark::Initialize(&argc, argv);64 benchmark::RunSpecifiedBenchmarks();65 benchmark::Shutdown();66 return 0;67}68