Workaround

NVIDIA · Feb 24, 2025 · 396f5a4 · 396f5a4
1 parent aabd073
commit 396f5a4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/thrust/thrust/iterator/iterator_traits.h b/thrust/thrust/iterator/iterator_traits.h
@@ -63,7 +63,13 @@ template <typename It>
 using it_reference_t = typename ::cuda::std::iterator_traits<It>::reference;
 
 template <typename It>
-using it_difference_t = typename ::cuda::std::iterator_traits<It>::difference_type;
+using it_difference_t = typename
+// FIXME(bgruber): switching to ::cuda::std::iterator_traits<T> breaks some tests, e.g.,
+// thrust.test.tabulate_output_iterator
+#if _CCCL_COMPILER(NVRTC)
+  ::cuda
+#endif // _CCCL_COMPILER(NVRTC)
+  ::std::iterator_traits<It>::difference_type;
 
 template <typename It>
 using it_pointer_t = typename ::cuda::std::iterator_traits<It>::pointer;

diff --git a/thrust/thrust/iterator/zip_iterator.h b/thrust/thrust/iterator/zip_iterator.h
@@ -66,7 +66,7 @@ struct make_zip_iterator_base
 template <typename... Its>
 struct make_zip_iterator_base<::cuda::std::tuple<Its...>>
 {
-  // reference type is the type of the tuple obtained from the iterators' reference types.
+  // reference type is the type of the tuple obtained from the iterator's reference types.
   using reference = tuple_of_iterator_references<it_reference_t<Its>...>;
 
   // Boost's Value type is the same as reference type. using value_type = reference;