mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 17:31:16 +08:00
12 lines
370 B
CMake
12 lines
370 B
CMake
|
function(build_benchmark SRCFILE)
|
||
|
|
get_filename_component(src_name ${SRCFILE} NAME_WE)
|
||
|
|
set(target "${src_name}")
|
||
|
|
add_executable(${target} ${SRCFILE})
|
||
|
|
target_link_libraries(${target} PRIVATE mlx)
|
||
|
|
endfunction(build_benchmark)
|
||
|
|
|
||
|
|
build_benchmark(single_ops.cpp)
|
||
|
|
build_benchmark(irregular_strides.cpp)
|
||
|
|
build_benchmark(compare_devices.cpp)
|
||
|
|
build_benchmark(autograd.cpp)
|