diff --git a/extension/llm/runner/test/test_text_llm_runner.cpp b/extension/llm/runner/test/test_text_llm_runner.cpp index 9a967dd82c1..f6f5137a90f 100644 --- a/extension/llm/runner/test/test_text_llm_runner.cpp +++ b/extension/llm/runner/test/test_text_llm_runner.cpp @@ -14,6 +14,7 @@ #include #include #include +#include #include #include @@ -145,6 +146,10 @@ class CallbackCounter { // Test fixture for Runner tests - minimal setup class RunnerTest : public Test { protected: + void SetUp() override { + executorch::runtime::runtime_init(); + } + // Helper functions to create and set up mock objects std::unique_ptr createMockTokenizer() { auto tokenizer = std::make_unique();