Use lean conda for faster initial setup

scripts/build-setup.sh

@@ -38,6 +38,7 @@ usage() {
     echo "  --help -h               : Display this message"
     echo "  --verbose -v            : Verbose printout"
     echo "  --use-unpinned-deps -ud : Use unpinned conda environment"
+    echo "  --use-lean-conda        : Install a leaner version of conda (no firesim && no marshal)"
 
     echo "  --skip -s N             : Skip step N in the list above. Use multiple times to skip multiple steps ('-s N -s M ...')."
     echo "  --skip-conda            : Skip Conda initialization (step 1)"
@@ -57,6 +58,7 @@ TOOLCHAIN_TYPE="riscv-tools"
 VERBOSE=false
 VERBOSE_FLAG=""
 USE_UNPINNED_DEPS=false
+USE_LEAN_CONDA=false
 SKIP_LIST=()
 
 # getopts does not support long options, and is inflexible
@@ -70,6 +72,9 @@ do
         --verbose | -v)
             VERBOSE_FLAG=$1
             set -x ;;
+        --use-lean-conda)
+            USE_LEAN_CONDA=true
+            SKIP_LIST+=(4 6 7 8 9) ;;
         -ud | --use-unpinned-deps )
             USE_UNPINNED_DEPS=true ;;
         --skip | -s)
@@ -161,13 +166,22 @@ if run_step "1"; then
     CONDA_REQS=$CYDIR/conda-reqs
     CONDA_LOCK_REQS=$CONDA_REQS/conda-lock-reqs
     # must match with the file generated by generate-conda-lockfile.sh
-    LOCKFILE=$CONDA_LOCK_REQS/conda-requirements-$TOOLCHAIN_TYPE-linux-64.conda-lock.yml
+    if [ "$USE_LEAN_CONDA" = false ]; then
+      LOCKFILE=$CONDA_LOCK_REQS/conda-requirements-$TOOLCHAIN_TYPE-linux-64.conda-lock.yml
+    else
+      if [ "$TOOLCHAIN_TYPE" != "riscv-tools" ]; then
+        echo "::ERROR:: Lean conda env is not supported for esp-tools"
+        exit 1
+      fi
+      LOCKFILE=$CONDA_LOCK_REQS/conda-requirements-$TOOLCHAIN_TYPE-linux-64-lean.conda-lock.yml
+    fi
 
     if [ "$USE_UNPINNED_DEPS" = true ]; then
         # auto-gen the lockfiles
         $CYDIR/scripts/generate-conda-lockfiles.sh
         exit_if_last_command_failed
     fi
+    echo "lockfile $LOCKFILE"
 
     # use conda-lock to create env
     conda-lock install --conda $(which conda) -p $CYDIR/.conda-env $LOCKFILE &&