Use alloc_align for all coroutine-related heap-allocated types

Signed-off-by: Petr Shumilov <p.shumilov@vkteam.ru>

Author: PetrShumilov

runtime-light/coroutine/detail/await-set.h

@@ -61,6 +61,11 @@ class await_broker {
     return kphp::memory::script::alloc(n);
   }
 
+  template<typename... Args>
+  auto operator new(size_t n, std::align_val_t al, [[maybe_unused]] Args&&... args) noexcept -> void* {
+    return kphp::memory::script::alloc_aligned(n, al);
+  }
+
   void operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept {
     kphp::memory::script::free(ptr);
   }
@@ -168,6 +173,11 @@ class await_set_task_promise_base : public kphp::coro::async_stack_element {
     return kphp::memory::script::alloc(n);
   }
 
+  template<typename... Args>
+  auto operator new(size_t n, std::align_val_t al, [[maybe_unused]] Args&&... args) noexcept -> void* {
+    return kphp::memory::script::alloc_aligned(n, al);
+  }
+
   void operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept {
     kphp::memory::script::free(ptr);
   }

runtime-light/coroutine/detail/task-self-deleting.h

@@ -35,6 +35,11 @@ struct promise_self_deleting : kphp::coro::async_stack_element {
     return kphp::memory::script::alloc(n);
   }
 
+  template<typename... Args>
+  auto operator new(size_t n, std::align_val_t al, [[maybe_unused]] Args&&... args) noexcept -> void* {
+    return kphp::memory::script::alloc_aligned(n, al);
+  }
+
   auto operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept -> void {
     kphp::memory::script::free(ptr);
   }

runtime-light/coroutine/detail/when-all.h

@@ -155,6 +155,11 @@ class when_all_task_promise_base : public kphp::coro::async_stack_element {
     return kphp::memory::script::alloc(n);
   }
 
+  template<typename... Args>
+  auto operator new(size_t n, std::align_val_t al, [[maybe_unused]] Args&&... args) noexcept -> void* {
+    return kphp::memory::script::alloc_aligned(n, al);
+  }
+
   auto operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept -> void {
     kphp::memory::script::free(ptr);
   }

runtime-light/coroutine/detail/when-any.h

@@ -165,6 +165,11 @@ class when_any_task_promise_base : public kphp::coro::async_stack_element {
     return kphp::memory::script::alloc(n);
   }
 
+  template<typename... Args>
+  auto operator new(size_t n, std::align_val_t al, [[maybe_unused]] Args&&... args) noexcept -> void* {
+    return kphp::memory::script::alloc_aligned(n, al);
+  }
+
   auto operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept -> void {
     kphp::memory::script::free(ptr);
   }

runtime-light/coroutine/shared-task.h

@@ -148,6 +148,11 @@ struct promise_base : kphp::coro::async_stack_element {
     return kphp::memory::script::alloc(n);
   }
 
+  template<typename... Args>
+  auto operator new(size_t n, std::align_val_t al, [[maybe_unused]] Args&&... args) noexcept -> void* {
+    return kphp::memory::script::alloc_aligned(n, al);
+  }
+
   auto operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept -> void {
     kphp::memory::script::free(ptr);
   }

runtime-light/coroutine/task.h

@@ -74,7 +74,7 @@ struct promise_base : kphp::coro::async_stack_element {
     return kphp::memory::script::alloc_aligned(n, al);
   }
 
-  auto operator delete([[maybe_unused]] void* ptr, [[maybe_unused]] size_t n) noexcept -> void {
+  auto operator delete(void* ptr, [[maybe_unused]] size_t n) noexcept -> void {
     kphp::memory::script::free(ptr);
   }