Skip to content

Commit

Permalink
ENH: Provide free memory estimate to unwarp_wf for better resources a…
Browse files Browse the repository at this point in the history
…llocation (#2910)

Takes advantage of nipreps/sdcflows#321 by providing an estimate of the free memory on the system to better select the number of threads to request.
  • Loading branch information
effigies authored Dec 12, 2022
2 parents c6a9671 + ad5b045 commit c59814f
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions fmriprep/workflows/bold/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -1112,6 +1112,7 @@ def init_func_preproc_wf(bold_file, has_fieldmap=False):
write_coeff=True,
)
unwarp_wf = init_unwarp_wf(
free_mem=config.environment.free_mem,
debug="fieldmaps" in config.execution.debug,
omp_nthreads=config.nipype.omp_nthreads,
)
Expand Down
2 changes: 1 addition & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ install_requires =
psutil >= 5.4
pybids >= 0.15.0
requests
sdcflows ~= 2.2.0
sdcflows ~= 2.2.1
smriprep ~= 0.10.0
tedana ~= 0.0.9
templateflow >= 0.6
Expand Down

0 comments on commit c59814f

Please sign in to comment.