Fix broken QNN models on benchmarks (#7630)

Co-authored-by: Github Executorch <[email protected]>

Author: 2 people

.ci/scripts/test_model.sh

@@ -169,6 +169,13 @@ test_model_with_qnn() {
     EXPORT_SCRIPT=inception_v3
   elif [[ "${MODEL_NAME}" == "vit" ]]; then
     EXPORT_SCRIPT=torchvision_vit
+  elif [[ "${MODEL_NAME}" == "edsr" ]]; then
+    EXPORT_SCRIPT=edsr
+    # Additional deps for edsr
+    pip install piq
+  else
+    echo "Unsupported model $MODEL_NAME"
+    exit 1
   fi
 
   # Use SM8450 for S22, SM8550 for S23, and SM8560 for S24

examples/qualcomm/scripts/edsr.py

@@ -102,14 +102,17 @@ def main(args):
             "Please specify a device serial by -s/--device argument."
         )
 
-    dataset = get_dataset(
-        args.hr_ref_dir, args.lr_dir, args.default_dataset, args.artifact
-    )
-
-    inputs, targets, input_list = dataset.lr, dataset.hr, dataset.get_input_list()
-    pte_filename = "edsr_qnn_q8"
     instance = EdsrModel()
+    if args.compile_only:
+        inputs = instance.get_example_inputs()
+    else:
+        dataset = get_dataset(
+            args.hr_ref_dir, args.lr_dir, args.default_dataset, args.artifact
+        )
 
+        inputs, targets, input_list = dataset.lr, dataset.hr, dataset.get_input_list()
+
+    pte_filename = "edsr_qnn_q8"
     build_executorch_binary(
         instance.get_eager_model().eval(),
         (inputs[0],),