From c5df984674f278e4b25269a0c337529e6ce97bbf Mon Sep 17 00:00:00 2001
From: Isuru Fernando <idf2@illinois.edu>
Date: Tue, 22 Feb 2022 01:34:01 -0600
Subject: [PATCH] update for new pymbolic fft
---
sumpy/tools.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/sumpy/tools.py b/sumpy/tools.py
index d0781bbb..669a3d63 100644
--- a/sumpy/tools.py
+++ b/sumpy/tools.py
@@ -870,16 +870,18 @@ def fft(seq, inverse=False, sac=None):
from pymbolic.algorithm import fft as _fft, ifft as _ifft
- def wrap(expr):
+ def wrap(level, expr):
if isinstance(expr, np.ndarray):
- res = [wrap(a) for a in expr]
+ res = [wrap(level, a) for a in expr]
return np.array(res, dtype=object).reshape(expr.shape)
return add_to_sac(sac, expr)
if inverse:
- return _ifft(list(seq), wrap_intermediate=wrap).tolist()
+ return _ifft(np.array(seq), wrap_intermediate_with_level=wrap,
+ complex_dtype=np.complex128).tolist()
else:
- return _fft(list(seq), wrap_intermediate=wrap).tolist()
+ return _fft(np.array(seq), wrap_intermediate_with_level=wrap,
+ complex_dtype=np.complex128).tolist()
def fft_toeplitz_upper_triangular(first_row, x, sac=None):
--
GitLab