102 lines
3.0 KiB
Bash
Executable File
102 lines
3.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# exit script if any command fails
|
|
set -e
|
|
set -o pipefail
|
|
|
|
# On macOS, use GNU readlink from 'coreutils' package in Homebrew/MacPorts
|
|
if [ "$(uname -s)" = "Darwin" ] ; then
|
|
READLINK=greadlink
|
|
else
|
|
READLINK=readlink
|
|
fi
|
|
|
|
# If BASH_SOURCE is undefined, we may be running under zsh, in that case
|
|
# provide a zsh-compatible alternative
|
|
DIR="$(dirname "$($READLINK -f "${BASH_SOURCE[0]:-${(%):-%x}}")")"
|
|
|
|
usage() {
|
|
echo "Usage: ${0} [OPTIONS] [riscv-tools | esp-tools]"
|
|
echo ""
|
|
echo "Helper script to initialize repository that wraps other scripts."
|
|
echo "Sets up conda environment, initializes submodules, and installs toolchain collateral."
|
|
echo ""
|
|
echo "Installation Types"
|
|
echo " riscv-tools: if set, builds the riscv toolchain (this is also the default)"
|
|
echo " esp-tools: if set, builds esp-tools toolchain used for the hwacha vector accelerator"
|
|
echo ""
|
|
echo "Options"
|
|
echo " --help -h : Display this message"
|
|
echo " --unpinned-deps -ud : Use unpinned conda environment"
|
|
echo " --batch : Skip prompt checking for tagged release/conda"
|
|
echo " --skip-conda : Skip conda env creation"
|
|
echo " --skip-toolchain-extra : Skip building extra RISC-V toolchain collateral (Spike, PK, tests, libgloos)"
|
|
exit "$1"
|
|
}
|
|
|
|
TOOLCHAIN="riscv-tools"
|
|
USE_PINNED_DEPS=true
|
|
BATCH_FLAG=""
|
|
SKIP_CONDA=false
|
|
SKIP_TOOLCHAIN=false
|
|
|
|
# getopts does not support long options, and is inflexible
|
|
while [ "$1" != "" ];
|
|
do
|
|
case $1 in
|
|
-h | --help )
|
|
usage 3 ;;
|
|
riscv-tools | esp-tools)
|
|
TOOLCHAIN=$1 ;;
|
|
-ud | --unpinned-deps )
|
|
USE_PINNED_DEPS=false ;;
|
|
--batch)
|
|
BATCH_FLAG=$1 ;;
|
|
--skip-conda)
|
|
SKIP_CONDA=true ;;
|
|
--skip-toolchain-extra)
|
|
SKIP_TOOLCHAIN=true ;;
|
|
* )
|
|
error "invalid option $1"
|
|
usage 1 ;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
if [ "$SKIP_CONDA" = false ]; then
|
|
# note: lock file must end in .conda-lock.yml - see https://github.com/conda-incubator/conda-lock/issues/154
|
|
LOCKFILE=$DIR/conda-requirements-$TOOLCHAIN-linux-64.conda-lock.yml
|
|
YAMLFILE=$DIR/conda-requirements-$TOOLCHAIN.yaml
|
|
|
|
if [ "$USE_PINNED_DEPS" = false ]; then
|
|
# auto-gen the lockfile
|
|
conda-lock -f $YAMLFILE -p linux-64 --lockfile $LOCKFILE
|
|
fi
|
|
|
|
# use conda-lock to create env
|
|
conda-lock install -p $DIR/.conda-env $LOCKFILE
|
|
|
|
source $DIR/.conda-env/etc/profile.d/conda.sh
|
|
conda activate $DIR/.conda-env
|
|
fi
|
|
|
|
if [ -z "$BATCH_FLAG" ]; then
|
|
if [ -z ${CONDA_DEFAULT_ENV+x} ]; then
|
|
error "ERROR: No conda environment detected. Did you activate the conda environment (e.x. 'conda activate base')?"
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
$DIR/scripts/init-submodules-no-riscv-tools.sh $BATCH_FLAG
|
|
|
|
if [ "$SKIP_TOOLCHAIN" = false ]; then
|
|
$DIR/scripts/build-toolchain-extra.sh $BATCH_FLAG $TOOLCHAIN
|
|
fi
|
|
|
|
$DIR/scripts/gen-tags.sh
|
|
|
|
cat << EOT >> env.sh
|
|
# line auto-generated by $0
|
|
conda activate $DIR/.conda-env
|
|
EOT
|