added a new mhp::copy signature

remove ifdef BENCH_SHP in sort
oneapi-src · Nov 15, 2023 · faacf2b · faacf2b
1 parent 4848345
commit faacf2b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/benchmarks/gbench/common/sort.cpp b/benchmarks/gbench/common/sort.cpp
@@ -21,11 +21,11 @@ class DRSortFixture : public benchmark::Fixture {
     a = new xhp::distributed_vector<T>(default_vector_size);
     std::vector<T> local(default_vector_size);
     fill_random(local);
-#ifdef BENCH_SHP
+    // #ifdef BENCH_SHP
     xhp::copy(local.begin(), local.end(), rng::begin(*a));
-#else
-    xhp::copy(local, rng::begin(*a));
-#endif
+    // #else
+    //     xhp::copy(local, rng::begin(*a));
+    // #endif
   }
 
   void TearDown(::benchmark::State &) { delete a; }

diff --git a/include/dr/mhp/algorithms/copy.hpp b/include/dr/mhp/algorithms/copy.hpp
@@ -25,6 +25,12 @@ void copy(DI_IN &&first, DI_IN &&last, dr::distributed_iterator auto &&out) {
   copy(rng::subrange(first, last), out);
 }
 
+template <std::contiguous_iterator CI_IN>
+void copy(CI_IN &&first, CI_IN &&last,
+          dr::distributed_contiguous_iterator auto out) {
+  copy(0, rng::subrange(first, last), out);
+}
+
 /// Copy distributed to local
 void copy(std::size_t root, dr::distributed_contiguous_range auto &&in,
           std::contiguous_iterator auto out) {