Migrate nvtext minhash, jaccard, and deduplicate APIs to cudf::memory_resources

Migrates minhash, minhash64, minhash_ngrams, minhash64_ngrams, jaccard_index,
build_suffix_array, resolve_duplicates, and resolve_duplicates_pair to
cudf::memory_resources and routes their temporary allocations through the temporary resource.

Contributes to NVIDIA#24268

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

Author: davidwendt

cpp/include/nvtext/deduplicate.hpp

@@ -35,14 +35,14 @@ namespace CUDF_EXPORT nvtext {
  * @param input Strings column to build suffix array for
  * @param min_width Minimum number of bytes that must match to identify a duplicate
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return Sorted suffix array and corresponding sizes
  */
 std::unique_ptr<rmm::device_uvector<cudf::size_type>> build_suffix_array(
   cudf::strings_column_view const& input,
   cudf::size_type min_width,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /**
  * @brief Returns duplicate strings found in the given input
@@ -61,15 +61,15 @@ std::unique_ptr<rmm::device_uvector<cudf::size_type>> build_suffix_array(
  * @param indices Suffix array from nvtext::build_suffix_array
  * @param min_width Minimum number of bytes that must match to identify a duplicate
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return New strings column with updated strings
  */
 std::unique_ptr<cudf::column> resolve_duplicates(
   cudf::strings_column_view const& input,
   cudf::device_span<cudf::size_type const> indices,
   cudf::size_type min_width,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /**
  * @brief Returns duplicate strings found from input1 found in the given input2
@@ -90,7 +90,7 @@ std::unique_ptr<cudf::column> resolve_duplicates(
  * @param indices2 Suffix array from nvtext::build_suffix_array for input2
  * @param min_width Minimum number of bytes that must match to identify a duplicate
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return New strings column with updated strings
  */
 std::unique_ptr<cudf::column> resolve_duplicates_pair(
@@ -99,8 +99,8 @@ std::unique_ptr<cudf::column> resolve_duplicates_pair(
   cudf::strings_column_view const& input2,
   cudf::device_span<cudf::size_type const> indices2,
   cudf::size_type min_width,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /** @} */  // end of group
 }  // namespace CUDF_EXPORT nvtext

cpp/include/nvtext/jaccard.hpp

@@ -57,15 +57,15 @@ namespace CUDF_EXPORT nvtext {
  * @param input2 Strings column to compare with `input1`
  * @param width The character width used for apply substrings
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return Index calculation values
  */
 std::unique_ptr<cudf::column> jaccard_index(
   cudf::strings_column_view const& input1,
   cudf::strings_column_view const& input2,
   cudf::size_type width,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /** @} */  // end of group
 }  // namespace CUDF_EXPORT nvtext

cpp/include/nvtext/minhash.hpp

@@ -56,7 +56,7 @@ namespace CUDF_EXPORT nvtext {
  * @param parameter_b Values used for the permuted calculation
  * @param width The character width of substrings to hash for each row
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return List column of minhash values for each string per seed
  */
 std::unique_ptr<cudf::column> minhash(
@@ -65,8 +65,8 @@ std::unique_ptr<cudf::column> minhash(
   cudf::device_span<uint32_t const> parameter_a,
   cudf::device_span<uint32_t const> parameter_b,
   cudf::size_type width,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /**
  * @brief Returns the minhash values for each string
@@ -103,7 +103,7 @@ std::unique_ptr<cudf::column> minhash(
  * @param parameter_b Values used for the permuted calculation
  * @param width The character width of substrings to hash for each row
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return List column of minhash values for each string per seed
  */
 std::unique_ptr<cudf::column> minhash64(
@@ -112,8 +112,8 @@ std::unique_ptr<cudf::column> minhash64(
   cudf::device_span<uint64_t const> parameter_a,
   cudf::device_span<uint64_t const> parameter_b,
   cudf::size_type width,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /**
  * @brief Returns the minhash values for each input row
@@ -150,7 +150,7 @@ std::unique_ptr<cudf::column> minhash64(
  * @param parameter_a Values used for the permuted calculation
  * @param parameter_b Values used for the permuted calculation
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return List column of minhash values for each string per seed
  */
 std::unique_ptr<cudf::column> minhash_ngrams(
@@ -159,8 +159,8 @@ std::unique_ptr<cudf::column> minhash_ngrams(
   uint32_t seed,
   cudf::device_span<uint32_t const> parameter_a,
   cudf::device_span<uint32_t const> parameter_b,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /**
  * @brief Returns the minhash values for each input row
@@ -197,7 +197,7 @@ std::unique_ptr<cudf::column> minhash_ngrams(
  * @param parameter_a Values used for the permuted calculation
  * @param parameter_b Values used for the permuted calculation
  * @param stream CUDA stream used for device memory operations and kernel launches
- * @param mr Device memory resource used to allocate the returned column's device memory
+ * @param mr Memory resources used for temporary allocations and the returned column
  * @return List column of minhash values for each string per seed
  */
 std::unique_ptr<cudf::column> minhash64_ngrams(
@@ -206,8 +206,8 @@ std::unique_ptr<cudf::column> minhash64_ngrams(
   uint64_t seed,
   cudf::device_span<uint64_t const> parameter_a,
   cudf::device_span<uint64_t const> parameter_b,
-  cuda::stream_ref stream           = cudf::get_default_stream(),
-  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
+  cuda::stream_ref stream   = cudf::get_default_stream(),
+  cudf::memory_resources mr = cudf::get_current_device_resource_ref());
 
 /** @} */  // end of group
 }  // namespace CUDF_EXPORT nvtext