Merge pull request #232457 from wegank/cudatoolkit-eval-1

python310Packages.torch: restrict platforms
This commit is contained in:
Weijia Wang 2023-05-17 21:53:41 +03:00 committed by GitHub
commit d1bc3181fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -414,7 +414,7 @@ in buildPythonPackage rec {
homepage = "https://pytorch.org/";
license = licenses.bsd3;
maintainers = with maintainers; [ teh thoughtpolice tscholak ]; # tscholak esp. for darwin-related builds
platforms = with platforms; linux ++ lib.optionals (!cudaSupport || !rocmSupport) darwin;
platforms = with platforms; linux ++ lib.optionals (!cudaSupport && !rocmSupport) darwin;
broken = rocmSupport && cudaSupport; # CUDA and ROCm are mutually exclusive
};
}