Skip to content

Commit 965ad7a

Browse files
authored
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.
1 parent 895b247 commit 965ad7a

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

StabilityMatrix.Core/Models/Packages/ComfyUI.cs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ await StandardPipInstallProcessAsync(
382382
var indexUrl = gfxArch switch
383383
{
384384
"gfx1151" => "https://rocm.nightlies.amd.com/v2/gfx1151",
385-
_ when gfxArch.StartsWith("gfx110") => "https://rocm.nightlies.amd.com/v2/gfx110X-dgpu",
385+
_ when gfxArch.StartsWith("gfx110") => "https://rocm.nightlies.amd.com/v2/gfx110X-all",
386386
_ when gfxArch.StartsWith("gfx120") => "https://rocm.nightlies.amd.com/v2/gfx120X-all",
387387
_ => throw new ArgumentOutOfRangeException(
388388
nameof(gfxArch),
@@ -870,6 +870,8 @@ private ImmutableDictionary<string, string> GetEnvVars(ImmutableDictionary<strin
870870
// set some experimental speed improving env vars for Windows ROCm
871871
return env.SetItem("PYTORCH_TUNABLEOP_ENABLED", "1")
872872
.SetItem("MIOPEN_FIND_MODE", "2")
873-
.SetItem("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "1");
873+
.SetItem("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "1")
874+
.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
875+
.SetItem("COMFYUI_USE_MIOPEN", "1"); // re-enables "cudnn" in ComfyUI as it's needed for MiOpen to function properly
874876
}
875877
}

0 commit comments

Comments
 (0)