commit e555375a11fa177252a4e7fccedce46caf93f923 · pyrox.dev/nixpkgs

+3 -1

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

···

       83
       83
        
       

     

       84
       84
        
         nativeBuildInputs = [

     

       85
       85
        
           cmake

     

       86
       86
       +
         ]

     

       87
       87
       +
         ++ lib.optionals cudaSupport [

     

       86
       88
        
           cudaPackages.cuda_nvcc

     

       87
       89
        
         ];

     

       88
       90
        
       

     
···

       96
       98
        
         cmakeFlags = [

     

       97
       99
        
           (lib.cmakeFeature "COMPUTE_BACKEND" (if cudaSupport then "cuda" else "cpu"))

     

       98
       100
        
         ];

     

       99
       99
       -
         CUDA_HOME = "${cuda-native-redist}";

     

       101
       101
       +
         CUDA_HOME = lib.optionalString cudaSupport "${cuda-native-redist}";

     

       100
       102
        
         NVCC_PREPEND_FLAGS = lib.optionals cudaSupport [

     

       101
       103
        
           "-I${cuda-native-redist}/include"

     

       102
       104
        
           "-L${cuda-native-redist}/lib"