movefasta/nixpkgs
0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-07-13 21:50:33 +03:00
Code Activity

python3Packages.pytorch: added BLAS provider passthru

Browse source 
`torch.fft` is available only when BLAS provider is MKL. Passing it thru
allows dependent derivations to easily check for FFT availability.
This commit is contained in:
Alexander Kiselyov 2021-08-29 00:33:01 +03:00
parent b9e35f1c86
commit 20a1ea5de7
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pkgs/development/python-modules/pytorch/default.nix
View file

@ -304,6 +304,8 @@ in buildPythonPackage rec {
passthru = {
inherit cudaSupport;
cudaArchList = final_cudaArchList;
# At least for 1.9.0 `torch.fft` is unavailable unless BLAS provider is MKL. This attribute allows for easy detection of its availability.
blasProvider = blas.provider;
};
meta = with lib; {