dev/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Merge pull request #136065 from yl3dy/torch_blas_passthru

Browse Source 
python3Packages.pytorch: added BLAS provider passthru
This commit is contained in:
Sandro 2021-09-11 22:11:21 +02:00 committed by GitHub
commit 9a58f7ea30
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats 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 = { passthru = {
inherit cudaSupport; inherit cudaSupport;
cudaArchList = final_cudaArchList; 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; { meta = with lib; {