commit f9ffac5a3e8db6fa1b9f06e1f70b002c2ba963fb · pyrox.dev/nixpkgs

+4 -2
pkgs/development/libraries/onnxruntime/default.nix
···

       81
       81
        
           hash = "sha256-9oORW0YlQ6SphqfbjcYb0dTlHc+1gzy9quH/Lj6By8Q=";

     

       82
       82
        
         };

     

       83
       83
        
       

     

       84
       84
       -
          cutlass = fetchFromGitHub {

     

       84
       84
       +
         cutlass = fetchFromGitHub {

     

       85
       85
        
           owner = "NVIDIA";

     

       86
       86
        
           repo = "cutlass";

     

       87
       87
        
           tag = "v3.5.1";

     
···

       95
       95
        
           hash = "sha256-YJdZ0cMtUncH5Z6TtAWBH0xtAIu2UcbjnVcCM4tfg20=";

     

       96
       96
        
         };

     

       97
       97
        
       

     

       98
       98
       +
         isCudaJetson = cudaSupport && cudaPackages.cudaFlags.isJetsonBuild;

     

       98
       99
        
       in

     

       99
       99
       -
       

     

       100
       100
        
       effectiveStdenv.mkDerivation rec {

     

       101
       101
        
         pname = "onnxruntime";

     

       102
       102
        
         inherit src version;

     
···

       128
       128
        
         ]) ++ lib.optionals cudaSupport [

     

       129
       129
        
           cudaPackages.cuda_nvcc

     

       130
       130
        
           cudaPackages.cudnn-frontend

     

       131
       131
       +
         ] ++ lib.optionals isCudaJetson [

     

       132
       132
       +
           cudaPackages.autoAddCudaCompatRunpath

     

       131
       133
        
         ];

     

       132
       134
        
       

     

       133
       135
        
         buildInputs = [