Commit 97579c6e authored by theos's avatar theos
Browse files

Made environmental variable NIFTY_FFTW_THREADS optional.

parent 7bcad43a
...@@ -381,6 +381,11 @@ class _fftw_plan_and_info(object): ...@@ -381,6 +381,11 @@ class _fftw_plan_and_info(object):
self.local_node_dimensions, self.local_node_dimensions,
self.offsetQ)) self.offsetQ))
try:
threads = int(os.environ['NIFTY_FFTW_THREADS'])
except(KeyError):
threads = 1
self.set_plan( self.set_plan(
pyfftw.create_mpi_plan( pyfftw.create_mpi_plan(
input_shape=self.global_input_shape, input_shape=self.global_input_shape,
...@@ -388,7 +393,7 @@ class _fftw_plan_and_info(object): ...@@ -388,7 +393,7 @@ class _fftw_plan_and_info(object):
output_dtype=self.output_dtype, output_dtype=self.output_dtype,
direction=self.direction, direction=self.direction,
flags=["FFTW_ESTIMATE"], flags=["FFTW_ESTIMATE"],
threads=int(os.environ['MY_FFTW_THREADS']), threads=threads,
**kwargs) **kwargs)
) )
......
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