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):
960
960
self ._use_cuda ()
961
961
962
962
def _num_threads_update (self ):
963
+ if self .inputs .use_cuda and gpu_count ()> 0 :
964
+ self .inputs .num_threads = 1
963
965
self ._num_threads = self .inputs .num_threads
964
966
if not isdefined (self .inputs .num_threads ):
965
967
if "OMP_NUM_THREADS" in self .inputs .environ :
@@ -969,6 +971,7 @@ def _num_threads_update(self):
969
971
970
972
def _use_cuda (self ):
971
973
if self .inputs .use_cuda and gpu_count ()> 0 :
974
+ self .inputs .num_threads = 1
972
975
# eddy_cuda usually link to eddy_cudaX.X but some versions miss the symlink
973
976
# anyway in newer fsl versions eddy automatically use cuda on cuda-capable systems
974
977
self ._cmd = "eddy_cuda" if which ("eddy_cuda" ) else "eddy"
You can’t perform that action at this time.
0 commit comments