[libcxx-commits] [libcxx] [libc++] Rewrite the std::pop_heap benchmark (PR #179911)

[via libcxx-commits]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-libcxx

Author: Nikolas Klauser (philnik777)

<details>
<summary>Changes</summary>



---
Full diff: https://github.com/llvm/llvm-project/pull/179911.diff


1 Files Affected:

- (added) libcxx/test/benchmarks/algorithms/sorting/pop_heap.bench.cpp (+52) 


``````````diff
diff --git a/libcxx/test/benchmarks/algorithms/sorting/pop_heap.bench.cpp b/libcxx/test/benchmarks/algorithms/sorting/pop_heap.bench.cpp
new file mode 100644
index 0000000000000..ad0ca8bff3a5b
--- /dev/null
+++ b/libcxx/test/benchmarks/algorithms/sorting/pop_heap.bench.cpp
@@ -0,0 +1,52 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+// UNSUPPORTED: c++03, c++11, c++14, c++17
+
+#include <algorithm>
+#include <deque>
+#include <iterator>
+#include <vector>
+
+#include "benchmark/benchmark.h"
+#include "../../GenerateInput.h"
+
+int main(int argc, char** argv) {
+  auto bm = []<class Container>(std::type_identity<Container>, std::string name) {
+    benchmark::RegisterBenchmark(name, [](benchmark::State& state) {
+      std::size_t size = state.range(0);
+
+      Container c;
+      std::generate_n(std::back_inserter(c), size, [] { return Generate<typename Container::value_type>::random(); });
+
+      while (state.KeepRunningBatch(size)) {
+        state.PauseTiming();
+        std::make_heap(c.begin(), c.end());
+        state.ResumeTiming();
+
+        for (auto first = c.begin(), last = c.end(); last != first; --last) {
+          std::pop_heap(first, last);
+        }
+      }
+    })->Arg(8)->Arg(1024)->Arg(8192);
+  };
+
+  bm(std::type_identity<std::vector<int>>{}, "std::pop_heap(vector<int>)");
+  bm(std::type_identity<std::vector<float>>{}, "std::pop_heap(vector<float>)");
+  bm(std::type_identity<std::vector<size_t>>{}, "std::pop_heap(vector<size_t>)");
+  bm(std::type_identity<std::vector<std::string>>{}, "std::pop_heap(vector<std::string>)");
+  bm(std::type_identity<std::deque<int>>{}, "std::pop_heap(deque<int>)");
+  bm(std::type_identity<std::deque<float>>{}, "std::pop_heap(deque<float>)");
+  bm(std::type_identity<std::deque<size_t>>{}, "std::pop_heap(deque<size_t>)");
+  bm(std::type_identity<std::deque<std::string>>{}, "std::pop_heap(deque<std::string>)");
+
+  benchmark::Initialize(&argc, argv);
+  benchmark::RunSpecifiedBenchmarks();
+  benchmark::Shutdown();
+  return 0;
+}

``````````

</details>


https://github.com/llvm/llvm-project/pull/179911