CMakeLists.txt 1.3 KB

1234567891011121314151617181920212223242526
  1. set(TARGET llama-eval-callback)
  2. add_executable(${TARGET} eval-callback.cpp)
  3. install(TARGETS ${TARGET} RUNTIME)
  4. target_link_libraries(${TARGET} PRIVATE common llama ${CMAKE_THREAD_LIBS_INIT})
  5. target_compile_features(${TARGET} PRIVATE cxx_std_17)
  6. if(LLAMA_BUILD_TESTS)
  7. if(NOT ${CMAKE_SYSTEM_PROCESSOR} MATCHES "s390x")
  8. set(MODEL_NAME "tinyllamas/stories15M-q4_0.gguf")
  9. set(MODEL_HASH "SHA256=66967fbece6dbe97886593fdbb73589584927e29119ec31f08090732d1861739")
  10. else()
  11. set(MODEL_NAME "tinyllamas/stories15M-be.Q4_0.gguf")
  12. set(MODEL_HASH "SHA256=9aec857937849d976f30397e97eb1cabb53eb9dcb1ce4611ba8247fb5f44c65d")
  13. endif()
  14. set(MODEL_DEST "${CMAKE_BINARY_DIR}/${MODEL_NAME}")
  15. set(TEST_TARGET test-eval-callback)
  16. add_test(NAME ${TEST_TARGET}-download-model COMMAND ${CMAKE_COMMAND}
  17. -DDEST=${MODEL_DEST}
  18. -DNAME=${MODEL_NAME}
  19. -DHASH=${MODEL_HASH}
  20. -P ${CMAKE_SOURCE_DIR}/cmake/download-models.cmake
  21. )
  22. set_tests_properties(${TEST_TARGET}-download-model PROPERTIES FIXTURES_SETUP ${TEST_TARGET}-download-model)
  23. add_test(NAME ${TEST_TARGET} COMMAND llama-eval-callback -m "${MODEL_DEST}" --prompt hello --seed 42 -ngl 0)
  24. set_tests_properties(${TEST_TARGET} PROPERTIES FIXTURES_REQUIRED ${TEST_TARGET}-download-model)
  25. endif()