Skip to content

Commit 5c8a730

Browse files
committed
llava cmakelists
1 parent d9c31fa commit 5c8a730

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.ci/scripts/test_llava.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ run_and_verify() {
154154
EXPECTED_PREFIX="ASSISTANT: image captures a basketball game in progress, with several players on the court. One of the players is dribbling the ball, while the others are in various"
155155
else
156156
# set the expected prefix to be the same as prompt because there's a bug in sdpa_with_kv_cache that causes <unk> tokens.
157-
EXPECTED_PREFIX="ASSISTANT:"
157+
EXPECTED_PREFIX="ASSISTANT: image"
158158
fi
159159
if [[ "${RESULT}" == *"${EXPECTED_PREFIX}"* ]]; then
160160
echo "Expected result prefix: ${EXPECTED_PREFIX}"

examples/models/llava/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
# ~~~
1616
# It should also be cmake-lint clean.
1717
#
18-
cmake_minimum_required(VERSION 3.19)
18+
cmake_minimum_required(VERSION 3.24) # 3.24 is required for WHOLE_ARCHIVE
1919
project(llava)
2020

2121
# Duplicating options as root CMakeLists.txt
@@ -124,7 +124,7 @@ target_link_options_shared_lib(quantized_ops_lib)
124124
list(APPEND link_libraries quantized_kernels quantized_ops_lib)
125125

126126
if(EXECUTORCH_BUILD_KERNELS_CUSTOM)
127-
list(APPEND link_libraries custom_ops)
127+
list(APPEND link_libraries $<LINK_LIBRARY:WHOLE_ARCHIVE,custom_ops>)
128128
endif()
129129

130130
set(XNNPACK_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/../../../backends/xnnpack)

0 commit comments

Comments
 (0)