Merge pull request #208485 from wegank/torch-darwin

python3Packages.torch: fix build on darwin
This commit is contained in:
Robert Scott 2023-01-03 23:49:06 +00:00 committed by GitHub
commit 2057adf46e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -167,7 +167,7 @@ in buildPythonPackage rec {
] ++ lib.optionals cudaSupport [ cudatoolkit_joined ];
buildInputs = [ blas blas.provider pybind11 ]
++ [ linuxHeaders_5_19 ] # TMP: avoid "flexible array member" errors for now
++ lib.optionals stdenv.isLinux [ linuxHeaders_5_19 ] # TMP: avoid "flexible array member" errors for now
++ lib.optionals cudaSupport [ cudnn magma nccl ]
++ lib.optionals stdenv.isLinux [ numactl ]
++ lib.optionals stdenv.isDarwin [ CoreServices libobjc ];