From fac016f1173aacbf4500a28ef63ed683365e4d3f Mon Sep 17 00:00:00 2001
From: Remi Lehe <remi.lehe@normalesup.org>
Date: Thu, 26 Sep 2019 09:00:22 -0700
Subject: Implemented iterator distance

---
 Source/Particles/Sorting/SortingUtils.H | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'Source/Particles/Sorting/SortingUtils.H')

diff --git a/Source/Particles/Sorting/SortingUtils.H b/Source/Particles/Sorting/SortingUtils.H
index ede59b53b..7d53a352e 100644
--- a/Source/Particles/Sorting/SortingUtils.H
+++ b/Source/Particles/Sorting/SortingUtils.H
@@ -42,6 +42,19 @@ ForwardIterator stablePartition(ForwardIterator index_begin,
     return sep;
 }
 
+// TODO: Add documentation
+template< typename ForwardIterator >
+int iteratorDistance(ForwardIterator first,
+                     ForwardIterator last)
+{
+#ifdef AMREX_USE_GPU
+    // On GPU: Use thrust
+    return thrust::distance( first, last );
+#else
+    return std::distance( first, last );
+#endif
+}
+
 // TODO: Add documentation
 class fillBufferFlag
 {
-- 
cgit v1.2.3