Commit ce24f3e

committed

force 1 thread is eddy run on gpu

1 parent 2f9aa4b commit ce24f3eCopy full SHA for ce24f3e

File tree

-0

lines changed

-0

lines changed

Lines changed: 3 additions & 0 deletions

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"`

Comments

(0)