From eb465ad574436c252d4bd88154d24b2fde9139a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=87=8E=E7=94=9F=E3=81=AE=E7=94=B7?=
 <Yasei-no-otoko@users.noreply.github.com>
Date: Fri, 3 Apr 2026 01:26:15 +0900
Subject: [PATCH] Restrict Windows ROCm SDK env setup to ROCm builds

---
 main.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/main.py b/main.py
index 569babbe6..51e062005 100644
--- a/main.py
+++ b/main.py
@@ -31,10 +31,21 @@ if __name__ == "__main__":
 setup_logger(log_level=args.verbose, use_stdout=args.log_stdout)
 
 
+def is_rocm_torch_build():
+    try:
+        torch_version = importlib.metadata.version("torch").lower()
+    except importlib.metadata.PackageNotFoundError:
+        return False
+    return "rocm" in torch_version
+
+
 def configure_windows_rocm_sdk_env():
     if os.name != "nt":
         return None
 
+    if not is_rocm_torch_build():
+        return None
+
     if any(os.environ.get(var) for var in ("ROCM_HOME", "ROCM_PATH", "HIP_PATH")):
         return None