From 0699874928757be3e6a8fd215637c33959f47604 Mon Sep 17 00:00:00 2001
From: Michael Wang <isVoid@users.noreply.github.com>
Date: Mon, 26 Jun 2023 19:21:12 +0000
Subject: [PATCH] Add thrust patches

---
 cpp/cmake/thirdparty/get_thrust.cmake         |  4 +++
 .../thirdparty/patches/thrust_override.json   | 13 ++++++++++
 ...ust_transform_iter_with_reduce_by_key.diff | 26 +++++++++++++++++++
 3 files changed, 43 insertions(+)
 create mode 100644 cpp/cmake/thirdparty/patches/thrust_override.json
 create mode 100644 cpp/cmake/thirdparty/patches/thrust_transform_iter_with_reduce_by_key.diff

diff --git a/cpp/cmake/thirdparty/get_thrust.cmake b/cpp/cmake/thirdparty/get_thrust.cmake
index 229e141c9..c881ff0e3 100644
--- a/cpp/cmake/thirdparty/get_thrust.cmake
+++ b/cpp/cmake/thirdparty/get_thrust.cmake
@@ -15,6 +15,10 @@
 # Use CPM to find or clone thrust
 function(find_and_configure_thrust)
         include(${rapids-cmake-dir}/cpm/thrust.cmake)
+        include(${rapids-cmake-dir}/cpm/package_override.cmake)
+
+        set(cuspatial_patch_dir "${CMAKE_CURRENT_FUNCTION_LIST_DIR}/patches")
+        rapids_cpm_package_override("${cuspatial_patch_dir}/thrust_override.json")
 
         rapids_cpm_thrust( NAMESPACE cuspatial
                            BUILD_EXPORT_SET cuspatial-exports
diff --git a/cpp/cmake/thirdparty/patches/thrust_override.json b/cpp/cmake/thirdparty/patches/thrust_override.json
new file mode 100644
index 000000000..62c379e68
--- /dev/null
+++ b/cpp/cmake/thirdparty/patches/thrust_override.json
@@ -0,0 +1,13 @@
+{
+  "packages" : {
+    "Thrust" : {
+      "patches" : [
+        {
+          "file" : "${current_json_dir}/thrust_transform_iter_with_reduce_by_key.diff",
+          "issue" : "Support transform_output_iterator as output of reduce by key [https://github.com/NVIDIA/thrust/pull/1805]",
+          "fixed_in" : "2.1"
+        }
+      ]
+    }
+  }
+}
diff --git a/cpp/cmake/thirdparty/patches/thrust_transform_iter_with_reduce_by_key.diff b/cpp/cmake/thirdparty/patches/thrust_transform_iter_with_reduce_by_key.diff
new file mode 100644
index 000000000..6a56af90d
--- /dev/null
+++ b/cpp/cmake/thirdparty/patches/thrust_transform_iter_with_reduce_by_key.diff
@@ -0,0 +1,26 @@
+diff --git a/thrust/iterator/transform_input_output_iterator.h b/thrust/iterator/transform_input_output_iterator.h
+index f512a36..a5f725d 100644
+--- a/thrust/iterator/transform_input_output_iterator.h
++++ b/thrust/iterator/transform_input_output_iterator.h
+@@ -102,6 +102,8 @@ template <typename InputFunction, typename OutputFunction, typename Iterator>
+   /*! \endcond
+    */
+
++  transform_input_output_iterator() = default;
++
+   /*! This constructor takes as argument a \c Iterator an \c InputFunction and an
+    * \c OutputFunction and copies them to a new \p transform_input_output_iterator
+    *
+diff --git a/thrust/iterator/transform_output_iterator.h b/thrust/iterator/transform_output_iterator.h
+index 66fb46a..4a68cb5 100644
+--- a/thrust/iterator/transform_output_iterator.h
++++ b/thrust/iterator/transform_output_iterator.h
+@@ -104,6 +104,8 @@ template <typename UnaryFunction, typename OutputIterator>
+   /*! \endcond
+    */
+
++  transform_output_iterator() = default;
++
+   /*! This constructor takes as argument an \c OutputIterator and an \c
+    * UnaryFunction and copies them to a new \p transform_output_iterator
+    *