Rewrite relocation to be compliant with P1144 (cmuparlay#67)

[Relocation](https://quuxplusone.github.io/blog/2018/07/18/announcing-trivially-relocatable/)
now mostly follows the API proposed in
[P1144](https://www.open-std.org/jtc1/sc22/wg21/docs/papers/2023/p1144r9.html).
Parlay will also defer the implementation of relocation operations to
the compiler/library if they are present, which currently works on
Arthur O'Dwyer's LLVM fork
[here](https://github.com/Quuxplusone/llvm-project).

Author: DanielLiamAnderson

CMakeLists.txt

@@ -6,7 +6,8 @@
 # -------------------------------------------------------------------
 
 cmake_minimum_required(VERSION 3.14)
-project(PARLAY VERSION 2.2.4
+
+project(PARLAY VERSION 2.3.1
         DESCRIPTION "A collection of parallel algorithms and other support for parallelism in C++"
         LANGUAGES CXX)
 

benchmark/bench_sequence.cpp

@@ -27,6 +27,61 @@ static void bench_short_subscript(benchmark::State& state) {
   }
 }
 
+static void bench_grow_int64(benchmark::State& state) {
+  parlay::sequence<int64_t> s;
+  for (auto _ : state) {
+    state.PauseTiming();
+    s = parlay::sequence<int64_t>(10000000);
+    state.ResumeTiming();
+    s.reserve(s.capacity() + 1);    // Trigger grow
+  }
+}
+
+// No annotation needed since this one should be detectable
+struct Relocatable {
+  std::unique_ptr<int> x;
+  Relocatable() = default;
+  Relocatable(int x_) : x(std::make_unique<int>(x_)) { }
+};
+
+#if defined(PARLAY_MUST_SPECIALIZE_IS_TRIVIALLY_RELOCATABLE)
+namespace parlay {
+template<>
+PARLAY_ASSUME_TRIVIALLY_RELOCATABLE(Relocatable);
+}
+#endif
+
+static_assert(parlay::is_trivially_relocatable_v<Relocatable>);
+
+struct NotRelocatable {
+  std::unique_ptr<int> x;
+  NotRelocatable() = default;
+  NotRelocatable(int x_) : x(std::make_unique<int>(x_)) { }
+  NotRelocatable(NotRelocatable&& other) noexcept : x(std::move(other.x)) { }
+  ~NotRelocatable() { }
+};
+static_assert(!parlay::is_trivially_relocatable_v<NotRelocatable>);
+
+static void bench_grow_relocatable(benchmark::State& state) {
+  parlay::sequence<Relocatable> s;
+  for (auto _ : state) {
+    state.PauseTiming();
+    s = parlay::sequence<Relocatable>(10000000);
+    state.ResumeTiming();
+    s.reserve(s.capacity() + 1);    // Trigger grow
+  }
+}
+
+static void bench_grow_nonrelocatable(benchmark::State& state) {
+  parlay::sequence<NotRelocatable> s;
+  for (auto _ : state) {
+    state.PauseTiming();
+    s = parlay::sequence<NotRelocatable>(10000000);
+    state.ResumeTiming();
+    s.reserve(s.capacity() + 1);    // Trigger grow
+  }
+}
+
 // ------------------------- Registration -------------------------------
 
 #define BENCH(NAME) BENCHMARK(bench_ ## NAME)               \
@@ -35,3 +90,6 @@ static void bench_short_subscript(benchmark::State& state) {
 
 BENCH(subscript);
 BENCH(short_subscript);
+BENCH(grow_int64);
+BENCH(grow_relocatable);
+BENCH(grow_nonrelocatable);

include/parlay/alloc.h

@@ -183,8 +183,11 @@ struct allocator {
   template <class U> /* implicit */ constexpr allocator(const allocator<U>&) noexcept { }
 };
 
-template<typename T>
-struct is_trivially_relocatable<allocator<T>> : std::true_type {};
+// Allocator should be trivially copyable since it is stateless and has no user-provided copy
+// constructor.  This should guarantee that it is also trivially relocatable.
+static_assert(std::is_trivially_copyable_v<allocator<int>>);
+static_assert(is_trivially_relocatable_v<allocator<int>>);
+
 
 template <class T, class U>
 bool operator==(const allocator<T>&, const allocator<U>&) { return true; }

include/parlay/internal/bucket_sort.h

@@ -42,7 +42,7 @@ void radix_step_(slice<InIterator, InIterator> A,
 
   for (size_t j = n; j > 0; j--) {
     auto x = --counts[keys[j-1]];
-    uninitialized_relocate(&B[x], &A[j-1]);
+    relocate_at(&A[j - 1], &B[x]);
   }
 }
 
@@ -128,7 +128,7 @@ void base_sort(slice<InIterator, InIterator> in,
   else {
     quicksort(in.begin(), in.size(), f);
     if (!inplace) {
-      uninitialized_relocate_n(out.begin(), in.begin(), in.size());
+      parlay::uninitialized_relocate(in.begin(), in.end(), out.begin());
     }
   }
 }

include/parlay/internal/collect_reduce.h

@@ -271,7 +271,7 @@ auto seq_collect_reduce_sparse(Slice A, Helper const &helper) {
   auto r = r_s.begin();
   size_t j = 0;
   for (size_t i = 0; i < table_size; i++)
-    if (flags[i]) uninitialized_relocate(&r[j++], &table[i]);
+    if (flags[i]) relocate_at(&table[i], &r[j++]);
   assert(j == count);
   return r_s;
 }

include/parlay/internal/counting_sort.h

@@ -320,7 +320,7 @@ auto count_sort_inplace(slice<InIterator, InIterator> In, KeyS const& Keys, size
   using value_type = typename slice<InIterator, InIterator>::value_type;
   auto Tmp = uninitialized_sequence<value_type>(In.size());
   auto a = count_sort<uninitialized_relocate_tag>(In, make_slice(Tmp), make_slice(Keys), num_buckets);
-  uninitialized_relocate_n(In.begin(), Tmp.begin(), In.size());
+  parlay::uninitialized_relocate(Tmp.begin(), Tmp.end(), In.begin());
   return a.first;
 }
 

include/parlay/internal/delayed/filter.h

@@ -96,7 +96,7 @@ struct block_delayed_filter_t :
       }
     }
     auto res = sequence<It>::uninitialized(n);
-    uninitialized_relocate_n(res.begin(), temp.begin(), n);
+    parlay::uninitialized_relocate_n(temp.begin(), n, res.begin());
     return res;
   }
 

include/parlay/internal/delayed/filter_op.h

@@ -85,7 +85,7 @@ struct block_delayed_filter_op_t :
       }
     }
     auto res = sequence<result_type>::uninitialized(n);
