From 965ad7a4857f272629f69d1c68146f86cbdb4d0a Mon Sep 17 00:00:00 2001
From: NeuralFault <65365345+NeuralFault@users.noreply.github.com>
Date: Sun, 18 Jan 2026 19:54:56 -0500
Subject: [PATCH] Corrected gfx110X Windows ROCm nightly index. Added Win-ROCm
 EnVars

Changed index from 110X-dgpu to 110X-all as former stopped getting builds and was merged with other 110X archs in the latter index.

Added new environment variables to enhance GPU stability and performance, including settings to prevent out-of-memory errors and re-enable CUDNN in ComfyUI to allow proper MiOpen support.
---
 StabilityMatrix.Core/Models/Packages/ComfyUI.cs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs
index 6af2b047..e0fe67e6 100644
--- a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs
+++ b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs
@@ -382,7 +382,7 @@ await StandardPipInstallProcessAsync(
             var indexUrl = gfxArch switch
             {
                 "gfx1151" => "https://rocm.nightlies.amd.com/v2/gfx1151",
-                _ when gfxArch.StartsWith("gfx110") => "https://rocm.nightlies.amd.com/v2/gfx110X-dgpu",
+                _ when gfxArch.StartsWith("gfx110") => "https://rocm.nightlies.amd.com/v2/gfx110X-all",
                 _ when gfxArch.StartsWith("gfx120") => "https://rocm.nightlies.amd.com/v2/gfx120X-all",
                 _ => throw new ArgumentOutOfRangeException(
                     nameof(gfxArch),
@@ -870,6 +870,8 @@ private ImmutableDictionary<string, string> GetEnvVars(ImmutableDictionary<strin
         // set some experimental speed improving env vars for Windows ROCm
         return env.SetItem("PYTORCH_TUNABLEOP_ENABLED", "1")
             .SetItem("MIOPEN_FIND_MODE", "2")
-            .SetItem("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "1");
+            .SetItem("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "1")
+            .SetItem("PYTORCH_ALLOC_CONF", "max_split_size_mb:6144,garbage_collection_threshold:0.8") // greatly helps prevent GPU OOM and instability/driver timeouts/OS hard locks and decreases dependency on Tiled VAE at standard res's
+            .SetItem("COMFYUI_USE_MIOPEN", "1"); // re-enables "cudnn" in ComfyUI as it's needed for MiOpen to function properly
     }
 }