#!/bin/bash
set -e

file_name=$(basename "$0" .sh)
cur_dir="$(cd "$(dirname "$0")"; pwd)"
source "${cur_dir}/common.sh"
output_path="$(resolve_output_path "${1:-}" "$0")"

testcase_result_dir="${output_path}/${file_name}"
init_testcase_dir "${testcase_result_dir}"
start_time=$(date "+%s")

python3 "${cur_dir}/src/train.py" --model acl_graph --prof-path "${testcase_result_dir}/result_dir" > "${testcase_result_dir}/plog.txt" 2>&1
check_run_exit_code "$?" "${file_name}" "${testcase_result_dir}"

python3 "${cur_dir}/src/${file_name}.py" "${testcase_result_dir}/result_dir"
check_run_exit_code "$?" "${file_name}" "${testcase_result_dir}"

end_time=$(date "+%s")
print_result "${file_name}" "${start_time}" "${end_time}"