unparallelled commited on
Commit
1dcb8d2
·
verified ·
1 Parent(s): fe49cf2

Upload scripts/eval_c_early.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. scripts/eval_c_early.sh +16 -0
scripts/eval_c_early.sh ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ set -uo pipefail; cd "$(dirname "$0")"
3
+ export LD_LIBRARY_PATH="$(cd ../.. && pwd)/Tensor-Implementations/lib:$(cd ../.. && pwd)/Profiler/lib:${LD_LIBRARY_PATH:-}"
4
+ export PATH=/usr/local/cuda/bin:$PATH; export PY=/root/Adhi/BluTrain/envv/bin/python
5
+ L=eval_c_early_logs; mkdir -p "$L"; M="$L/master.log"
6
+ A680="CP_N_EMBD=1536 CP_N_LAYER=24 CP_N_HEAD=12 CP_N_KVHEAD=4 CP_FFN=4096 CP_WEIGHT_TYING=1"
7
+ say(){ echo "[$(date -u +%H:%M:%S)] $*" | tee -a "$M" >&2; }
8
+ blk(){ local PL="$1"; local O="ec_early_${PL}"
9
+ A1B="$A680" HAY="data/mix128_b6_60/val_shard_00000.bin" GEN_DATA_ROOT="data/mix128_b6_60" \
10
+ CKPT_EXTRA="c100=t128c_s100.ckpt c200=t128c_s200.ckpt" CKPTS="c100 c200" TASKS="niah_single" N=100 \
11
+ T=131072 YARN=32 YARN_ORIG=4096 PROMPT_LEN="$PL" GPUS="0,1,2,3" NP=1 JOBS_PER_GPU=2 OUT="$O" \
12
+ ./run_longeval.sh > "$L/len_${PL}.log" 2>&1
13
+ say " len $PL: $(head -1 $O/summary.csv|tr -d '\r') -> $(grep -h '^niah_single,' $O/summary.csv|tr -d '\r')"; }
14
+ say "=== t128c EARLY steps 100/200 (the missing baseline) ==="
15
+ blk 4096 & blk 32744 & wait
16
+ say "EVAL_C_EARLY_DONE"