Unverified Commit dbd5b7af authored by j-k's avatar j-k Committed by GitHub
Browse files

python3Packages.bitsandbytes: move env vars into env for structuredAttrs (#492268)

parents db897912 b5add006

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

+8 −5
Original line number Diff line number Diff line
@@ -163,11 +163,14 @@ buildPythonPackage { 


    (lib.cmakeFeature "CMAKE_HIP_ARCHITECTURES" (builtins.concatStringsSep ";" rocmGpuTargets))

  ];

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

  NVCC_PREPEND_FLAGS = lib.optionals cudaSupport [



  env = lib.optionalAttrs cudaSupport {

    CUDA_HOME = cuda-native-redist;

    NVCC_PREPEND_FLAGS = toString [

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

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

    ];

  };



  preBuild = ''

    make -j $NIX_BUILD_CORES