mlx/examples/export/CMakeLists.txt

cmake_minimum_required(VERSION 3.27)

project(import_mlx LANGUAGES CXX)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

find_package(
  Python 3.9
  COMPONENTS Interpreter Development.Module
  REQUIRED)
execute_process(
  COMMAND "${Python_EXECUTABLE}" -m mlx --cmake-dir
  OUTPUT_STRIP_TRAILING_WHITESPACE
  OUTPUT_VARIABLE MLX_ROOT)
find_package(MLX CONFIG REQUIRED)

add_executable(eval_mlp eval_mlp.cpp)
target_link_libraries(eval_mlp PRIVATE mlx)

add_executable(train_mlp train_mlp.cpp)
target_link_libraries(train_mlp PRIVATE mlx)
Export / import functions to / from a file (#1642) * export and import functions * refactor + works for few primitives * nit * allow primitives with state * nit * nit * simplify serialize / deserialize * fix for constants * python bindings * maybe fix serialize failure case * add example * more primitives, training kind of works * same result for python and c++ * some fixes * fix export * template it up * some simplificatoin * rebase * allow kwargs and multiple functions * exporter * more primitives for exporting * deal with endianness * handle invalid stream * add docstring 2024-12-25 03:19:13 +08:00			`cmake_minimum_required(VERSION 3.27)`

			`project(import_mlx LANGUAGES CXX)`

			`set(CMAKE_CXX_STANDARD 17)`
			`set(CMAKE_CXX_STANDARD_REQUIRED ON)`

			`find_package(`
			`Python 3.9`
			`COMPONENTS Interpreter Development.Module`
			`REQUIRED)`
			`execute_process(`
MLX in C++ example (#1736) * MLX in C++ example * nits * fix docs 2025-01-03 11:09:04 +08:00			`COMMAND "${Python_EXECUTABLE}" -m mlx --cmake-dir`
Export / import functions to / from a file (#1642) * export and import functions * refactor + works for few primitives * nit * allow primitives with state * nit * nit * simplify serialize / deserialize * fix for constants * python bindings * maybe fix serialize failure case * add example * more primitives, training kind of works * same result for python and c++ * some fixes * fix export * template it up * some simplificatoin * rebase * allow kwargs and multiple functions * exporter * more primitives for exporting * deal with endianness * handle invalid stream * add docstring 2024-12-25 03:19:13 +08:00			`OUTPUT_STRIP_TRAILING_WHITESPACE`
			`OUTPUT_VARIABLE MLX_ROOT)`
			`find_package(MLX CONFIG REQUIRED)`

			`add_executable(eval_mlp eval_mlp.cpp)`
			`target_link_libraries(eval_mlp PRIVATE mlx)`

			`add_executable(train_mlp train_mlp.cpp)`
			`target_link_libraries(train_mlp PRIVATE mlx)`