Update index url and check for macos

Author: rohanbabbar04

pylops/signalprocessing/fft.py

@@ -22,7 +22,7 @@
     import pyfftw
 
 if mkl_fft_message is None:
-    import mkl_fft.interfaces.scipy_fft as mkl_backend
+    import mkl_fft.interfaces.numpy_fft as mkl_backend
 
 logger = logging.getLogger(__name__)
 
@@ -414,21 +414,6 @@ def __init__(
         dtype: DTypeLike = "complex128",
         **kwargs_fft,
     ) -> None:
-        if np.dtype(dtype) == np.float16:
-            warnings.warn(
-                "mkl_fft backend is unavailable with float16 dtype. Will use float32."
-            )
-            dtype = np.float32
-        elif np.dtype(dtype) == np.complex256:
-            warnings.warn(
-                "mkl_fft backend is unavailable with complex256 dtype. Will use complex128."
-            )
-            dtype = np.complex128
-        elif np.dtype(dtype) == np.float128:
-            warnings.warn(
-                "mkl_fft backend is unavailable with float128 dtype. Will use float64."
-            )
-            dtype = np.float64
         super().__init__(
             dims=dims,
             axis=axis,
@@ -452,12 +437,6 @@ def __init__(
 
     @reshaped
     def _matvec(self, x: NDArray) -> NDArray:
-        if x.dtype == np.float16:
-            x = x.astype(np.float32)
-        elif x.dtype == np.float128:
-            x = x.astype(np.float64)
-        elif x.dtype == np.complex256:
-            x = x.astype(np.complex128)
         if self.ifftshift_before:
             x = mkl_backend.ifftshift(x, axes=self.axis)
         if not self.clinear:
@@ -477,12 +456,6 @@ def _matvec(self, x: NDArray) -> NDArray:
 
     @reshaped
     def _rmatvec(self, x: NDArray) -> NDArray:
-        if x.dtype == np.float16:
-            x = x.astype(np.float32)
-        elif x.dtype == np.float128:
-            x = x.astype(np.float64)
-        elif x.dtype == np.complex256:
-            x = x.astype(np.complex128)
         if self.fftshift_after:
             x = mkl_backend.ifftshift(x, axes=self.axis)
         if self.real:

pylops/signalprocessing/fft2d.py

@@ -16,7 +16,7 @@
 mkl_fft_message = deps.mkl_fft_import("the mkl fft module")
 
 if mkl_fft_message is None:
-    import mkl_fft.interfaces.scipy_fft as mkl_backend
+    import mkl_fft.interfaces.numpy_fft as mkl_backend
 
 
 class _FFT2D_numpy(_BaseFFTND):
@@ -257,21 +257,6 @@ def __init__(
         dtype: DTypeLike = "complex128",
         **kwargs_fft,
     ) -> None:
-        if np.dtype(dtype) == np.float16:
-            warnings.warn(
-                "mkl_fft backend is unavailable with float16 dtype. Will use float32."
-            )
-            dtype = np.float32
-        elif np.dtype(dtype) == np.complex256:
-            warnings.warn(
-                "mkl_fft backend is unavailable with complex256 dtype. Will use complex128."
-            )
-            dtype = np.complex128
-        elif np.dtype(dtype) == np.float128:
-            warnings.warn(
-                "mkl_fft backend is unavailable with float128 dtype. Will use float64."
-            )
-            dtype = np.float64
         super().__init__(
             dims=dims,
             axes=axes,
@@ -305,12 +290,6 @@ def __init__(
 
     @reshaped
     def _matvec(self, x):
-        if x.dtype == np.float16:
-            x = x.astype(np.float32)
-        elif x.dtype == np.float128:
-            x = x.astype(np.float64)
-        elif x.dtype == np.complex256:
-            x = x.astype(np.complex128)
         if self.ifftshift_before.any():
             x = mkl_backend.ifftshift(x, axes=self.axes[self.ifftshift_before])
         if not self.clinear:
@@ -335,12 +314,6 @@ def _matvec(self, x):
 
     @reshaped
     def _rmatvec(self, x):
-        if x.dtype == np.float16:
-            x = x.astype(np.float32)
-        elif x.dtype == np.float128:
-            x = x.astype(np.float64)
-        elif x.dtype == np.complex256:
-            x = x.astype(np.complex128)
         if self.fftshift_after.any():
             x = mkl_backend.ifftshift(x, axes=self.axes[self.fftshift_after])
         if self.real:

pylops/signalprocessing/fftnd.py

@@ -16,7 +16,7 @@
 mkl_fft_message = deps.mkl_fft_import("the mkl fft module")
 
 if mkl_fft_message is None:
-    import mkl_fft.interfaces.scipy_fft as mkl_backend
+    import mkl_fft.interfaces.numpy_fft as mkl_backend
 
 
 class _FFTND_numpy(_BaseFFTND):
@@ -238,21 +238,6 @@ def __init__(
         dtype: DTypeLike = "complex128",
         **kwargs_fft,
     ) -> None:
-        if np.dtype(dtype) == np.float16:
-            warnings.warn(
-                "mkl_fft backend is unavailable with float16 dtype. Will use float32."
-            )
-            dtype = np.float32
-        elif np.dtype(dtype) == np.complex256:
-            warnings.warn(
-                "mkl_fft backend is unavailable with complex256 dtype. Will use complex128."
-            )
-            dtype = np.complex128
-        elif np.dtype(dtype) == np.float128:
-            warnings.warn(
-                "mkl_fft backend is unavailable with float128 dtype. Will use float64."
-            )
-            dtype = np.float64
         super().__init__(
             dims=dims,
             axes=axes,
@@ -275,12 +260,6 @@ def __init__(
 
     @reshaped
     def _matvec(self, x: NDArray) -> NDArray:
-        if x.dtype == np.float16:
-            x = x.astype(np.float32)
-        elif x.dtype == np.float128:
-            x = x.astype(np.float64)
-        elif x.dtype == np.complex256:
-            x = x.astype(np.complex128)
         if self.ifftshift_before.any():
             x = mkl_backend.ifftshift(x, axes=self.axes[self.ifftshift_before])
         if not self.clinear:
@@ -305,12 +284,6 @@ def _matvec(self, x: NDArray) -> NDArray:
 
     @reshaped
     def _rmatvec(self, x: NDArray) -> NDArray:
-        if x.dtype == np.float16:
-            x = x.astype(np.float32)
-        elif x.dtype == np.float128:
-            x = x.astype(np.float64)
-        elif x.dtype == np.complex256:
-            x = x.astype(np.complex128)
         if self.fftshift_after.any():
             x = mkl_backend.ifftshift(x, axes=self.axes[self.fftshift_after])
         if self.real:

requirements-dev.txt

@@ -1,3 +1,6 @@
+--index-url https://software.repos.intel.com/python/pypi
+--extra-index-url https://pypi.org/simple
+mkl_fft
 numpy>=2.0.0
 scipy>=1.13.0
 jax
@@ -30,4 +33,3 @@ black
 flake8
 mypy
 pytensor>=2.28.0
-mkl-fft; sys_platform != "darwin"

requirements-doc.txt

@@ -1,3 +1,6 @@
+--index-url https://software.repos.intel.com/python/pypi
+--extra-index-url https://pypi.org/simple
+mkl_fft
 numpy>=2.0.0
 scipy>=1.13.0
 jax
@@ -31,4 +34,3 @@ flake8
 mypy
 pytensor>=2.28.0
 pymc>=5.21.0
-mkl-fft; sys_platform != "darwin"