...
 
Commits (2)
......@@ -52,14 +52,14 @@ variable_verbosity = keepers.Variable('verbosity',
nifty_configuration = keepers.get_Configuration(
'NIFTy',
[variable_fft_module,
variable_lm2gl,
variable_use_healpy,
variable_use_libsharp,
variable_verbosity,
],
path=os.path.expanduser('~') + "/.nifty/nifty_config")
name='NIFTy',
variables=[variable_fft_module,
variable_lm2gl,
variable_use_healpy,
variable_use_libsharp,
variable_verbosity],
file_name='nifty_config')
########
### Compatibility variables
########
......
......@@ -381,6 +381,11 @@ class _fftw_plan_and_info(object):
self.local_node_dimensions,
self.offsetQ))
try:
threads = int(os.environ['NIFTY_FFTW_THREADS'])
except(KeyError):
threads = 1
self.set_plan(
pyfftw.create_mpi_plan(
input_shape=self.global_input_shape,
......@@ -388,7 +393,7 @@ class _fftw_plan_and_info(object):
output_dtype=self.output_dtype,
direction=self.direction,
flags=["FFTW_ESTIMATE"],
threads=int(os.environ['MY_FFTW_THREADS']),
threads=threads,
**kwargs)
)
......