@@ -156,25 +156,27 @@ if(EXECUTORCH_BUILD_LLAMA_JNI)
156156    ${CMAKE_CURRENT_BINARY_DIR} /../../examples/models/llama2/runner
157157  )
158158
159-     target_sources (
160-     executorch_jni PRIVATE 
161-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/mtk_llama_runner.cpp
162-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/LlamaModelChunk.cpp
163-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/LlamaRuntime.cpp
164-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/ModelChunk.cpp
165-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/MultiModelLoader.cpp
166-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/llm_helper/mask_builder.cpp
167-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/llm_helper/rotary_embedding.cpp
168-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/llm_helper/token_embedding.cpp
169-   )
170-   target_include_directories (
171-     executorch_jni PRIVATE 
172-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/
173-     ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner
174-   )
175-   ADD_LIBRARY (libneuron_buffer_allocator SHARED IMPORTED )
176-   SET_PROPERTY (TARGET  libneuron_buffer_allocator PROPERTY IMPORTED_LOCATION  ${NEURON_BUFFER_ALLOCATOR_LIB} )
177-   list (APPEND  link_libraries  neuron_backend libneuron_buffer_allocator)
159+   if (NEURON_BUFFER_ALLOCATOR_LIB)
160+       target_sources (
161+       executorch_jni PRIVATE 
162+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/mtk_llama_runner.cpp
163+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/LlamaModelChunk.cpp
164+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/LlamaRuntime.cpp
165+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/ModelChunk.cpp
166+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/MultiModelLoader.cpp
167+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/llm_helper/mask_builder.cpp
168+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/llm_helper/rotary_embedding.cpp
169+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner/llm_helper/token_embedding.cpp
170+     )
171+     target_include_directories (
172+       executorch_jni PRIVATE 
173+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/
174+       ${EXECUTORCH_ROOT} /examples/mediatek/executor_runner/llama_runner
175+     )
176+     ADD_LIBRARY (libneuron_buffer_allocator SHARED IMPORTED )
177+     SET_PROPERTY (TARGET  libneuron_buffer_allocator PROPERTY IMPORTED_LOCATION  ${NEURON_BUFFER_ALLOCATOR_LIB} )
178+     list (APPEND  link_libraries  neuron_backend libneuron_buffer_allocator)
179+   endif ()
178180endif ()
179181
180182target_include_directories (
0 commit comments