Arm backend: Make memory pool sizes configrable from cmake (#5841)

Summary: Pull Request resolved: #5841

Reviewed By: mergennachin

Differential Revision: D63832189

Pulled By: digantdesai

fbshipit-source-id: 1ef435045b8821436f6b7272482ae0f7e30df33d

Author: zingo

examples/arm/executor_runner/CMakeLists.txt

@@ -7,6 +7,9 @@ cmake_minimum_required(VERSION 3.20)
 project(arm_executor_runner)
 
 option(SEMIHOSTING "Enable semihosting" OFF)
+option(ET_ARM_BAREMETAL_METHOD_ALLOCATOR_POOL_SIZE "Set ET_ARM_BAREMETAL_METHOD_ALLOCATOR_POOL_SIZE to specify memory alloction pool size" OFF)
+option(ET_ARM_BAREMETAL_TEMP_ALLOCATOR_POOL_SIZE "Set ET_ARM_BAREMETAL_TEMP_ALLOCATOR_POOL_SIZE to specify temp alloction pool size" OFF)
+
 
 if(NOT DEFINED ET_PTE_FILE_PATH AND NOT ${SEMIHOSTING})
   message(
@@ -172,6 +175,14 @@ else()
   add_dependencies(arm_executor_runner gen_model_header)
 endif()
 
+if(ET_ARM_BAREMETAL_METHOD_ALLOCATOR_POOL_SIZE)
+  target_compile_definitions(arm_executor_runner PUBLIC ET_ARM_BAREMETAL_METHOD_ALLOCATOR_POOL_SIZE=${ET_ARM_BAREMETAL_METHOD_ALLOCATOR_POOL_SIZE})
+endif()
+
+if(ET_ARM_BAREMETAL_TEMP_ALLOCATOR_POOL_SIZE)
+  target_compile_definitions(arm_executor_runner PUBLIC ET_ARM_BAREMETAL_TEMP_ALLOCATOR_POOL_SIZE=${ET_ARM_BAREMETAL_TEMP_ALLOCATOR_POOL_SIZE})
+endif()
+
 # Fixup compilation of retarget.c
 if(SEMIHOSTING)
   # Remove this when MLBEDSW-8910 is closed.

examples/arm/executor_runner/arm_executor_runner.cpp

@@ -392,6 +392,11 @@ int main(int argc, const char* argv[]) {
         (unsigned int)method_meta.error());
   }
 
+  ET_LOG(
+      Info,
+      "Setup Method allocator pool. Size: %lu bytes.",
+      method_allocation_pool_size);
+
   ArmMemoryAllocator method_allocator(
       method_allocation_pool_size, method_allocation_pool);