occlum/demos/openvino/run_benchmark_on_occlum.sh
2023-11-07 09:26:20 +08:00

34 lines
1.2 KiB
Bash
Executable File

#!/bin/bash
benchmark=benchmark_app
set -e
SCRIPT_DIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
bomfile=${SCRIPT_DIR}/openvino.yaml
# 1. Init Occlum Workspace
rm -rf occlum_instance
mkdir occlum_instance
cd occlum_instance
occlum init
cpu_cc=`cat /proc/cpuinfo | grep processor | wc -l`
#new_json="$(jq '.resource_limits.user_space_size = "4GB" |
# .resource_limits.kernel_space_heap_size = "128MB" |
# .resource_limits.kernel_space_stack_size = "16MB" |
# .resource_limits.max_num_of_threads = 128 |
# .process.default_stack_size = "8MB" |
# .process.default_heap_size = "32MB" |
# .metadata.debuggable = false ' Occlum.json)" && \
new_json="$(jq '.resource_limits.max_num_of_threads = 64 |
.resource_limits.user_space_size = "1MB" |
.resource_limits.user_space_max_size = "600MB"' Occlum.json)" && \
echo "${new_json}" > Occlum.json
# 2. Copy files into Occlum Workspace and Build
rm -rf image
copy_bom -f $bomfile --root image --include-dir /opt/occlum/etc/template
occlum build
# 3. Run benchmark
occlum run /bin/$benchmark -m /model/age-gender-recognition-retail-0013.xml