#!/bin/bash
# Smoke gate: run the small 0.5B merge experiment on an L40S to confirm the GPU pipeline works
# on CX3 before the 4 h 7B run. ~5 min.  submit: qsub hpc/llm_smoke.pbs
#PBS -l select=1:ncpus=8:mem=48gb:ngpus=1:gpu_type=L40S
#PBS -l walltime=00:20:00
#PBS -N lam_llm_smoke

cd "$PBS_O_WORKDIR"
export HF_HOME="$EPHEMERAL/hf_cache"
export TOKENIZERS_PARALLELISM=false
source .venv/bin/activate
nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader
python -m llm.experiment configs/llm/merge.yaml   # 0.5B, writes results/llm_merge/
echo "done: $(date)"