Skip to content

Commit 7020c1a

Browse files
committed
protect cmakelist for extension under NEURON_BUFFER_ALLOCATOR_LIB flag
1 parent ba70f2d commit 7020c1a

File tree

1 file changed

+21
-19
lines changed

1 file changed

+21
-19
lines changed

extension/android/CMakeLists.txt

Lines changed: 21 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -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()
178180
endif()
179181

180182
target_include_directories(

0 commit comments

Comments
 (0)