Fix iterator traits

NVIDIA · Feb 6, 2025 · d52cf58 · d52cf58
1 parent 9df7a86
commit d52cf58
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/thrust/testing/scan.cu b/thrust/testing/scan.cu
@@ -574,6 +574,16 @@ struct iterator_traits<only_set_when_expected_it>
 };
 THRUST_NAMESPACE_END
 
+namespace std
+{
+template <>
+struct iterator_traits<only_set_when_expected_it>
+{
+  using value_type = long long;
+  using reference  = only_set_when_expected_it;
+};
+} // namespace std
+
 void TestInclusiveScanWithBigIndexesHelper(int magnitude)
 {
   thrust::constant_iterator<long long> begin(1);