diff --git a/examples/arm/run.sh b/examples/arm/run.sh index 137aa171e95..44173ed8db7 100755 --- a/examples/arm/run.sh +++ b/examples/arm/run.sh @@ -39,7 +39,8 @@ function help() { echo "Usage: $(basename $0) [options]" echo "Options:" echo " --model_name= Model file .py/.pth/.pt, builtin model or a model from examples/models. Passed to aot_arm_compiler" - echo " --model_input= Provide model input .pt file to override the input in the model file. Passed to aot_arm_compiler" + echo " --model_input= Provide model input .pt file to override the input in the model file. Passed to aot_arm_compiler" + echo " NOTE: Inference in FVP is done with a dummy input full of ones. Use bundleio flag to run the model in FVP with the custom input or the input from the model file." echo " --aot_arm_compiler_flags= Only used if --model_name is used Default: ${aot_arm_compiler_flags}" echo " --portable_kernels= Comma separated list of portable (non delagated) kernels to include Default: ${portable_kernels}" echo " --target= Target to build and run for Default: ${target}" @@ -200,7 +201,11 @@ for i in "${!test_model[@]}"; do # Remove old pte files rm -f "${output_folder}/${model_filename}" - + + if [ "$model_input_set" = true ]; then + model_compiler_flags="${model_compiler_flags} --model_input=${model_input}" + fi + ARM_AOT_CMD="python3 -m examples.arm.aot_arm_compiler --model_name=${model} --target=${target} ${model_compiler_flags} --intermediate=${output_folder} --output=${pte_file} --so_library=$SO_LIB --system_config=${system_config} --memory_mode=${memory_mode} $bundleio_flag" echo "CALL ${ARM_AOT_CMD}" >&2 ${ARM_AOT_CMD} 1>&2