fix segfault in defragmentation (#1404)

Author: apolyakov

runtime-common/core/memory-resource/details/memory_ordered_chunk_list.cpp

@@ -31,14 +31,18 @@ void memory_ordered_chunk_list::save_next(list_node* node, const list_node* next
 }
 
 void memory_ordered_chunk_list::add_from_array(list_node** first, list_node** last) noexcept {
-  if (first == last) {
-    return;
-  }
-
+  // There may be memory blocks in the range [first, last) that do not belong to
+  // this memory resource's main arena, such as those from extra allocations.
+  // For now, simply skip these blocks.
   last = std::partition(first, last, [this](const auto* mem) {
     return reinterpret_cast<uintptr_t>(mem) >= reinterpret_cast<uintptr_t>(this->memory_resource_begin_) &&
            reinterpret_cast<uintptr_t>(mem) < reinterpret_cast<uintptr_t>(this->memory_resource_end_);
   });
+
+  if (first == last) {
+    return;
+  }
+
   std::sort(first, last, std::greater<>{});
 
   if (!head_) {

runtime-light/state/component-state.h

@@ -38,8 +38,7 @@ struct ComponentState final : private vk::not_copyable {
 private:
   static constexpr std::string_view INI_ARG_PREFIX = "ini ";
   static constexpr std::string_view RUNTIME_CONFIG_ARG = "runtime-config";
-  // FIXME: Temporary workaround (increased INIT_COMPONENT_ALLOCATOR_SIZE) to prevent segmentation fault during defragmentation
-  static constexpr auto INIT_COMPONENT_ALLOCATOR_SIZE = static_cast<size_t>(128U * 1024U * 1024U); // 128MB
+  static constexpr auto INIT_COMPONENT_ALLOCATOR_SIZE = static_cast<size_t>(1024U * 1024U); // 1MB
 
   void parse_env() noexcept;
 

runtime-light/state/image-state.h

@@ -87,5 +87,5 @@ struct ImageState final : private vk::not_copyable {
   }
 
 private:
-  static constexpr auto INIT_IMAGE_ALLOCATOR_SIZE = static_cast<size_t>(16U * 1024U * 1024U); // 16 MB
+  static constexpr auto INIT_IMAGE_ALLOCATOR_SIZE = static_cast<size_t>(1024U * 1024U); // 1MB
 };