diff options
| author | Sandro <sandro.jaeckel@gmail.com> | 2021-09-11 22:11:21 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-09-11 22:11:21 +0200 |
| commit | 9a58f7ea301a4f047b7a27e8b2319d422050dd03 (patch) | |
| tree | c1bf264bd4771ee9a637bb1377ba2db3d33c99a1 /pkgs/development | |
| parent | 956e3411019a84fe1908311325ccdca48c694f8d (diff) | |
| parent | 20a1ea5de7f484709935f58da8f6682ccd77c5a7 (diff) | |
Merge pull request #136065 from yl3dy/torch_blas_passthru
python3Packages.pytorch: added BLAS provider passthru
Diffstat (limited to 'pkgs/development')
| -rw-r--r-- | pkgs/development/python-modules/pytorch/default.nix | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix index 0de0015ab1ec7..a69268cd5aecd 100644 --- a/pkgs/development/python-modules/pytorch/default.nix +++ b/pkgs/development/python-modules/pytorch/default.nix @@ -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; { |