flatironinstitute / jax-finufft

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
Apache License 2.0
77 stars 2 forks source link

Add check for OpenMP compile state to options #82

Closed dfm closed 4 months ago

dfm commented 4 months ago

This temporarily fixes #73 until https://github.com/flatironinstitute/finufft/pull/431 gets merged and pulled into our git submodule. It works by forcing nthreads=1 if FINUFFT was compiled without OpenMP support. It's a bit of a hack and https://github.com/flatironinstitute/finufft/pull/431 will be better, but this could be useful in the meantime!