Reneabled QNN flow

Author: suryasidd

.ci/scripts/setup-openvino.sh

@@ -10,17 +10,40 @@ set -ex
 # shellcheck source=/dev/null
 source "$(dirname "${BASH_SOURCE[0]}")/utils.sh"
 
+# Parse arguments
+USE_NIGHTLY=false
+for arg in "$@"; do
+  case $arg in
+    --nightly) USE_NIGHTLY=true ;;
+  esac
+done
+
 # Download and install OpenVINO from release packages
 OPENVINO_VERSION="2026.0"
 OPENVINO_BUILD="2026.0.0.20965.c6d6a13a886"
-OPENVINO_URL="https://storage.openvinotoolkit.org/repositories/openvino/packages/${OPENVINO_VERSION}/linux/openvino_toolkit_ubuntu22_${OPENVINO_BUILD}_x86_64.tgz"
+OPENVINO_STABLE_URL="https://storage.openvinotoolkit.org/repositories/openvino/packages/${OPENVINO_VERSION}/linux/openvino_toolkit_ubuntu22_${OPENVINO_BUILD}_x86_64.tgz"
+
+OPENVINO_NIGHTLY_BUILD_ID="2026.1.0-21296-4589d335731"
+OPENVINO_NIGHTLY_BUILD="2026.1.0.dev20260311"
+OPENVINO_NIGHTLY_URL="https://storage.openvinotoolkit.org/repositories/openvino/packages/nightly/${OPENVINO_NIGHTLY_BUILD_ID}/openvino_toolkit_ubuntu22_${OPENVINO_NIGHTLY_BUILD}_x86_64.tgz"
+
+if [ "${USE_NIGHTLY}" = true ]; then
+  OPENVINO_URL="${OPENVINO_NIGHTLY_URL}"
+  OPENVINO_EXTRACTED_DIR="openvino_toolkit_ubuntu22_${OPENVINO_NIGHTLY_BUILD}_x86_64"
+  echo "Using OpenVINO nightly build: ${OPENVINO_NIGHTLY_BUILD_ID}"
+else
+  OPENVINO_URL="${OPENVINO_STABLE_URL}"
+  OPENVINO_EXTRACTED_DIR="openvino_toolkit_ubuntu22_${OPENVINO_BUILD}_x86_64"
+  echo "Using OpenVINO stable release: ${OPENVINO_BUILD}"
+fi
 
 curl -Lo /tmp/openvino_toolkit.tgz --retry 3 --fail ${OPENVINO_URL}
 tar -xzf /tmp/openvino_toolkit.tgz
-mv openvino_toolkit_ubuntu22_${OPENVINO_BUILD}_x86_64 openvino
+mv "${OPENVINO_EXTRACTED_DIR}" openvino
 
+python_version=${python_version:-}
 source openvino/setupvars.sh
 cd backends/openvino
 pip install -r requirements.txt
 cd scripts
-./openvino_build.sh --enable_python
+./openvino_build.sh --enable_python

.ci/scripts/test_backend.sh

@@ -80,7 +80,7 @@ fi
 
 if [[ "$FLOW" == *openvino* ]]; then
     # Setup OpenVINO environment
-    source .ci/scripts/setup-openvino.sh
+    source .ci/scripts/setup-openvino.sh --nightly
     EXTRA_BUILD_ARGS+=" -DEXECUTORCH_BUILD_OPENVINO=ON"
 fi
 

backends/openvino/requirements.txt

@@ -1 +1 @@
-nncf==3.0.0
+git+https://github.com/openvinotoolkit/nncf@b101e7e#egg=nncf

backends/test/suite/flow.py

@@ -4,6 +4,7 @@
 # LICENSE file in the root directory of this source tree.
 
 import logging
+import os
 
 from dataclasses import dataclass, field
 from typing import Callable
@@ -118,26 +119,29 @@ def all_flows() -> dict[str, TestFlow]:
     except Exception as e:
         logger.info(f"Skipping OpenVINO flow registration: {e}")
 
-    # try:
-    #     from executorch.backends.test.suite.flows.qualcomm import (
-    #         QNN_16A16W_TEST_FLOW,
-    #         QNN_16A4W_BLOCK_TEST_FLOW,
-    #         QNN_16A4W_TEST_FLOW,
-    #         QNN_16A8W_TEST_FLOW,
-    #         QNN_8A8W_TEST_FLOW,
-    #         QNN_TEST_FLOW,
-    #     )
-
-    #     flows += [
-    #         QNN_TEST_FLOW,
-    #         QNN_16A16W_TEST_FLOW,
-    #         QNN_16A8W_TEST_FLOW,
-    #         QNN_16A4W_TEST_FLOW,
-    #         QNN_16A4W_BLOCK_TEST_FLOW,
-    #         QNN_8A8W_TEST_FLOW,
-    #     ]
-    # except Exception as e:
-    #     logger.info(f"Skipping QNN flow registration: {e}")
+    if os.environ.get("QNN_SDK_ROOT"):
+        try:
+            from executorch.backends.test.suite.flows.qualcomm import (
+                QNN_16A16W_TEST_FLOW,
+                QNN_16A4W_BLOCK_TEST_FLOW,
+                QNN_16A4W_TEST_FLOW,
+                QNN_16A8W_TEST_FLOW,
+                QNN_8A8W_TEST_FLOW,
+                QNN_TEST_FLOW,
+            )
+
+            flows += [
+                QNN_TEST_FLOW,
+                QNN_16A16W_TEST_FLOW,
+                QNN_16A8W_TEST_FLOW,
+                QNN_16A4W_TEST_FLOW,
+                QNN_16A4W_BLOCK_TEST_FLOW,
+                QNN_8A8W_TEST_FLOW,
+            ]
+        except Exception as e:
+            logger.info(f"Skipping QNN flow registration: {e}")
+    else:
+        logger.info("Skipping QNN flow registration: QNN_SDK_ROOT not set")
 
     try:
         from executorch.backends.test.suite.flows.arm import (