diff --git a/backends/qualcomm/__init__.py b/backends/qualcomm/__init__.py index 5770dfb0fcd..4864340d811 100644 --- a/backends/qualcomm/__init__.py +++ b/backends/qualcomm/__init__.py @@ -1,5 +1,8 @@ import os +import cpuinfo +import torch + from .scripts.download_qnn_sdk import install_qnn_sdk, is_linux_x86 @@ -11,3 +14,8 @@ ok = install_qnn_sdk() if not ok: raise RuntimeError("Failed to install QNN SDK. Please check the logs above.") + +info = cpuinfo.get_cpu_info() +vendor = info.get("vendor_id_raw", "").lower() +if "amd" in vendor: + torch.backends.mkldnn.enabled = False diff --git a/backends/qualcomm/scripts/build.sh b/backends/qualcomm/scripts/build.sh index e3b31a2fe1c..aebbb9bbd6c 100755 --- a/backends/qualcomm/scripts/build.sh +++ b/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