File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments