[libcxx-commits] [PATCH] D108906: Add std::assume_aligned

[Gonzalo Brito Gadeschi via Phabricator via libcxx-commits]

gonzalobg updated this revision to Diff 369392.
gonzalobg added a comment.

Fix missing N in __builtin_assume_aligned


CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D108906/new/

https://reviews.llvm.org/D108906

Files:
  libcxx/include/memory
  libcxx/test/libcxx/memory/assume_aligned.pass.cpp


Index: libcxx/test/libcxx/memory/assume_aligned.pass.cpp
===================================================================
--- /dev/null
+++ libcxx/test/libcxx/memory/assume_aligned.pass.cpp
@@ -0,0 +1,31 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+
+// <memory>
+// UNSUPPORTED: c++03, c++11, c++14, c++17
+
+// template<std::size_t N, typename T>
+// [[nodiscard]] T* assume_aligned(T*);
+
+#include <memory>
+
+constexpr int test() {
+  int a;
+  int* b = std::assume_aligned<4>(&a);
+  return sizeof(int);
+}
+
+int main(int, char**)
+{
+  constexpr int N = test();
+  int* a = (int*)malloc(N);
+  int* b = std::assume_aligned<4>(a);
+
+  return 0;
+}
Index: libcxx/include/memory
===================================================================
--- libcxx/include/memory
+++ libcxx/include/memory
@@ -953,6 +953,19 @@
   }
 };
 
+#if _LIBCPP_STD_VER >= 20
+template <size_t N, typename T>
+[[nodiscard]]
+_LIBCPP_CONSTEXPR
+_LIBCPP_ALWAYS_INLINE
+T* assume_aligned(T* __ptr) {
+  #if defined(__clang__) || (defined(__GNUC__) && !defined(__ICC))
+    return reinterpret_cast<T*>(__builtin_assume_aligned(__ptr, N));
+  #else
+    return __ptr;
+  #endif
+}
+#endif // _LIBCPP_STD_VER >= 20
 
 _LIBCPP_END_NAMESPACE_STD
 


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D108906.369392.patch
Type: text/x-patch
Size: 1570 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/libcxx-commits/attachments/20210830/4222e877/attachment.bin>