FIX: Tag T2SMap with memory requirements for loading two copies of al…

…l echos
nipreps · Dec 2, 2022 · 0bc23f5 · 0bc23f5
1 parent d500199
commit 0bc23f5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/fmriprep/workflows/bold/base.py b/fmriprep/workflows/bold/base.py
@@ -564,7 +564,7 @@ def init_func_preproc_wf(bold_file, has_fieldmap=False):
         # create optimal combination, adaptive T2* map
         bold_t2s_wf = init_bold_t2s_wf(
             echo_times=tes,
-            mem_gb=mem_gb["resampled"],
+            mem_gb=mem_gb["filesize"],
             omp_nthreads=omp_nthreads,
             name="bold_t2smap_wf",
         )

diff --git a/fmriprep/workflows/bold/t2s.py b/fmriprep/workflows/bold/t2s.py
@@ -98,7 +98,11 @@ def init_bold_t2s_wf(echo_times, mem_gb, omp_nthreads, name='bold_t2s_wf'):
 
     LOGGER.log(25, 'Generating T2* map and optimally combined ME-EPI time series.')
 
-    t2smap_node = pe.Node(T2SMap(echo_times=list(echo_times)), name='t2smap_node')
+    t2smap_node = pe.Node(
+        T2SMap(echo_times=list(echo_times)),
+        name='t2smap_node',
+        mem_gb=2 * mem_gb * len(echo_times),
+    )
     # fmt:off
     workflow.connect([
         (inputnode, t2smap_node, [('bold_file', 'in_files'),