Test proclaim_copyable_arguments for lambdas

NVIDIA · Nov 15, 2024 · 9c84e62 · 9c84e62
1 parent bb1c7e7
commit 9c84e62
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/thrust/testing/address_stability.cmake b/thrust/testing/address_stability.cmake
@@ -0,0 +1 @@
+target_compile_options(${test_target} PRIVATE $<$<COMPILE_LANG_AND_ID:CUDA,NVIDIA>:--extended-lambda>)
diff --git a/thrust/testing/address_stability.cu b/thrust/testing/address_stability.cu
@@ -83,3 +83,38 @@ void TestAddressStabilityUserDefinedFunctionObject()
   static_assert(proclaims_copyable_arguments<decltype(proclaim_copyable_arguments(my_plus<const int&&>{}))>::value, "");
 }
 DECLARE_UNITTEST(TestAddressStabilityUserDefinedFunctionObject);
+
+void TestAddressStabilityLambda()
+{
+  using ::cuda::proclaim_copyable_arguments;
+  using ::cuda::proclaims_copyable_arguments;
+
+  {
+    auto l = [](const int& i) {
+      return i + 2;
+    };
+    static_assert(!proclaims_copyable_arguments<decltype(l)>::value, "");
+    auto pr_l = proclaim_copyable_arguments(l);
+    ASSERT_EQUAL(pr_l(3), 5);
+    static_assert(proclaims_copyable_arguments<decltype(pr_l)>::value, "");
+  }
+
+  {
+    auto l = [] __device__(const int& i) {
+      return i + 2;
+    };
+    static_assert(!proclaims_copyable_arguments<decltype(l)>::value, "");
+    auto pr_device_l = proclaim_copyable_arguments(l);
+    (void) pr_device_l;
+    static_assert(proclaims_copyable_arguments<decltype(pr_device_l)>::value, "");
+  }
+
+  {
+    auto l = [] __host__ __device__(const int& i) { return i + 2; };
+    static_assert(!proclaims_copyable_arguments<decltype(l)>::value, "");
+    auto pr_l = proclaim_copyable_arguments(l);
+    ASSERT_EQUAL(pr_l(3), 5);
+    static_assert(proclaims_copyable_arguments<decltype(pr_l)>::value, "");
+  }
+}
+DECLARE_UNITTEST(TestAddressStabilityLambda);