CMakeLists.txt 1002 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. #
  2. # mnist
  3. set(TEST_TARGET mnist)
  4. add_executable(${TEST_TARGET} main.cpp)
  5. target_link_libraries(${TEST_TARGET} PRIVATE ggml common)
  6. #
  7. # mnist-cnn
  8. set(TEST_TARGET mnist-cnn)
  9. add_executable(${TEST_TARGET} main-cnn.cpp)
  10. target_link_libraries(${TEST_TARGET} PRIVATE ggml common)
  11. #
  12. # mnist-cpu
  13. set(TEST_TARGET mnist-cpu)
  14. add_executable(${TEST_TARGET} main-cpu.cpp)
  15. target_link_libraries(${TEST_TARGET} PRIVATE ggml)
  16. if (APPLE)
  17. #
  18. # mnist-mtl
  19. find_library(FOUNDATION_LIBRARY Foundation REQUIRED)
  20. find_library(METAL_FRAMEWORK Metal REQUIRED)
  21. find_library(METALKIT_FRAMEWORK MetalKit REQUIRED)
  22. find_library(METALPERFORMANCE_FRAMEWORK MetalPerformanceShaders REQUIRED)
  23. set(TEST_TARGET mnist-mtl)
  24. add_executable(${TEST_TARGET} main-mtl.cpp main-mtl.h main-mtl.m)
  25. target_link_libraries(${TEST_TARGET} PRIVATE
  26. ggml
  27. ${FOUNDATION_LIBRARY}
  28. ${METAL_FRAMEWORK}
  29. ${METALKIT_FRAMEWORK}
  30. ${METALPERFORMANCE_FRAMEWORK}
  31. )
  32. endif()