Finufft3d fails with Incompatible function arguments for single precision setting (runtime.use_single_prec=true)
Error in streaming mode on cori gpu when various GPU related flags are switched off
[runtime]
use_cuda = true
use_cufinufft = false
use_cupy = false
Traceback (most recent call last):
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/spinifel/__main__.py", line 25, in <module>
main_psana2()
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/site-packages/PyNVTX/__init__.py", line 33, in wrapper
ret = func(*args, **kwargs)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/spinifel/mpi/main_psana2.py", line 361, in main
ac = mg.solve_ac(generation)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/site-packages/PyNVTX/__init__.py", line 33, in wrapper
ret = func(*args, **kwargs)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/spinifel/mpi/autocorrelation.py", line 165, in solve_ac
W, d = self.setup_linops(H, K, L, ac_support, x0)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/site-packages/PyNVTX/__init__.py", line 33, in wrapper
ret = func(*args, **kwargs)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/spinifel/mpi/autocorrelation.py", line 93, in setup_linops
ugrid_conv = self.nufft.adjoint(
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/site-packages/PyNVTX/__init__.py", line 33, in wrapper
ret = func(*args, **kwargs)
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/spinifel/extern/nufft_ext.py", line 490, in adjoint
assert not finufft.nufft3d1(
File "/global/project/projectdirs/lcls/chuck/temp/spinifel/setup/conda/envs/myenv/lib/python3.8/site-packages/finufftpy/_interfaces.py", line 457, in nufft3d1
return finufftpy_cpp.finufft3d1_cpp(x,y,z,c,isign,eps,ms,mt,mu,f,debug,spread_debug,spread_sort,fftw,modeord,chkbnds,upsampfac)
TypeError: finufft3d1_cpp(): incompatible function arguments. The following argument types are supported:
1. (xj: numpy.ndarray[numpy.float64], yj: numpy.ndarray[numpy.float64], zj: numpy.ndarray[numpy.float64], cj: numpy.ndarray[numpy.complex128], iflag: int, eps: float, ms: int, mt: int, mu: int, fk: numpy.ndarray[numpy.complex128], debug: int, spread_debug: int, spread_sort: int, fftw: int, modeord: int, chkbnds: int, upsampfac: float) -> int
Invoked with: array([-2.48019188, -2.47531443, -2.4704406 , ..., 2.66082175,
2.69253805, 2.72423416]), array([-0.74826405, -0.72593415, -0.70365277, ..., -1.26888202,
-1.25659757, -1.24433789]), array([-1.77717593, -1.75186601, -1.72646421, ..., -0.9427877 ,
-0.94572608, -0.94856553]), array([1.+0.j, 1.+0.j, 1.+0.j, ..., 1.+0.j, 1.+0.j, 1.+0.j]), -1, 1e-12, 162, 162, 162, array([[[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
...,
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j]],
[[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
...,
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j]],
[[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
...,
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j]],
...,
[[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
...,
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j]],
[[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
...,
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j]],
[[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
...,
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j],
[0.+0.j, 0.+0.j, 0.+0.j, ..., 0.+0.j, 0.+0.j, 0.+0.j]]],
dtype=complex64), 0, 0, 2, 0, 0, 1, 2.0