From 57c78bcf04cae2af4b3aeb26d2b39b3ba428b77d Mon Sep 17 00:00:00 2001
From: samaid <55950596+samaid@users.noreply.github.com>
Date: Tue, 28 Feb 2023 10:26:01 -0600
Subject: [PATCH 1/2] Enable MKL in SciPy

Enable MKL in SciPy in case of presence of mkl_fft package in the environment
---
 fft_bench.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/fft_bench.py b/fft_bench.py
index c2d932b..c3ad151 100644
--- a/fft_bench.py
+++ b/fft_bench.py
@@ -12,6 +12,13 @@
 import re
 import sys
 
+try: # Workaround for disabled mkl_fft in latest Intel SciPy
+    import mkl_fft
+    import mkl_fft._scipy_fft_backend as sfbn
+    scipy.fft.set_global_backend(sfbn)
+except:
+    print("Unable to use mkl_fft as scipy.fft backend")
+
 # Mark which FFT submodules are available...
 fft_modules = {'numpy.fft': np.fft, 'scipy.fft': scipy.fft}
 

From a744f802a53cfa60a66ad84da57abaebb959b222 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <oleksandr.pavlyk@intel.com>
Date: Tue, 28 Feb 2023 14:08:25 -0600
Subject: [PATCH 2/2] Update fft_bench.py

Provide list of exceptions that triggers customer warning
---
 fft_bench.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fft_bench.py b/fft_bench.py
index c3ad151..9211c26 100644
--- a/fft_bench.py
+++ b/fft_bench.py
@@ -16,7 +16,7 @@
     import mkl_fft
     import mkl_fft._scipy_fft_backend as sfbn
     scipy.fft.set_global_backend(sfbn)
-except:
+except (ImportError, ModuleNotFoundError, ValueError):
     print("Unable to use mkl_fft as scipy.fft backend")
 
 # Mark which FFT submodules are available...