From 757bd5b82529c9532f5ee7c068e592253821d255 Mon Sep 17 00:00:00 2001 From: winskuo-quic Date: Wed, 11 Mar 2026 04:19:44 -0700 Subject: [PATCH 1/2] Temp fix on amd vendor --- backends/qualcomm/__init__.py | 8 +++++++- backends/qualcomm/scripts/build.sh | 1 + 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/backends/qualcomm/__init__.py b/backends/qualcomm/__init__.py index 5770dfb0fcd..64323c022da 100644 --- a/backends/qualcomm/__init__.py +++ b/backends/qualcomm/__init__.py @@ -1,5 +1,6 @@ import os - +import cpuinfo +import torch from .scripts.download_qnn_sdk import install_qnn_sdk, is_linux_x86 @@ -11,3 +12,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 \ No newline at end of file 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 From 6455008fb509938a2681237c7adb7cb58fb582bb Mon Sep 17 00:00:00 2001 From: winskuo-quic Date: Wed, 11 Mar 2026 20:58:26 -0700 Subject: [PATCH 2/2] lint --- backends/qualcomm/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backends/qualcomm/__init__.py b/backends/qualcomm/__init__.py index 64323c022da..4864340d811 100644 --- a/backends/qualcomm/__init__.py +++ b/backends/qualcomm/__init__.py @@ -1,6 +1,8 @@ import os + import cpuinfo import torch + from .scripts.download_qnn_sdk import install_qnn_sdk, is_linux_x86 @@ -16,4 +18,4 @@ info = cpuinfo.get_cpu_info() vendor = info.get("vendor_id_raw", "").lower() if "amd" in vendor: - torch.backends.mkldnn.enabled = False \ No newline at end of file + torch.backends.mkldnn.enabled = False