Temp fix on amd vendor

Author: winskuo-quic

backends/qualcomm/__init__.py

@@ -1,7 +1,9 @@
 import os
 
-from .scripts.download_qnn_sdk import install_qnn_sdk, is_linux_x86, QNN_ZIP_URL
+import cpuinfo
+import torch
 
+from .scripts.download_qnn_sdk import install_qnn_sdk, is_linux_x86, QNN_ZIP_URL
 
 env_flag = os.getenv("EXECUTORCH_BUILDING_WHEEL", "0").lower()
 # If users have preinstalled QNN_SDK_ROOT, we will use it.
@@ -27,3 +29,8 @@
                 "       export LD_LIBRARY_PATH="
                 "$QNN_SDK_ROOT/lib/x86_64-linux-clang/:$LD_LIBRARY_PATH"
             )
+
+info = cpuinfo.get_cpu_info()
+vendor = info.get("vendor_id_raw", "").lower()
+if "amd" in vendor:
+    torch.backends.mkldnn.enabled = False

backends/qualcomm/scripts/build.sh

@@ -7,6 +7,7 @@
 set -e
 
 pip install pydot
+pip install py-cpuinfo
 
 # Check if running on macOS/Darwin
 if [[ "$(uname -s)" == "Darwin" ]]; then