File tree Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -155,6 +155,26 @@ if(EXECUTORCH_BUILD_LLAMA_JNI)
155
155
${EXECUTORCH_ROOT} /examples/models/llama2/runner
156
156
${CMAKE_CURRENT_BINARY_DIR} /../../examples/models/llama2/runner
157
157
)
158
+
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} /libneuron_buffer_allocator.so )
177
+ list (APPEND link_libraries neuron_backend libneuron_buffer_allocator )
158
178
endif ()
159
179
160
180
target_include_directories (
You can’t perform that action at this time.
0 commit comments