Skip to content

Commit a526c4e

Browse files
committed
Simplified type 3. Fixed 1D and 2D _invoke_plan input for type 3
1 parent eeb9d6a commit a526c4e

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

python/cufinufft/cufinufft/_simple.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,17 +22,14 @@ def nufft3d1(x, y, z, data, n_modes=None, out=None, eps=1e-6, isign=1,
2222
def nufft3d2(x, y, z, data, out=None, eps=1e-6, isign=-1, **kwargs):
2323
return _invoke_plan(3, 2, x, y, z, data, None, None, None, out, isign, eps, None, kwargs)
2424

25-
# Then nufft3d3 is as simple as:
2625
def nufft3d3(x, y, z, data, s, t, u, out=None, eps=1e-6, isign=1, **kwargs):
2726
return _invoke_plan(3, 3, x, y, z, data, s, t, u, out, isign, eps, None, kwargs)
2827

29-
# For 2d3:
3028
def nufft2d3(x, y, data, s, t, out=None, eps=1e-6, isign=1, **kwargs):
31-
return _invoke_plan(2, 3, x, y, None, data, s, t, out, isign, eps, None, kwargs)
29+
return _invoke_plan(2, 3, x, y, None, data, s, t, None, out, isign, eps, None, kwargs)
3230

33-
# For 1d3:
3431
def nufft1d3(x, data, s, out=None, eps=1e-6, isign=1, **kwargs):
35-
return _invoke_plan(1, 3, x, None, None, data, s, out, isign, eps, None, kwargs)
32+
return _invoke_plan(1, 3, x, None, None, data, s, None, None, out, isign, eps, None, kwargs)
3633

3734
def _invoke_plan(dim, nufft_type, x, y, z, data, s, t, u, out, isign, eps,
3835
n_modes=None, kwargs=None):

0 commit comments

Comments
 (0)