Unverified Commit 52fc4c0e authored by Doron Behar's avatar Doron Behar Committed by GitHub
Browse files

python312Packages.mlx: upgrade and fix build (#367011)

parents a710670a 32ff266f

pkgs/by-name/op/openmpi/package.nix

+2 −2
Original line number Diff line number Diff line
@@ -85,6 +85,7 @@ stdenv.mkDerivation (finalAttrs: { 
      zlib

      libevent

      hwloc

      prrte

    ]

    ++ lib.optionals stdenv.hostPlatform.isLinux [

      libnl
@@ -92,7 +93,6 @@ stdenv.mkDerivation (finalAttrs: { 
      pmix

      ucx

      ucc

      prrte

    ]

    ++ lib.optionals cudaSupport [ cudaPackages.cuda_cudart ]

    ++ lib.optionals (stdenv.hostPlatform.isLinux || stdenv.hostPlatform.isFreeBSD) [ rdma-core ]
@@ -119,7 +119,7 @@ stdenv.mkDerivation (finalAttrs: { 
    "--with-pmix=${lib.getDev pmix}"

    "--with-pmix-libdir=${lib.getLib pmix}/lib"

    # Puts a "default OMPI_PRTERUN" value to mpirun / mpiexec executables

    (lib.withFeatureAs stdenv.hostPlatform.isLinux "prrte" (lib.getBin prrte))

    (lib.withFeatureAs true "prrte" (lib.getBin prrte))

    (lib.withFeature enableSGE "sge")

    (lib.enableFeature enablePrefix "mpirun-prefix-by-default")

    # TODO: add UCX support, which is recommended to use with cuda for the most robust OpenMPI build

pkgs/by-name/pr/prrte/package.nix

+1 −1
Original line number Diff line number Diff line
@@ -75,6 +75,6 @@ stdenv.mkDerivation rec { 
    homepage = "https://docs.prrte.org/";

    license = lib.licenses.bsd3;

    maintainers = with lib.maintainers; [ markuskowa ];

    platforms = lib.platforms.linux;

    platforms = lib.platforms.unix;

  };

}

pkgs/development/python-modules/mlx/default.nix

+6 −3
Original line number Diff line number Diff line
@@ -28,13 +28,13 @@ let 
in

buildPythonPackage rec {

  pname = "mlx";

  version = "0.18.0";

  version = "0.21.1";



  src = fetchFromGitHub {

    owner = "ml-explore";

    repo = "mlx";

    rev = "refs/tags/v${version}";

    hash = "sha256-eFKjCrutqrmhZKzRrLq5nYl0ieqLvoXpbnTxA1NEhWo=";

    hash = "sha256-wxv9bA9e8VyFv/FMh63sUTTNgkXHGQJNQhLuVynczZA=";

  };



  pyproject = true;
@@ -83,6 +83,9 @@ buildPythonPackage rec { 
    changelog = "https://github.com/ml-explore/mlx/releases/tag/v${version}";

    license = licenses.mit;

    platforms = [ "aarch64-darwin" ];

    maintainers = with maintainers; [ viraptor ];

    maintainers = with maintainers; [

      viraptor

      Gabriella439

    ];

  };

}