|
1 | 1 | /* |
2 | | - * SPDX-FileCopyrightText: Copyright (c) 2020-2025, NVIDIA CORPORATION. |
| 2 | + * SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION. |
3 | 3 | * SPDX-License-Identifier: Apache-2.0 |
4 | 4 | */ |
5 | 5 | #pragma once |
@@ -79,8 +79,8 @@ struct device_isend_tuple_iterator_element_impl { |
79 | 79 | int base_tag, |
80 | 80 | raft::comms::request_t* requests) const |
81 | 81 | { |
82 | | - using output_value_t = typename thrust:: |
83 | | - tuple_element<I, typename std::iterator_traits<OutputIterator>::value_type>::type; |
| 82 | + using output_value_t = |
| 83 | + cuda::std::tuple_element_t<I, typename std::iterator_traits<OutputIterator>::value_type>; |
84 | 84 | auto tuple_element_input_first = cuda::std::get<I>(input_first.get_iterator_tuple()); |
85 | 85 | device_isend_impl<decltype(tuple_element_input_first), output_value_t>( |
86 | 86 | comm, tuple_element_input_first, count, dst, static_cast<int>(base_tag + I), requests + I); |
@@ -139,8 +139,8 @@ struct device_irecv_tuple_iterator_element_impl { |
139 | 139 | int base_tag, |
140 | 140 | raft::comms::request_t* requests) const |
141 | 141 | { |
142 | | - using input_value_t = typename thrust:: |
143 | | - tuple_element<I, typename std::iterator_traits<InputIterator>::value_type>::type; |
| 142 | + using input_value_t = |
| 143 | + cuda::std::tuple_element_t<I, typename std::iterator_traits<InputIterator>::value_type>; |
144 | 144 | auto tuple_element_output_first = cuda::std::get<I>(output_first.get_iterator_tuple()); |
145 | 145 | device_irecv_impl<input_value_t, decltype(tuple_element_output_first)>( |
146 | 146 | comm, tuple_element_output_first, count, src, static_cast<int>(base_tag + I), requests + I); |
@@ -211,8 +211,8 @@ struct device_sendrecv_tuple_iterator_element_impl { |
211 | 211 | int src, |
212 | 212 | rmm::cuda_stream_view stream_view) const |
213 | 213 | { |
214 | | - using output_value_t = typename thrust:: |
215 | | - tuple_element<I, typename std::iterator_traits<OutputIterator>::value_type>::type; |
| 214 | + using output_value_t = |
| 215 | + cuda::std::tuple_element_t<I, typename std::iterator_traits<OutputIterator>::value_type>; |
216 | 216 | auto tuple_element_input_first = cuda::std::get<I>(input_first.get_iterator_tuple()); |
217 | 217 | auto tuple_element_output_first = cuda::std::get<I>(output_first.get_iterator_tuple()); |
218 | 218 | device_sendrecv_impl<decltype(tuple_element_input_first), decltype(tuple_element_output_first)>( |
@@ -301,8 +301,8 @@ struct device_multicast_sendrecv_tuple_iterator_element_impl { |
301 | 301 | raft::host_span<int const> rx_src_ranks, |
302 | 302 | rmm::cuda_stream_view stream_view) const |
303 | 303 | { |
304 | | - using output_value_t = typename thrust:: |
305 | | - tuple_element<I, typename std::iterator_traits<OutputIterator>::value_type>::type; |
| 304 | + using output_value_t = |
| 305 | + cuda::std::tuple_element_t<I, typename std::iterator_traits<OutputIterator>::value_type>; |
306 | 306 | auto tuple_element_input_first = cuda::std::get<I>(input_first.get_iterator_tuple()); |
307 | 307 | auto tuple_element_output_first = cuda::std::get<I>(output_first.get_iterator_tuple()); |
308 | 308 | device_multicast_sendrecv_impl<decltype(tuple_element_input_first), |
|
0 commit comments