Add CustomMin auxilary snippet

NVIDIA · Mar 11, 2024 · 868d50c · 868d50c
1 parent d21af18
commit 868d50c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/cub/cub/device/device_segmented_reduce.cuh b/cub/cub/device/device_segmented_reduce.cuh
@@ -470,6 +470,12 @@ public:
   //! .. literalinclude:: ../../test/catch2_test_device_segmented_reduce_api.cu
   //!     :language: c++
   //!     :dedent:
+  //!     :start-after: example-begin segmented-reduce-custommin
+  //!     :end-before: example-end segmented-reduce-custommin
+  //!
+  //! .. literalinclude:: ../../test/catch2_test_device_segmented_reduce_api.cu
+  //!     :language: c++
+  //!     :dedent:
   //!     :start-after: example-begin segmented-reduce-min
   //!     :end-before: example-end segmented-reduce-min
   //!

diff --git a/cub/test/catch2_test_device_segmented_reduce_api.cu b/cub/test/catch2_test_device_segmented_reduce_api.cu
@@ -36,6 +36,7 @@
 #include "catch2_test_helper.h"
 #include "thrust/detail/raw_pointer_cast.h"
 
+// example-begin segmented-reduce-custommin
 struct CustomMin
 {
   template <typename T>
@@ -45,6 +46,8 @@ struct CustomMin
   }
 };
 
+// example-end segmented-reduce-custommin
+
 struct is_equal
 {
   __device__ bool operator()(cub::KeyValuePair<int, int> lhs, cub::KeyValuePair<int, int> rhs)