From 15d7373d21090c40a43f4ef023ec145719af1229 Mon Sep 17 00:00:00 2001
From: Bernhard Manfred Gruber <bernhardmgruber@gmail.com>
Date: Mon, 4 Nov 2024 14:03:56 +0100
Subject: [PATCH] Revert some feedback from miscco

We cannot include <cuda/functional> into non-CUDA translation units
---
 thrust/benchmarks/bench/transform/basic.cu | 2 +-
 thrust/testing/address_stability.cu        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/thrust/benchmarks/bench/transform/basic.cu b/thrust/benchmarks/bench/transform/basic.cu
index ffa4d3ca1d9..360f37951aa 100644
--- a/thrust/benchmarks/bench/transform/basic.cu
+++ b/thrust/benchmarks/bench/transform/basic.cu
@@ -33,7 +33,7 @@
 #include <thrust/transform.h>
 #include <thrust/zip_function.h>
 
-#include <cuda/functional>
+#include <cuda/__functional/address_stability.h>
 
 #include <nvbench_helper.cuh>
 
diff --git a/thrust/testing/address_stability.cu b/thrust/testing/address_stability.cu
index 65aa4121009..6336b6a8265 100644
--- a/thrust/testing/address_stability.cu
+++ b/thrust/testing/address_stability.cu
@@ -1,4 +1,4 @@
-#include <cuda/functional>
+#include <cuda/__functional/address_stability.h>
 
 #include <unittest/unittest.h>