File size: 1,043 Bytes
9c19cff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
#!/bin/bash
ARCHS=( arm32 aarch64 x64 )
DTYPES=( float double )
for arch in ${ARCHS[@]}
do
for dtype in ${DTYPES[@]}
do
DIR=compiled/${arch}/${dtype}
mkdir -p $DIR
echo "### Running ${arch} ${dtype} test ###"
python3 -m remend.compile_dataset --file ./prim_fwd/prim_fwd.test --prefix test \
--impl cse_c dag_c cse_fortran dag_fortran -O 0 1 2 --pick 1 --outdir $DIR \
--arch ${arch} --dtype ${dtype}
echo "### Running ${arch} ${dtype} valid ###"
python3 -m remend.compile_dataset --file ./prim_fwd/prim_fwd.valid --prefix valid \
--impl cse_c dag_c cse_fortran dag_fortran -O 0 1 2 --pick 1 --outdir $DIR \
--arch ${arch} --dtype ${dtype}
echo "### Running ${arch} ${dtype} train ###"
python3 -m remend.compile_dataset --file ./prim_fwd/prim_fwd.train --prefix train \
--impl cse_c dag_c cse_fortran dag_fortran -O 0 1 2 --pick 0.0005 --outdir $DIR \
--arch ${arch} --dtype ${dtype}
done
done
|