python3Packages.pytorch: added BLAS provider passthru

`torch.fft` is available only when BLAS provider is MKL. Passing it thru allows dependent derivations to easily check for FFT availability.
2021-08-29 00:33:01 +03:00
parent b9e35f1c86
commit 20a1ea5de7
1 changed files with 2 additions and 0 deletions
@@ -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; {