Skip to content

Commit ce24f3e

Browse files
force 1 thread is eddy run on gpu
1 parent 2f9aa4b commit ce24f3e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Diff for: nipype/interfaces/fsl/epi.py

+3
Original file line numberDiff line numberDiff line change
@@ -960,6 +960,8 @@ def __init__(self, **inputs):
960960
self._use_cuda()
961961

962962
def _num_threads_update(self):
963+
if self.inputs.use_cuda and gpu_count()>0:
964+
self.inputs.num_threads = 1
963965
self._num_threads = self.inputs.num_threads
964966
if not isdefined(self.inputs.num_threads):
965967
if "OMP_NUM_THREADS" in self.inputs.environ:
@@ -969,6 +971,7 @@ def _num_threads_update(self):
969971

970972
def _use_cuda(self):
971973
if self.inputs.use_cuda and gpu_count()>0:
974+
self.inputs.num_threads = 1
972975
# eddy_cuda usually link to eddy_cudaX.X but some versions miss the symlink
973976
# anyway in newer fsl versions eddy automatically use cuda on cuda-capable systems
974977
self._cmd = "eddy_cuda" if which("eddy_cuda") else "eddy"

0 commit comments

Comments
 (0)