python3Packages.flashinfer: move env vars into env for structuredAttrs

This commit is contained in:
Stefan Frijters
2026-02-28 13:38:13 +01:00
parent 236c631156
commit 7fc539e635
@@ -87,7 +87,7 @@ buildPythonPackage rec {
export MAX_JOBS="$NIX_BUILD_CORES"
'';
FLASHINFER_CUDA_ARCH_LIST = lib.concatStringsSep ";" torch.cudaCapabilities;
env.FLASHINFER_CUDA_ARCH_LIST = lib.concatStringsSep ";" torch.cudaCapabilities;
pythonRemoveDeps = [
"nvidia-cudnn-frontend"