enable std::assume_aligned

Author: fsb4000

stl/inc/memory

@@ -3079,15 +3079,21 @@ inline void* align(size_t _Bound, size_t _Size, void*& _Ptr, size_t& _Space) noe
     return _Ptr;
 }
 
-#if _HAS_CXX20 && 0
+#if _HAS_CXX20
 template <size_t _Nx, class _Ty>
 _NODISCARD constexpr _Ty* assume_aligned(_Ty* _Ptr) noexcept /* strengthened */ {
-    // this enforces the requirement that _Nx be a power of two
-    __builtin_assume_aligned(_Ptr, _Nx);
+#if 1 // TRANSITION: delete after __builtin_assume_aligned will be constexpr
+    if (!std::is_constant_evaluated()) {
+        return static_cast<_Ty*>(__builtin_assume_aligned(_Ptr, _Nx));
+    }
 
     return _Ptr;
+#else
+    // this enforces the requirement that _Nx be a power of two
+    return static_cast<_Ty*>(__builtin_assume_aligned(_Ptr, _Nx));
+#endif // TRANSITION: delete after __builtin_assume_aligned will be constexpr
 }
-#endif // _HAS_CXX20 && 0
+#endif // _HAS_CXX20
 
 // SPIN LOCKS
 _EXTERN_C

stl/inc/yvals_core.h

@@ -176,6 +176,7 @@
 // P0966R1 string::reserve() Should Not Shrink
 // P1001R2 execution::unseq
 // P1006R1 constexpr For pointer_traits<T*>::pointer_to()
+// P1007R3 std::assume_aligned
 // P1023R0 constexpr For std::array Comparisons
 // P1024R3 Enhancing span Usability
 // P1032R1 Miscellaneous constexpr

tests/std/test.lst

@@ -336,6 +336,7 @@ tests\P0898R3_identity
 tests\P0912R5_coroutine
 tests\P0919R3_heterogeneous_unordered_lookup
 tests\P0966R1_string_reserve_should_not_shrink
+tests\P1007R3_assume_aligned
 tests\P1023R0_constexpr_for_array_comparisons
 tests\P1032R1_miscellaneous_constexpr
 tests\P1135R6_atomic_flag_test

tests/std/tests/P1007R3_assume_aligned/env.lst

@@ -0,0 +1,4 @@
+# Copyright (c) Microsoft Corporation.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+RUNALL_INCLUDE ..\usual_latest_matrix.lst

tests/std/tests/P1007R3_assume_aligned/test.cpp

@@ -0,0 +1,44 @@
+// Copyright (c) Microsoft Corporation.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+#include <cassert>
+#include <memory>
+
+static void can_change_value_via_assume_aligned(int* p) {
+    int* p1 = std::assume_aligned<256>(p);
+    p1[0]   = 10;
+    p1[1]   = -15;
+}
+
+static constexpr int can_read_constexpr_value_via_assume_aligned_and_sum_it(int const* p, size_t size) {
+    int const* p1 = std::assume_aligned<256>(p);
+    int result    = 0;
+    for (size_t i = 0; i < size; ++i) {
+        result += p1[i];
+    }
+    return result;
+}
+
+
+static void assume_aligned_can_change_value() {
+    alignas(256) int arr[2] = {0, 0};
+    can_change_value_via_assume_aligned(&arr[0]);
+    assert(arr[0] == 10);
+    assert(arr[1] == -15);
+}
+
+static constexpr void constexpr_assume_aligned_can_read_value() {
+    alignas(256) constexpr int arr[3] = {10, 11, 9};
+    static_assert(can_read_constexpr_value_via_assume_aligned_and_sum_it(arr, 3) == 30);
+}
+
+static void assume_aligned_can_read_value() {
+    alignas(256) constexpr int arr[3] = {10, 11, 9};
+    assert(can_read_constexpr_value_via_assume_aligned_and_sum_it(arr, 3) == 30);
+}
+
+int main() {
+    assume_aligned_can_change_value();
+    constexpr_assume_aligned_can_read_value();
+    assume_aligned_can_read_value();
+}