# | |
# mnist | |
set(TEST_TARGET mnist) | |
add_executable(${TEST_TARGET} main.cpp) | |
target_link_libraries(${TEST_TARGET} PRIVATE ggml common) | |
# | |
# mnist-cnn | |
set(TEST_TARGET mnist-cnn) | |
add_executable(${TEST_TARGET} main-cnn.cpp) | |
target_link_libraries(${TEST_TARGET} PRIVATE ggml common) | |
# | |
# mnist-cpu | |
set(TEST_TARGET mnist-cpu) | |
add_executable(${TEST_TARGET} main-cpu.cpp) | |
target_link_libraries(${TEST_TARGET} PRIVATE ggml) | |
if (APPLE) | |
# | |
# mnist-mtl | |
find_library(FOUNDATION_LIBRARY Foundation REQUIRED) | |
find_library(METAL_FRAMEWORK Metal REQUIRED) | |
find_library(METALKIT_FRAMEWORK MetalKit REQUIRED) | |
find_library(METALPERFORMANCE_FRAMEWORK MetalPerformanceShaders REQUIRED) | |
set(TEST_TARGET mnist-mtl) | |
add_executable(${TEST_TARGET} main-mtl.cpp main-mtl.h main-mtl.m) | |
target_link_libraries(${TEST_TARGET} PRIVATE | |
ggml | |
${FOUNDATION_LIBRARY} | |
${METAL_FRAMEWORK} | |
${METALKIT_FRAMEWORK} | |
${METALPERFORMANCE_FRAMEWORK} | |
) | |
endif() | |