commit e6afb967b698b45fdd9931b4aec94e63c4fce869 · pyrox.dev/nixpkgs

+15 -8

pkgs/development/python-modules/paddlepaddle/binary-hashes.nix

···

       2
        
         x86_64-linux = {

     

       3
        
           platform = "manylinux1_x86_64";

     

       4
        
           cpu = {

     

       5
       -
             cp39 = "sha256-Yu/FWoMhYp+behAth/jH0FKlf2LJr8TyvL9MBwmuews=";

     

       6
       -
             cp310 = "sha256-O7d/5LY2dEMf5gW5WrN3xzIIEi2vT0RWoMeVOk5lATk=";

     

       7
        
           };

     

       8
        
           gpu = {

     

       9
       -
             cp39 = "sha256-XHREY27jc+BrVyCJgpMvPVOFiKgPwuiNXPXO3biMLnc=";

     

       10
       -
             cp310 = "sha256-oTEBa26o5g6ruuTBgUljjDqign5fXmCn0EnL/0mv+ao=";

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       11
        
           };

     

       12
        
         };

     

       13
        
         x86_64-darwin = {

     

       14
        
           platform = "macosx_10_9_x86_64";

     

       15
        
           cpu = {

     

       16
       -
             cp39 = "sha256-5g9b2gC6uosMpoJiobpj8yToIS6ifAFRvLEqnc/o/QQ=";

     

       17
       -
             cp310 = "sha256-2c1hjwNCOOOx9tVfBk+Pyk/pF0m/2tAmRsBH91834eM=";

     

       18
        
           };

     

       19
        
         };

     

       20
        
         aarch64-darwin = {

     

       21
        
           platform = "macosx_11_0_arm64";

     

       22
        
           cpu = {

     

       23
       -
             cp39 = "sha256-JhYNTOx1UkuNf/63lHXBDry6FQjPnbIB8jU5jKcyX2k=";

     

       24
       -
             cp310 = "sha256-4ltYEYm2OzPBc6D2bQt2dEh6Sz+5m1mMKGGYgQGLSAY=";

     

       25
        
           };

     

       26
        
         };

     

       27
        
       }

···

       2
        
         x86_64-linux = {

     

       3
        
           platform = "manylinux1_x86_64";

     

       4
        
           cpu = {

     

       5
       +
             cp312 = "sha256-gafFsQFQsHUh0c0Ukdyh+3b/YhsU2xDomdlZ86d5Neo=";

     

       6
       +
             cp313 = "sha256-j8SGXv02Vu6ZQkEkeSy4imQhUbTVkafW1KXGr9rpWVk=";

     

       7
        
           };

     

       8
        
           gpu = {

     

       9
       +
             cp311 = "sha256-KWlGhjg9k1+wlm3Tk/mvMqh9LWZ0yGA1g99bCPlFf0U=";

     

       10
       +
             cp312 = "sha256-KJ2drJWLuwdaYsCj7egh1nQV4j35vT+UgH0qTdxoyHk=";

     

       11
       +
           };

     

       12
       +
         };

     

       13
       +
         aarch64-linux = {

     

       14
       +
           platform = "manylinux2014_aarch64";

     

       15
       +
           cpu = {

     

       16
       +
             cp312 = "sha256-3aqZaosKANvkJp2iHWUFKHfsNpOiLswHucraPs0RaIY=";

     

       17
       +
             cp313 = "sha256-u8TVc7NdJKJi4C1yaW6A9bSu5B9phnGvlXTe6xqD5vc=";

     

       18
        
           };

     

       19
        
         };

     

       20
        
         x86_64-darwin = {

     

       21
        
           platform = "macosx_10_9_x86_64";

     

       22
        
           cpu = {

     

       23
       +
             cp312 = "sha256-3P6/sQ3rFaoz0qLWbVoS2d5lRh2KQNJofi+zIhFQ0Lo=";

     

       24
       +
             cp313 = "sha256-UsQB/+Sq5WMWZgozAVpv11XNoj09cKKLE7c9cMvbuMs=";

     

       25
        
           };

     

       26
        
         };

     

       27
        
         aarch64-darwin = {

     

       28
        
           platform = "macosx_11_0_arm64";

     

       29
        
           cpu = {

     

       30
       +
             cp312 = "sha256-hnfo1C/2b3T7yjL/Mti2S749Vu0pqS1D3EGPDxaPy2k=";

     

       31
       +
             cp313 = "sha256-nRBR8uII2h1Dna7nyGG8tQJA8JcSSW62Hpzoxhj68vk=";

     

       32
        
           };

     

       33
        
         };

     

       34
        
       }

+51 -46

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

