52 lines
1.6 KiB
Bash
52 lines
1.6 KiB
Bash
#!/bin/bash
|
|
|
|
################################### Include variables and constants
|
|
|
|
. "common_520.sh"
|
|
|
|
echo "[WARNING] This script is deprecated. Please use the python API package ktc instead."
|
|
|
|
if [[ $# -lt 1 ]]; then
|
|
echo "You need to provide an bie model or an onnx to run this script."
|
|
exit 1
|
|
fi
|
|
|
|
model=$1
|
|
|
|
|
|
#################################### Main Process
|
|
|
|
# Prepare compiler input
|
|
mkdir -p $TMP_FOLDER
|
|
echo "running compiler and IP evaluator..."
|
|
delFile $TMP_FOLDER/config_compiler.json
|
|
delFolder $RESULT_FOLDER/compiler
|
|
mkdir -p $RESULT_FOLDER/compiler
|
|
|
|
python $LIBS_FOLDER/compiler/gen_config.py -t 520 -v model_rel -o $TMP_FOLDER/config_compiler.json -a ''{\"ip_evaluator_cfg\":\"$SCRIPT_FOLDER/res/ip_config_520.json\"}''
|
|
chk_flow $? "Generating compiler config fails!"
|
|
echo "Compiler config generated."
|
|
|
|
# # Run old compiler
|
|
# cd $LIBS_FOLDER/compiler/build/bin && ./compile $TMP_FOLDER/output.onnx $TMP_FOLDER/config_compiler.json
|
|
# chk_flow $? "compiler fails"
|
|
# Run piano compiler
|
|
cd $TMP_FOLDER && $LIBS_FOLDER/compiler/compile 520 $model $TMP_FOLDER/config_compiler.json warning compile.log
|
|
chk_flow $? "compuler and ip evaluator fails!"
|
|
|
|
# cp $TMP_FOLDER/ProfileResult.txt $RESULT_FOLDER/compiler/
|
|
# echo "finish save ip evaluator result"
|
|
|
|
if [[ $debug_mode == "true" ]]
|
|
then
|
|
cp $TMP_FOLDER/*.txt $RESULT_FOLDER/compiler/
|
|
cp $TMP_FOLDER/*.v $RESULT_FOLDER/compiler/
|
|
cp $TMP_FOLDER/compiler_config.json $RESULT_FOLDER/compiler/
|
|
fi
|
|
cp $TMP_FOLDER/*.bin $RESULT_FOLDER/compiler/
|
|
cp $TMP_FOLDER/ioinfo.csv $RESULT_FOLDER/compiler/
|
|
cp $TMP_FOLDER/ip_eval_prof.txt $RESULT_FOLDER/compiler/
|
|
echo "Compilation and IP Evaluation finished."
|
|
|
|
rm -rf $TMP_FOLDER
|