Skip to content
Snippets Groups Projects

Bf/startup

Merged Matthew Webster requested to merge bf/startup into master
1 file
+ 14
1
Compare changes
  • Side-by-side
  • Inline
@@ -4,10 +4,13 @@
import os
import shlex
import sys
import tempfile
import subprocess as sp
from pathlib import Path
def run(cmd, ompthreads=None, blasthreads=None, fslskipglobal=None):
env = os.environ.copy()
@@ -33,12 +36,22 @@ def run(cmd, ompthreads=None, blasthreads=None, fslskipglobal=None):
def main():
env = os.environ.copy()
activatePath = Path(f'''{env['FSLDIR']}/bin/activate''')
if not activatePath.is_file():
activatePath='activate'
buildcmds = ['source activate $FSLDIR',
outdir = Path(sys.argv[1])
buildcmds = [f'''source {activatePath} $FSLDIR''',
'source $FSLDIR/etc/fslconf/fsl-devel.sh',
f'''cp Makefile test_fslStartup.cc {outdir}''',
f'''cd {outdir}''',
'make']
sp.run('; '.join(buildcmds), check=True, shell=True)
os.chdir(outdir)
# Default behaviour should be: OMP multi-threaded, BLAS single threaded.
assert run('./test_fslStartup', 8, 8) == '8 1 8'
assert run('./test_fslStartup', 4, 4) == '4 1 4'
Loading