···

       1
        
       {

     

       2
       -
         stdenv,

     

       3
        
         config,

     

       4
        
         lib,

     

       0
        
       
     

       5
        
         buildPythonPackage,

     

       6
        
         fetchPypi,

     

       7
        
         python,

     

       8
        
         pythonOlder,

     

       9
        
         pythonAtLeast,

     

       10
       -
         openssl_1_1,

     

       11
        
         zlib,

     

       12
        
         setuptools,

     

       13
        
         cudaSupport ? config.cudaSupport or false,

     
···

       20
        
         pillow,

     

       21
        
         decorator,

     

       22
        
         astor,

     

       23
       -
         paddle-bfloat,

     

       24
        
         opt-einsum,

     

       0
        
       
     

       25
        
       }:

     

       26
        
       

     

       27
        
       let

     

       28
        
         pname = "paddlepaddle" + lib.optionalString cudaSupport "-gpu";

     

       29
       -
         version = "2.5.0";

     

       30
        
         format = "wheel";

     

       31
        
         pyShortVersion = "cp${builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion}";

     

       32
        
         cpuOrGpu = if cudaSupport then "gpu" else "cpu";

     

       33
        
         allHashAndPlatform = import ./binary-hashes.nix;

     

       34
        
         hash =

     

       35
       -
           allHashAndPlatform."${stdenv.system}"."${cpuOrGpu}"."${pyShortVersion}"

     

       36
       -
             or (throw "${pname} has no binary-hashes.nix entry for '${stdenv.system}.${cpuOrGpu}.${pyShortVersion}' attribute");

     

       37
       -
         platform = allHashAndPlatform."${stdenv.system}".platform;

     

       38
       -
         src = fetchPypi ({

     

       39
        
           inherit

     

       40
        
             version

     

       41
        
             format

     
···

       46
        
           dist = pyShortVersion;

     

       47
        
           python = pyShortVersion;

     

       48
        
           abi = pyShortVersion;

     

       49
       -
         });

     

       50
        
       in

     

       51
        
       buildPythonPackage {

     

       52
        
         inherit

     
···

       56
        
           src

     

       57
        
           ;

     

       58
        
       

     

       59
       -
         disabled = pythonOlder "3.9" || pythonAtLeast "3.11";

     

       60
       -
       

     

       61
       -
         libraryPath = lib.makeLibraryPath (

     

       62
       -
           # TODO: remove openssl_1_1 and zlib, maybe by building paddlepaddle from

     

       63
       -
           # source as suggested in the following comment:

     

       64
       -
           # https://github.com/NixOS/nixpkgs/pull/243583#issuecomment-1641450848

     

       65
       -
           [

     

       66
       -
             openssl_1_1

     

       67
       -
             zlib

     

       68
       -
           ]

     

       69
       -
           ++ lib.optionals cudaSupport (

     

       70
       -
             with cudaPackages_11;

     

       71
       -
             [

     

       72
       -
               cudatoolkit.lib

     

       73
       -
               cudatoolkit.out

     

       74
       -
               cudnn

     

       75
       -
             ]

     

       76
       -
           )

     

       77
       -
         );

     

       78
       -
       

     

       79
       -
         postFixup = lib.optionalString stdenv.hostPlatform.isLinux ''

     

       80
       -
           function fixRunPath {

     

       81
       -
             p=$(patchelf --print-rpath $1)

     

       82
       -
             patchelf --set-rpath "$p:$libraryPath" $1

     

       83
       -
             ${lib.optionalString cudaSupport ''

     

       84
       -
               addDriverRunpath $1

     

       85
       -
             ''}

     

       86
       -
           }

     

       87
       -
           fixRunPath $out/${python.sitePackages}/paddle/fluid/libpaddle.so

     

       88
       -
         '';

     

       89
        
       

     

       90
        
         nativeBuildInputs = [ addDriverRunpath ];

     

       91
        
       

     

       92
       -
         propagatedBuildInputs = [

     

       93
        
           setuptools

     

       94
        
           httpx

     

       95
        
           numpy

     
···

       97
        
           pillow

     

       98
        
           decorator

     

       99
        
           astor

     

       100
       -
           paddle-bfloat

     

       101
        
           opt-einsum

     

       0
        
       
     

       102
        
         ];

     

       103
        
       

     

       104
        
         pythonImportsCheck = [ "paddle" ];

     
···

       106
        
         # no tests

     

       107
        
         doCheck = false;

     

       108
        
       

     

       109
       -
         meta = with lib; {

     

       110
       -
           description = "PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署";

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       111
        
           homepage = "https://github.com/PaddlePaddle/Paddle";

     

       112
       -
           license = licenses.asl20;

     

       113
       -
           maintainers = with maintainers; [ happysalada ];

     

       114
        
           platforms =

     

       115
        
             [ "x86_64-linux" ]

     

       116
       -
             ++ optionals (!cudaSupport) [

     

       0
        
       
     

       117
        
               "x86_64-darwin"

     

       118
        
               "aarch64-darwin"

     

       119
        
             ];

···

       1
        
       {

     

       0
        
       
     

       2
        
         config,

     

       3
        
         lib,

     

       4
       +
         stdenv,

     

       5
        
         buildPythonPackage,

     

       6
        
         fetchPypi,

     

       7
        
         python,

     

       8
        
         pythonOlder,

     

       9
        
         pythonAtLeast,

     

       0
        
       
     

       10
        
         zlib,

     

       11
        
         setuptools,

     

       12
        
         cudaSupport ? config.cudaSupport or false,

     
···

       19
        
         pillow,

     

       20
        
         decorator,

     

       21
        
         astor,

     

       0
        
       
     

       22
        
         opt-einsum,

     

       23
       +
         typing-extensions,

     

       24
        
       }:

     

       25
        
       

     

       26
        
       let

     

       27
        
         pname = "paddlepaddle" + lib.optionalString cudaSupport "-gpu";

     

       28
       +
         version = if cudaSupport then "2.6.2" else "3.0.0";

     

       29
        
         format = "wheel";

     

       30
        
         pyShortVersion = "cp${builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion}";

     

       31
        
         cpuOrGpu = if cudaSupport then "gpu" else "cpu";

     

       32
        
         allHashAndPlatform = import ./binary-hashes.nix;

     

       33
        
         hash =

     

       34
       +
           allHashAndPlatform."${stdenv.hostPlatform.system}"."${cpuOrGpu}"."${pyShortVersion}"

     

       35
       +
             or (throw "${pname} has no binary-hashes.nix entry for '${stdenv.hostPlatform.system}.${cpuOrGpu}.${pyShortVersion}' attribute");

     

       36
       +
         platform = allHashAndPlatform."${stdenv.hostPlatform.system}".platform;

     

       37
       +
         src = fetchPypi {

     

       38
        
           inherit

     

       39
        
             version

     

       40
        
             format

     
···

       45
        
           dist = pyShortVersion;

     

       46
        
           python = pyShortVersion;

     

       47
        
           abi = pyShortVersion;

     

       48
       +
         };

     

       49
        
       in

     

       50
        
       buildPythonPackage {

     

       51
        
         inherit

     
···

       55
        
           src

     

       56
        
           ;

     

       57
        
       

     

       58
       +
         disabled =

     

       59
       +
           if cudaSupport then

     

       60
       +
             (pythonOlder "3.11" || pythonAtLeast "3.13")

     

       61
       +
           else

     

       62
       +
             (pythonOlder "3.12" || pythonAtLeast "3.14");

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       63
        
       

     

       64
        
         nativeBuildInputs = [ addDriverRunpath ];

     

       65
        
       

     

       66
       +
         dependencies = [

     

       67
        
           setuptools

     

       68
        
           httpx

     

       69
        
           numpy

     
···

       71
        
           pillow

     

       72
        
           decorator

     

       73
        
           astor

     

       0
        
       
     

       74
        
           opt-einsum

     

       75
       +
           typing-extensions

     

       76
        
         ];

     

       77
        
       

     

       78
        
         pythonImportsCheck = [ "paddle" ];

     
···

       80
        
         # no tests

     

       81
        
         doCheck = false;

     

       82
        
       

     

       83
       +
         postFixup = lib.optionalString stdenv.hostPlatform.isLinux (

     

       84
       +
           let

     

       85
       +
             libraryPath = lib.makeLibraryPath (

     

       86
       +
               [

     

       87
       +
                 zlib

     

       88
       +
                 (lib.getLib stdenv.cc.cc)

     

       89
       +
               ]

     

       90
       +
               ++ lib.optionals cudaSupport (

     

       91
       +
                 with cudaPackages_11;

     

       92
       +
                 [

     

       93
       +
                   cudatoolkit.lib

     

       94
       +
                   cudatoolkit.out

     

       95
       +
                   cudnn

     

       96
       +
                 ]

     

       97
       +
               )

     

       98
       +
             );

     

       99
       +
           in

     

       100
       +
           ''

     

       101
       +
             function fixRunPath {

     

       102
       +
               p=$(patchelf --print-rpath $1)

     

       103
       +
               patchelf --set-rpath "$p:${libraryPath}" $1

     

       104
       +
               ${lib.optionalString cudaSupport ''

     

       105
       +
                 addDriverRunpath $1

     

       106
       +
               ''}

     

       107
       +
             }

     

       108
       +
             fixRunPath $out/${python.sitePackages}/paddle/base/libpaddle.so

     

       109
       +
             fixRunPath $out/${python.sitePackages}/paddle/libs/lib*.so

     

       110
       +
           ''

     

       111
       +
         );

     

       112
       +
       

     

       113
       +
         meta = {

     

       114
       +
           description = "Machine Learning Framework from Industrial Practice";

     

       115
        
           homepage = "https://github.com/PaddlePaddle/Paddle";

     

       116
       +
           license = lib.licenses.asl20;

     

       117
       +
           maintainers = with lib.maintainers; [ happysalada ];

     

       118
        
           platforms =

     

       119
        
             [ "x86_64-linux" ]

     

       120
       +
             ++ lib.optionals (!cudaSupport) [

     

       121
       +
               "aarch64-linux"

     

       122
        
               "x86_64-darwin"

     

       123
        
               "aarch64-darwin"

     

       124
        
             ];