Fix arm CI test (pytorch#1917)

Summary:

https://github.com/pytorch/executorch/actions/runs/7849522096/job/21423067611

Let's use the new shiny API instead

Differential Revision: D53627686

Author: mergennachin

examples/arm/ethos-u-setup/core_platform/patches/0006-Executorch-Add-executorch_runner-test.patch

@@ -6,9 +6,9 @@ Subject: [Executorch 6/7] Add executorch_runner test
 ---
  applications/CMakeLists.txt                  |   2 +
  applications/executorch_tests/CMakeLists.txt |  76 +++++++++++
- applications/executorch_tests/runner.cpp     | 133 +++++++++++++++++++
+ applications/executorch_tests/runner.cpp     | 136 +++++++++++++++++++
  cmake/helpers.cmake                          |  13 +-
- 4 files changed, 222 insertions(+), 2 deletions(-)
+ 4 files changed, 225 insertions(+), 2 deletions(-)
  create mode 100644 applications/executorch_tests/CMakeLists.txt
  create mode 100644 applications/executorch_tests/runner.cpp
 
@@ -17,9 +17,9 @@ index 1fa2b2e..68e5427 100644
 --- a/applications/CMakeLists.txt
 +++ b/applications/CMakeLists.txt
 @@ -28,6 +28,8 @@ add_subdirectory(threadx_demo)
- 
+
  add_subdirectory(message_handler_openamp)
- 
+
 +add_subdirectory(executorch_tests)
 +
  if (CMAKE_CXX_COMPILER_ID STREQUAL "ARMClang")
@@ -109,10 +109,10 @@ index 0000000..c95d53e
 +# TODO Memory setup
 diff --git a/applications/executorch_tests/runner.cpp b/applications/executorch_tests/runner.cpp
 new file mode 100644
-index 0000000..7ef920d
+index 0000000..4f9d334
 --- /dev/null
 +++ b/applications/executorch_tests/runner.cpp
-@@ -0,0 +1,133 @@
+@@ -0,0 +1,136 @@
 +/* Copyright (c) Meta Platforms, Inc. and affiliates.
 + * All rights reserved.
 + *
@@ -125,11 +125,11 @@ index 0000000..7ef920d
 +#include <memory>
 +
 +#include <executorch/extension/data_loader/buffer_data_loader.h>
++#include <executorch/extension/runner_util/inputs.h>
 +#include <executorch/runtime/executor/program.h>
 +#include <executorch/runtime/platform/log.h>
 +#include <executorch/runtime/platform/platform.h>
 +#include <executorch/runtime/platform/runtime.h>
-+#include <executorch/util/util.h>
 +
 +// Model file - TODO make this configurable through CMake
 +#include "model_pte.h"
@@ -224,7 +224,10 @@ index 0000000..7ef920d
 +    ET_LOG(Info,"Method loaded.");
 +
 +    ET_LOG(Info,"Preparing inputs...");
-+    auto inputs = torch::executor::util::PrepareInputTensors(*method);
++    auto inputs = torch::executor::util::prepare_input_tensors(*method);
++    if (!inputs.ok()) {
++        ET_LOG(Info, "Preparing inputs tensors for method %s failed with status 0x%" PRIx32, method_name, inputs.error());
++    }
 +    ET_LOG(Info,"Input prepared.");
 +
 +    ET_LOG(Info,"Starting the model execution...");
@@ -252,17 +255,17 @@ index a21d9f0..036f189 100644
 +++ b/cmake/helpers.cmake
 @@ -85,7 +85,7 @@ endfunction()
  #############################################################################
- 
+
  function(ethosu_add_executable target)
 -    cmake_parse_arguments(ARGS "" "TARGET_LIBRARY" "SOURCES;LIBRARIES" ${ARGN})
 +    cmake_parse_arguments(ARGS "WHOLE_ARCHIVE" "TARGET_LIBRARY" "SOURCES;LIBRARIES" ${ARGN})
      add_executable(${target})
- 
+
      target_sources(${target} PRIVATE
 @@ -95,8 +95,17 @@ function(ethosu_add_executable target)
          set(ARGS_TARGET_LIBRARY ethosu_target_init)
      endif()
- 
+
 +    if (ARGS_WHOLE_ARCHIVE)
 +        set(PRE_LINKER_FLAGS "-Wl,--whole-archive")
 +        set(POST_LINKER_FLAGS "-Wl,--no-whole-archive")
@@ -271,13 +274,12 @@ index a21d9f0..036f189 100644
      target_link_libraries(${target} PRIVATE
 -        ${ARGS_TARGET_LIBRARY} ${ARGS_LIBRARIES})
 +        ${PRE_LINKER_FLAGS}
-+        ${ARGS_TARGET_LIBRARY} 
++        ${ARGS_TARGET_LIBRARY}
 +        ${ARGS_LIBRARIES}
 +        ${POST_LINKER_FLAGS}
 +        )
- 
+
      ethosu_eval_link_options(${target})
- 
--- 
-2.42.0
 
+--
+2.42.0