Minimal MLX CMake

CMakeLists.txt

cmake_minimum_required(VERSION 3.27)

project(_ext LANGUAGES CXX)

# ----------------------------- Setup -----------------------------
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_POSITION_INDEPENDENT_CODE ON)

option(BUILD_SHARED_LIBS "Build as a shared library" ON)

# ----------------------------- Dependencies -----------------------------
find_package(MLX CONFIG REQUIRED)

add_executable(main main.cpp)
target_link_libraries(main PUBLIC mlx)

main.cpp

#include <iostream>
#include <mlx/mlx.h>

using namespace mlx::core;

int main () {
  auto a = array({1,2,3});
  std::cout << a << std::endl;
  return 0;
}

To build you should have MLX installed. If you installed it with Python you can do something like this:

mkdir build
cd build
MLX_DIR=`python -c "import mlx.core as mx; import os; print(os.path.dirname(mx.__file__))"` cmake ..
make -j4