From df418fc3fde66bdf86e62e0fc50fda2e7fd2f539 Mon Sep 17 00:00:00 2001
From: patientx <patient.x@gmail.com>
Date: Sun, 19 Jan 2025 17:49:55 +0300
Subject: [PATCH] added rocm and hip hiding for certain nodes getting errors
 under zluda

---
 comfy/zluda.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/comfy/zluda.py b/comfy/zluda.py
index 4f02aeb2e..1005fcdfc 100644
--- a/comfy/zluda.py
+++ b/comfy/zluda.py
@@ -1,3 +1,15 @@
+# hide rocm and hip
+import os
+os.environ.pop("ROCM_HOME",None)
+os.environ.pop("HIP_HOME",None)
+os.environ.pop("ROCM_VERSION",None)
+paths=os.environ["PATH"].split(";")
+paths_no_rocm=[]
+for path_ in paths:
+    if "rocm" not in path_.lower():
+        paths_no_rocm.append(path_)
+os.environ["PATH"]=";".join(paths_no_rocm)
+
 import torch
 
 if torch.cuda.get_device_name().endswith("[ZLUDA]"):