Commit bc355549 authored by Martin Reinecke's avatar Martin Reinecke

fix FFTW benchmark; pypocketfft now looks much better :)

parent c0f74f61
......@@ -45,7 +45,7 @@ def measure_fftw(a, nrepeat):
f1 = pyfftw.empty_aligned(a.shape, dtype=a.dtype)
f1[()] = a
f2 = pyfftw.empty_aligned(a.shape, dtype=a.dtype)
fftw = pyfftw.FFTW(f1, f2, flags=('FFTW_MEASURE',), threads=nthr)
fftw = pyfftw.FFTW(f1, f2, flags=('FFTW_MEASURE',), axes=range(a.ndim), threads=nthr)
tmin = 1e38
for i in range(nrepeat):
t0 = time()
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment