#!/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