commit 19500d862e3334785083ea5637f9b34d8c2ea42d · pyrox.dev/nixpkgs

+18 -7
pkgs/development/python-modules/mlx/default.nix
···

       1
       1
        
       {

     

       2
       2
       +
         lib,

     

       3
       3
       +
         stdenv,

     

       2
       4
        
         buildPythonPackage,

     

       3
       5
        
         fetchFromGitHub,

     

       4
       4
       -
         lib,

     

       5
       6
        
         replaceVars,

     

       6
       6
       -
         stdenv,

     

       7
       7
        
       

     

       8
       8
        
         # build-system

     

       9
       9
        
         setuptools,

     
···

       13
       13
        
       

     

       14
       14
        
         # buildInputs

     

       15
       15
        
         apple-sdk_14,

     

       16
       16
       -
         fmt_10,

     

       16
       16
       +
         fmt_11,

     

       17
       17
        
         nanobind,

     

       18
       18
        
         nlohmann_json,

     

       19
       19
        
         pybind11,

     
···

       36
       36
        
       

     

       37
       37
        
         mlx = buildPythonPackage rec {

     

       38
       38
        
           pname = "mlx";

     

       39
       39
       -
           version = "0.26.3";

     

       39
       39
       +
           version = "0.28.0";

     

       40
       40
        
           pyproject = true;

     

       41
       41
        
       

     

       42
       42
        
           src = fetchFromGitHub {

     

       43
       43
        
             owner = "ml-explore";

     

       44
       44
        
             repo = "mlx";

     

       45
       45
        
             tag = "v${version}";

     

       46
       46
       -
             hash = "sha256-hbqV/2KYGJ1gyExZd5bgaxTdhl5+Gext+U/+1KAztMU=";

     

       46
       46
       +
             hash = "sha256-+2dVZ89a09q8mWIbv6fBsySp7clzRV1tOyqr5hjFrNU=";

     

       47
       47
        
           };

     

       48
       48
        
       

     

       49
       49
        
           patches = [

     
···

       80
       80
        
               # hatches which let you interact with a native install of Xcode, such as `composeXcodeWrapper`

     

       81
       81
        
               # or by changing the upstream (e.g., https://github.com/zed-industries/zed/discussions/7016).

     

       82
       82
        
               (lib.cmakeBool "MLX_BUILD_METAL" false)

     

       83
       83
       +
               (lib.cmakeBool "USE_SYSTEM_FMT" true)

     

       83
       84
        
               (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_GGUFLIB" "${gguf-tools}")

     

       84
       85
        
               (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_JSON" "${nlohmann_json.src}")

     

       85
       85
       -
               (lib.cmakeOptionType "filepath" "FETCHCONTENT_SOURCE_DIR_FMT" "${fmt_10.src}")

     

       86
       86
        
             ];

     

       87
       87
        
           };

     

       88
       88
        
       

     
···

       96
       96
        
       

     

       97
       97
        
           buildInputs = [

     

       98
       98
        
             apple-sdk_14

     

       99
       99
       -
             fmt_10

     

       99
       99
       +
             fmt_11

     

       100
       100
        
             gguf-tools

     

       101
       101
        
             nanobind

     

       102
       102
        
             nlohmann_json

     
···

       113
       113
        
       

     

       114
       114
        
           enabledTestPaths = [

     

       115
       115
        
             "python/tests/"

     

       116
       116
       +
           ];

     

       117
       117
       +
       

     

       118
       118
       +
           disabledTests = [

     

       119
       119
       +
             # AssertionError

     

       120
       120
       +
             "test_numpy_conv"

     

       121
       121
       +
             "test_tensordot"

     

       122
       122
       +
           ];

     

       123
       123
       +
       

     

       124
       124
       +
           disabledTestPaths = [

     

       125
       125
       +
             # AssertionError

     

       126
       126
       +
             "python/tests/test_blas.py"

     

       116
       127
        
           ];

     

       117
       128
        
       

     

       118
       129
        
           # Additional testing by executing the example Python scripts supplied with mlx