From 0b93be7551673ac5812db3236f63683f7772363d Mon Sep 17 00:00:00 2001
From: Patrick Stotko <stotko@cs.uni-bonn.de>
Date: Sat, 7 Feb 2026 13:12:22 +0100
Subject: [PATCH] General: Add support for CUDA 13.1

---
 src/stdgpu/impl/iterator_detail.h | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/stdgpu/impl/iterator_detail.h b/src/stdgpu/impl/iterator_detail.h
index 513d2e5fd..a8287cf9b 100644
--- a/src/stdgpu/impl/iterator_detail.h
+++ b/src/stdgpu/impl/iterator_detail.h
@@ -19,6 +19,7 @@
 #include <cstdio>
 #include <thrust/iterator/discard_iterator.h>
 #include <thrust/iterator/iterator_adaptor.h>
+#include <thrust/version.h>
 #include <type_traits>
 
 namespace stdgpu
@@ -392,6 +393,17 @@ THRUST_NAMESPACE_BEGIN
 namespace detail
 {
 
+// thrust 3.1 (CUDA 13.1) refactored their internal type traits towards C++17 style
+#if THRUST_MAJOR_VERSION >= 3 && THRUST_MINOR_VERSION >= 1
+template <typename Container>
+inline constexpr bool is_proxy_reference_v<stdgpu::detail::back_insert_iterator_proxy<Container>> = true;
+
+template <typename Container>
+inline constexpr bool is_proxy_reference_v<stdgpu::detail::front_insert_iterator_proxy<Container>> = true;
+
+template <typename Container>
+inline constexpr bool is_proxy_reference_v<stdgpu::detail::insert_iterator_proxy<Container>> = true;
+#else
 template <typename Container>
 struct is_proxy_reference<stdgpu::detail::back_insert_iterator_proxy<Container>> : public thrust::detail::true_type
 {
@@ -406,6 +418,7 @@ template <typename Container>
 struct is_proxy_reference<stdgpu::detail::insert_iterator_proxy<Container>> : public thrust::detail::true_type
 {
 };
+#endif
 
 } // namespace detail
 THRUST_NAMESPACE_END