From 20a1ea5de7f484709935f58da8f6682ccd77c5a7 Mon Sep 17 00:00:00 2001
From: Alexander Kiselyov <aleksandr.kiselyov@gmail.com>
Date: Sun, 29 Aug 2021 00:33:01 +0300
Subject: [PATCH] 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.
---
 pkgs/development/python-modules/pytorch/default.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix
index 0de0015ab1ec..a69268cd5aec 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; {