-    uninitialized_relocate_n(res.begin(), temp.begin(), n);
+    parlay::uninitialized_relocate_n(temp.begin(), n, res.begin());
     return res;
   }
 

include/parlay/internal/integer_sort.h

@@ -80,10 +80,10 @@ void seq_radix_sort_(slice<InIterator, InIterator> In,
   }
 
   if (swapped && inplace) {
-    uninitialized_relocate_n(In.begin(), Out.begin(), In.size());
+    parlay::uninitialized_relocate(Out.begin(), Out.end(), In.begin());
   }
   else if (!swapped && !inplace) {
-    uninitialized_relocate_n(Out.begin(), In.begin(), Out.size());
+    parlay::uninitialized_relocate(In.begin(), In.end(), Out.begin());
   }
 }
 
@@ -105,10 +105,10 @@ void seq_radix_sort(slice<InIterator, InIterator> In,
     size_t n = In.size();
     if (odd) {
       // We could just use assign_dispatch(Tmp[i], In[i]) for each i, but we
-      // can optimize better by calling destructive_move_slice, since this
+      // can optimize better by calling uninitialized_relocate, since this
       // has the ability to memcpy multiple elements at once
       if constexpr (std::is_same_v<assignment_tag, uninitialized_relocate_tag>) {
-        uninitialized_relocate_n(Tmp.begin(), In.begin(), Tmp.size());
+        parlay::uninitialized_relocate(In.begin(), In.end(), Tmp.begin());
       }
       else {
         for (size_t i = 0; i < n; i++)
@@ -117,7 +117,7 @@ void seq_radix_sort(slice<InIterator, InIterator> In,
       seq_radix_sort_(Tmp, Out, g, key_bits, false);
     } else {
       if constexpr (std::is_same_v<assignment_tag, uninitialized_relocate_tag>) {
-        uninitialized_relocate_n(Out.begin(), In.begin(), Out.size());
+        parlay::uninitialized_relocate(In.begin(), In.end(), Out.begin());
       }
       else {
         for (size_t i = 0; i < n; i++)
@@ -219,7 +219,7 @@ sequence<size_t> integer_sort_r(slice<InIterator, InIterator> In,
       // uninitialized_relocate_n, which can memcpy multiple elements at a time
       // to save on performing every copy individually.
       if constexpr (std::is_same_v<assignment_tag, uninitialized_relocate_tag>) {
-        uninitialized_relocate_n(Out.begin(), In.begin(), Out.size());
+        parlay::uninitialized_relocate(In.begin(), In.end(), Out.begin());
       }
       else {
         parallel_for(0, In.size(), [&](size_t i) {
@@ -248,7 +248,7 @@ sequence<size_t> integer_sort_r(slice<InIterator, InIterator> In,
 
     if constexpr (inplace_tag::value == true) {
       if (!one_bucket) {
-        uninitialized_relocate_n(In.begin(), Out.begin(), In.size());
+        parlay::uninitialized_relocate(Out.begin(), Out.end(), In.begin());
       }
     }
 

include/parlay/internal/merge_sort.h

@@ -28,7 +28,7 @@ void merge_sort_(slice<InIterator, InIterator> In,
     insertion_sort(In.begin(), In.size(), f);
     if (!inplace) {
       for (size_t i = 0; i < In.size(); i++) {
-        uninitialized_relocate(&Out[i], &In[i]);
+        relocate_at(&In[i], &Out[i]);
       }
     }
   }