commit b4d192a9a03ed0a492443947efb17e1b6c1d2848 · pyrox.dev/nixpkgs

+15
pkgs/development/python-modules/vllm/default.nix
···

       5
       5
        
         buildPythonPackage,

     

       6
       6
        
         pythonAtLeast,

     

       7
       7
        
         fetchFromGitHub,

     

       8
       8
       +
         fetchpatch,

     

       8
       9
        
         symlinkJoin,

     

       9
       10
        
         autoAddDriverRunpath,

     

       10
       11
        
       

     
···

       144
       145
        
             rev = "57b4e68b9f9d94750b46de8f8dbd2bfcc86edd4f";

     

       145
       146
        
             hash = "sha256-c7L7WZVVEnXMOTPBoSp7jhkl9d4TA4sj11QvOSWTDIE=";

     

       146
       147
        
           };

     

       148
       148
       +
       

     

       149
       149
       +
           patches = [

     

       150
       150
       +
             # fix Hopper build failure

     

       151
       151
       +
             # https://github.com/Dao-AILab/flash-attention/pull/1719

     

       152
       152
       +
             # https://github.com/Dao-AILab/flash-attention/pull/1723

     

       153
       153
       +
             (fetchpatch {

     

       154
       154
       +
               url = "https://github.com/Dao-AILab/flash-attention/commit/dad67c88d4b6122c69d0bed1cebded0cded71cea.patch";

     

       155
       155
       +
               hash = "sha256-JSgXWItOp5KRpFbTQj/cZk+Tqez+4mEz5kmH5EUeQN4=";

     

       156
       156
       +
             })

     

       157
       157
       +
             (fetchpatch {

     

       158
       158
       +
               url = "https://github.com/Dao-AILab/flash-attention/commit/e26dd28e487117ee3e6bc4908682f41f31e6f83a.patch";

     

       159
       159
       +
               hash = "sha256-NkCEowXSi+tiWu74Qt+VPKKavx0H9JeteovSJKToK9A=";

     

       160
       160
       +
             })

     

       161
       161
       +
           ];

     

       147
       162
        
       

     

       148
       163
        
           dontConfigure = true;

     

       149
       164