Add --verbose flag for riscv64 scripts

Author: luhenry

.ci/scripts/test_riscv_qemu.sh

@@ -17,6 +17,7 @@ et_root_dir=$(realpath "${script_dir}/../..")
 model="add"
 xnnpack=false
 quantize=false
+verbose=false
 
 usage() {
     cat <<EOF
@@ -34,6 +35,7 @@ for arg in "$@"; do
         --model=*) model="${arg#*=}" ;;
         --xnnpack) xnnpack=true ;;
         --quantize) quantize=true ;;
+        --verbose) verbose=true ;;
         -h|--help) usage; exit 0 ;;
         *) echo "Unknown option: $arg" >&2; usage; exit 1 ;;
     esac
@@ -46,6 +48,9 @@ fi
 if ${quantize}; then
     run_extra_args+=(--quantize)
 fi
+if ${verbose}; then
+    run_extra_args+=(--verbose)
+fi
 
 bash "${et_root_dir}/examples/riscv/setup.sh"
 bash "${et_root_dir}/examples/riscv/run.sh" --model="${model}" "${run_extra_args[@]}"

examples/riscv/aot_riscv.py

@@ -11,6 +11,7 @@
 """
 
 import argparse
+import logging
 from pathlib import Path
 
 import torch
@@ -75,8 +76,16 @@ def main() -> None:
         action="store_true",
         help="Produce an 8-bit quantized model",
     )
+    parser.add_argument(
+        "--verbose",
+        action="store_true",
+        help="Enable XNNPACK partitioner DEBUG logging and dump the lowered graph",
+    )
     args = parser.parse_args()
 
+    if args.verbose:
+        logging.basicConfig(level=logging.DEBUG)
+
     if args.output is None:
         args.output = Path(f"{args.model}_riscv.bpte")
 
@@ -100,7 +109,8 @@ def main() -> None:
         from executorch.backends.xnnpack.partition.xnnpack_partitioner import (
             XnnpackPartitioner,
         )
-        partitioners.append(XnnpackPartitioner())
+
+        partitioners.append(XnnpackPartitioner(verbose=args.verbose))
 
     compile_config = None
     if args.quantize:
@@ -121,6 +131,11 @@ def main() -> None:
         f"quantize={args.quantize} delegated_nodes={delegated}"
     )
 
+    if args.verbose:
+        from executorch.exir.backend.utils import print_delegated_graph
+
+        print_delegated_graph(edge.exported_program().graph_module)
+
     et_program = edge.to_executorch()
 
     test_suite = MethodTestSuite(

examples/riscv/run.sh

@@ -23,6 +23,7 @@ qemu_timeout="600"
 model="add"
 xnnpack=false
 quantize=false
+verbose=false
 
 usage() {
     cat <<EOF
@@ -31,6 +32,7 @@ Options:
   --model=<NAME>          Which model to export and run (default: ${model})
   --xnnpack               Enable the XNNPACK backend (AOT partitioner + runtime)
   --quantize              Produce an 8-bit quantized model
+  --verbose               Enable XNNPACK partitioner DEBUG logging and dump the lowered graph
   --build_only            Only export and cross-compile; do not invoke QEMU
   --build_dir=<DIR>       CMake build directory (default: ${build_dir})
   --output_dir=<DIR>      Directory for the exported .bpte (default: ${output_dir})
@@ -45,6 +47,7 @@ for arg in "$@"; do
         --model=*) model="${arg#*=}" ;;
         --xnnpack) xnnpack=true ;;
         --quantize) quantize=true ;;
+        --verbose) verbose=true ;;
         --build_only) build_only=true ;;
         --build_dir=*) build_dir="${arg#*=}" ;;
         --output_dir=*) output_dir="${arg#*=}" ;;
@@ -66,6 +69,9 @@ fi
 if ${quantize}; then
     aot_extra_args+=(--quantize)
 fi
+if ${verbose}; then
+    aot_extra_args+=(--verbose)
+fi
 python "${script_dir}/aot_riscv.py" --model "${model}" "${aot_extra_args[@]}" --output "${bpte_path}"
 
 echo "[run.sh] Step 2/3: cross-compile executor_runner for riscv64-linux"