CMakeLists.txt 3.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. function(llama_build_executable source)
  2. get_filename_component(TEST_TARGET ${source} NAME_WE)
  3. add_executable(${TEST_TARGET} ${source})
  4. install(TARGETS ${TEST_TARGET} RUNTIME)
  5. target_link_libraries(${TEST_TARGET} PRIVATE common)
  6. endfunction()
  7. function(llama_test_executable name source)
  8. get_filename_component(TEST_TARGET ${source} NAME_WE)
  9. add_test(NAME ${name} COMMAND $<TARGET_FILE:${TEST_TARGET}> ${ARGN})
  10. endfunction()
  11. function(llama_build_and_test_executable source)
  12. get_filename_component(TEST_TARGET ${source} NAME_WE)
  13. add_executable(${TEST_TARGET} ${source})
  14. install(TARGETS ${TEST_TARGET} RUNTIME)
  15. target_link_libraries(${TEST_TARGET} PRIVATE common)
  16. add_test(NAME ${TEST_TARGET} COMMAND $<TARGET_FILE:${TEST_TARGET}> ${ARGN})
  17. endfunction()
  18. # llama_build_and_test_executable(test-double-float.cpp) # SLOW
  19. llama_build_and_test_executable(test-quantize-fns.cpp)
  20. llama_build_and_test_executable(test-quantize-perf.cpp)
  21. llama_build_and_test_executable(test-sampling.cpp)
  22. llama_build_executable(test-tokenizer-0-llama.cpp)
  23. llama_test_executable (test-tokenizer-0-llama test-tokenizer-0-llama.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-llama.gguf)
  24. llama_build_executable(test-tokenizer-0-falcon.cpp)
  25. llama_test_executable (test-tokenizer-0-falcon test-tokenizer-0-falcon.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-falcon.gguf)
  26. llama_build_executable(test-tokenizer-1-llama.cpp)
  27. llama_test_executable (test-tokenizer-1-llama test-tokenizer-1-llama.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-llama.gguf)
  28. llama_test_executable (test-tokenizer-1-baichuan test-tokenizer-1-llama.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-baichuan.gguf)
  29. llama_build_executable(test-tokenizer-1-bpe.cpp)
  30. llama_test_executable (test-tokenizer-1-falcon test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-falcon.gguf)
  31. llama_test_executable (test-tokenizer-1-aquila test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-aquila.gguf)
  32. llama_test_executable (test-tokenizer-1-mpt test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-mpt.gguf)
  33. llama_test_executable (test-tokenizer-1-stablelm-3b-4e1t test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-stablelm-3b-4e1t.gguf)
  34. llama_test_executable (test-tokenizer-1-gpt-neox test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-gpt-neox.gguf)
  35. llama_test_executable (test-tokenizer-1-refact test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-refact.gguf)
  36. llama_test_executable (test-tokenizer-1-starcoder test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-starcoder.gguf)
  37. llama_test_executable (test-tokenizer-1-gpt2 test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-gpt2.gguf)
  38. # llama_test_executable (test-tokenizer-1-bloom test-tokenizer-1-bpe.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../models/ggml-vocab-bloom.gguf) # BIG
  39. llama_build_and_test_executable(test-grammar-parser.cpp)
  40. llama_build_and_test_executable(test-llama-grammar.cpp)
  41. llama_build_and_test_executable(test-grad0.cpp)
  42. # llama_build_and_test_executable(test-opt.cpp) # SLOW
  43. llama_build_and_test_executable(test-backend-ops.cpp)
  44. llama_build_and_test_executable(test-rope.cpp)
  45. # dummy executable - not installed
  46. get_filename_component(TEST_TARGET test-c.c NAME_WE)
  47. add_executable(${TEST_TARGET} test-c.c)
  48. target_link_libraries(${TEST_TARGET} PRIVATE llama)