78 lines
2 KiB
Nix
78 lines
2 KiB
Nix
{ stdenv
|
|
, lib
|
|
, fetchFromGitHub
|
|
, buildPythonPackage
|
|
, python3Packages
|
|
, pybind11
|
|
, cmake
|
|
, xcbuild
|
|
, zsh
|
|
, darwin
|
|
, blas
|
|
, lapack
|
|
}:
|
|
|
|
let
|
|
# static dependencies included directly during compilation
|
|
gguf-tools = fetchFromGitHub {
|
|
owner = "antirez";
|
|
repo = "gguf-tools";
|
|
rev = "af7d88d808a7608a33723fba067036202910acb3";
|
|
hash = "sha256-LqNvnUbmq0iziD9VP5OTJCSIy+y/hp5lKCUV7RtKTvM=";
|
|
};
|
|
nlohmann_json = fetchFromGitHub {
|
|
owner = "nlohmann";
|
|
repo = "json";
|
|
rev = "v3.11.3";
|
|
hash = "sha256-7F0Jon+1oWL7uqet5i1IgHX0fUw/+z0QwEcA3zs5xHg=";
|
|
};
|
|
in
|
|
buildPythonPackage rec {
|
|
pname = "mlx";
|
|
version = "0.6.0";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "ml-explore";
|
|
repo = "mlx";
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-FihdI+3ACKMJfPT2POjTRdtkXs7x+KiQpdpo3RcczBE=";
|
|
};
|
|
|
|
pyproject = true;
|
|
|
|
patches = [
|
|
# With Darwin SDK 11 we cannot include vecLib/cblas_new.h, this needs to wait for PR #229210
|
|
# In the meantime, pretend Accelerate is not available and use blas/lapack instead.
|
|
./disable-accelerate.patch
|
|
];
|
|
|
|
postPatch = ''
|
|
substituteInPlace CMakeLists.txt \
|
|
--replace "/usr/bin/xcrun" "${xcbuild}/bin/xcrun" \
|
|
'';
|
|
|
|
dontUseCmakeConfigure = true;
|
|
|
|
env = {
|
|
PYPI_RELEASE = version;
|
|
# we can't use Metal compilation with Darwin SDK 11
|
|
CMAKE_ARGS = toString [
|
|
(lib.cmakeBool "MLX_BUILD_METAL" false)
|
|
(lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_GGUFLIB" "${gguf-tools}")
|
|
(lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_JSON" "${nlohmann_json}")
|
|
];
|
|
};
|
|
|
|
nativeBuildInputs = [ cmake pybind11 xcbuild zsh gguf-tools nlohmann_json ] ++ (with python3Packages; [ setuptools ]);
|
|
|
|
buildInputs = [ blas lapack ];
|
|
|
|
meta = with lib; {
|
|
homepage = "https://github.com/ml-explore/mlx";
|
|
description = "An array framework for Apple silicon";
|
|
changelog = "https://github.com/ml-explore/mlx/releases/tag/v${version}";
|
|
license = licenses.mit;
|
|
platforms = [ "aarch64-darwin" ];
|
|
maintainers = with maintainers; [ viraptor ];
|
|
};
|
|
}
|