Unverified Commit b5add006 authored by Stefan Frijters's avatar Stefan Frijters
Browse files

python3Packages.bitsandbytes: move env vars into env for structuredAttrs

parent c6147e10
Loading
Loading
Loading
Loading
+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