Qualcomm AI Engine Direct - AMD backend error (pytorch#18098)

Author: winskuo-quic

.ci/scripts/setup-qnn-deps.sh

@@ -11,4 +11,5 @@ source "$(dirname "${BASH_SOURCE[0]}")/../../backends/qualcomm/scripts/install_q
 
 setup_libcpp 12
 setup_android_ndk
-install_qnn
+install_qnn
+pip install -r backends/qualcomm/requirements.txt

.ci/scripts/test_wheel_package_qnn.sh

@@ -177,6 +177,9 @@ PY
   "$PIPBIN" install . --no-build-isolation
   popd > /dev/null
 
+  # Install qualcomm backend dependencies  
+  "$PIPBIN" install -r "$REPO_ROOT/backends/qualcomm/requirements.txt"
+
   echo "=== [$LABEL] Import smoke tests ==="
   "$PYBIN" -c "import executorch; print('executorch imported successfully')"
   "$PYBIN" -c "import executorch.backends.qualcomm; print('executorch.backends.qualcomm imported successfully')"

backends/qualcomm/__init__.py

@@ -1,7 +1,19 @@
 import os
 
+import torch
+
 from .scripts.download_qnn_sdk import install_qnn_sdk, is_linux_x86, QNN_ZIP_URL
 
+try:
+    import cpuinfo
+
+    info = cpuinfo.get_cpu_info()
+    vendor = info.get("vendor_id_raw", "").lower()
+    if "amd" in vendor:
+        torch.backends.mkldnn.enabled = False
+except ImportError:
+    raise ImportError("Please install the cpuinfo with pip install py-cpuinfo.")
+
 
 env_flag = os.getenv("EXECUTORCH_BUILDING_WHEEL", "0").lower()
 # If users have preinstalled QNN_SDK_ROOT, we will use it.

backends/qualcomm/requirements.txt

@@ -0,0 +1,5 @@
+graphviz
+pydot
+py-cpuinfo
+requests
+tabulate

backends/qualcomm/scripts/build.sh

@@ -6,7 +6,7 @@
 # LICENSE file in the root directory of this source tree.
 set -e
 
-pip install pydot
+pip install -r backends/qualcomm/requirements.txt
 
 # Check if running on macOS/Darwin
 if [[ "$(uname -s)" == "Darwin" ]]; then