Merge pull request #2081 from FCP-INDI/also-throttle

⚡️ Throttle `calc_ants_warp`, `cal_DVARS` and all `cosine_filter`s
FCP-INDI · Mar 13, 2024 · 41bba7d · 41bba7d
2 parents 955155e + a469562
commit 41bba7d
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/CPAC/generate_motion_statistics/generate_motion_statistics.py b/CPAC/generate_motion_statistics/generate_motion_statistics.py
@@ -207,7 +207,8 @@ def motion_power_statistics(name='motion_stats',
                         name='cal_DVARS',
                         mem_gb=0.4,
                         mem_x=(739971956005215 / 151115727451828646838272,
-                               'in_file'))
+                               'in_file'),
+                        throttle=True)
 
     cal_DVARS_strip = pe.Node(Function(input_names=['file_1D'],
                                        output_names=['out_file', 'DVARS_val'],

diff --git a/CPAC/nuisance/nuisance.py b/CPAC/nuisance/nuisance.py
@@ -1283,7 +1283,7 @@ def create_regressor_workflow(nuisance_selectors,
                                         imports=cosfilter_imports),
                             name=f'{regressor_type}_cosine_filter',
                             mem_gb=8.0,
-                            throttle=regressor_type.lower() == "acompcor")
+                            throttle=True)
                         nuisance_wf.connect(
                             summary_filter_input[0], summary_filter_input[1],
                             cosfilter_node, 'input_image_path'

diff --git a/CPAC/qc/xcp.py b/CPAC/qc/xcp.py
@@ -112,7 +112,8 @@ def _connect_motion(wf, nodes, strat_pool, qc_file, pipe_num):
                         name=f'cal_DVARS_{pipe_num}',
                         mem_gb=0.4,
                         mem_x=(739971956005215 / 151115727451828646838272,
-                               'in_file'))
+                               'in_file'),
+                        throttle=True)
     cal_DVARS_strip = pe.Node(Function(input_names=['file_1D'],
                                        output_names=['out_file'],
                                        function=DVARS_strip_t0,

diff --git a/CPAC/registration/registration.py b/CPAC/registration/registration.py
@@ -1209,7 +1209,8 @@ def create_wf_calculate_ants_warp(
                                         imports=reg_imports),
                 name='calc_ants_warp',
                 mem_gb=2.8,
-                mem_x=(2e-7, 'moving_brain', 'xyz'))
+                mem_x=(2e-7, 'moving_brain', 'xyz'),
+                throttle=True)
 
     calculate_ants_warp.interface.num_threads = num_threads