Files
chipyard/scripts/generate-ckpt.sh
2023-04-17 13:19:16 -07:00

134 lines
3.4 KiB
Bash
Executable File

#!/bin/bash
set -e
usage() {
echo "Usage: $0 [OPTIONS] -- [SPIKEFLAGS]"
echo ""
echo "Options"
echo " --help -h : Display this message"
echo " -n <n> : Mumber of harts"
echo " -b <elf> : Binary to run in spike"
echo " -p <pc> : PC to take checkpoint at [default 0x80000000]"
echo " -c <cycles>: Cycles after PC to take checkpoint at [default 0]"
echo " -m <isa> : ISA to pass to spike for checkpoint generation [defualt rv64gc]"
exit "$1"
}
NHARTS=1
BINARY=""
PC="0x80000000"
CYCLES=0
ISA="rv64gc"
while [ "$1" != "" ];
do
case $1 in
-h | --help )
usage 3 ;;
-n )
shift
NHARTS=$1 ;;
-b )
shift
BINARY=$1 ;;
-p )
shift
PC=$1 ;;
-c )
shift
CYCLES=$1 ;;
-m )
shift
ISA=$1 ;;
* )
error "Invalid option $1"
usage 1 ;;
esac
shift
done
BASEMEM="$((0x80000000)):$((0x10000000))"
SPIKEFLAGS="-p$NHARTS --pmpregions=0 --isa=$ISA -m$BASEMEM"
BASENAME=$(basename -- $BINARY)
DIRNAME=$BASENAME.$PC.$CYCLES.loadarch
echo "Generating loadarch directory $DIRNAME"
rm -rf $DIRNAME
mkdir -p $DIRNAME
LOADARCH_FILE=$DIRNAME/loadarch
RAWMEM_ELF=$DIRNAME/raw.elf
LOADMEM_ELF=$DIRNAME/mem.elf
CMDS_FILE=$DIRNAME/cmds_tmp.txt
SPIKECMD_FILE=$DIRNAME/spikecmd.sh
echo "Generating state capture spike interactive commands in $CMDS_FILE"
echo "until pc 0 $PC" >> $CMDS_FILE
echo "rs $CYCLES" >> $CMDS_FILE
echo "dump" >> $CMDS_FILE
for (( h=0; h<$NHARTS; h++ ))
do
echo "pc $h" >> $CMDS_FILE
echo "priv $h" >> $CMDS_FILE
echo "reg $h fcsr" >> $CMDS_FILE
echo "reg $h vstart" >> $CMDS_FILE
echo "reg $h vxsat" >> $CMDS_FILE
echo "reg $h vxrm" >> $CMDS_FILE
echo "reg $h vcsr" >> $CMDS_FILE
echo "reg $h vtype" >> $CMDS_FILE
echo "reg $h stvec" >> $CMDS_FILE
echo "reg $h sscratch" >> $CMDS_FILE
echo "reg $h sepc" >> $CMDS_FILE
echo "reg $h scause" >> $CMDS_FILE
echo "reg $h stval" >> $CMDS_FILE
echo "reg $h satp" >> $CMDS_FILE
echo "reg $h mstatus" >> $CMDS_FILE
echo "reg $h medeleg" >> $CMDS_FILE
echo "reg $h mideleg" >> $CMDS_FILE
echo "reg $h mie" >> $CMDS_FILE
echo "reg $h mtvec" >> $CMDS_FILE
echo "reg $h mscratch" >> $CMDS_FILE
echo "reg $h mepc" >> $CMDS_FILE
echo "reg $h mcause" >> $CMDS_FILE
echo "reg $h mtval" >> $CMDS_FILE
echo "reg $h mip" >> $CMDS_FILE
echo "reg $h mcycle" >> $CMDS_FILE
echo "reg $h minstret" >> $CMDS_FILE
echo "mtime" >> $CMDS_FILE
echo "mtimecmp $h" >> $CMDS_FILE
for (( fr=0; fr<32; fr++ ))
do
echo "freg $h $fr" >> $CMDS_FILE
done
for (( xr=0; xr<32; xr++ ))
do
echo "reg $h $xr" >> $CMDS_FILE
done
echo "vreg $h" >> $CMDS_FILE
done
echo "quit" >> $CMDS_FILE
#cat $CMDS_FILE
echo "spike -d --debug-cmd=$CMDS_FILE $SPIKEFLAGS $BINARY" > $SPIKECMD_FILE
echo "Capturing state at checkpoint to spikeout"
spike -d --debug-cmd=$CMDS_FILE $SPIKEFLAGS $BINARY 2> $LOADARCH_FILE
echo "Finding tohost/fromhost in elf file"
TOHOST=$(riscv64-unknown-elf-nm $BINARY | grep tohost | head -c 16)
FROMHOST=$(riscv64-unknown-elf-nm $BINARY | grep fromhost | head -c 16)
echo "Compiling memory to elf"
riscv64-unknown-elf-objcopy -I binary -O elf64-littleriscv mem.0x80000000.bin $RAWMEM_ELF
rm -rf mem.0x80000000.bin
riscv64-unknown-elf-ld -Tdata=0x80000000 -nmagic --defsym tohost=0x$TOHOST --defsym fromhost=0x$FROMHOST -o $LOADMEM_ELF $RAWMEM_ELF
rm -rf $RAWMEM_ELF