Skip to content

Commit c990823

Browse files
authored
Update environment-variables.md
updated envars auto passed on windows rocm installs
1 parent 87393c5 commit c990823

1 file changed

Lines changed: 9 additions & 1 deletion

File tree

docs/advanced/environment-variables.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,15 @@ Most users should leave these alone unless they are troubleshooting a specific R
143143
| `COMFYUI_ENABLE_MIOPEN` | `1` | Tells ComfyUI to keep the MIOpen-backed path enabled on ROCm builds where it may otherwise be disabled by default. Without this enabled, ComfyUI disables the `cudnn` backend path in its backend calls for RDNA 3, RDNA 4, and newer AMD GPUs, which in turn disables the MIOpen-backed functions that rely on that path. This variable is needed for MIOpen to function properly in those setups. |
144144
| `TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL` | `1` | Enables the experimental ROCm AOTriton path in compatible PyTorch builds. In Stability Matrix's Windows ROCm ComfyUI integration, this is used for TheRock technical-preview PyTorch builds to enable AOTriton-provided built-in Flash Attention and PyTorch SDPA memory-efficient attention paths. |
145145

146-
For some Windows ROCm-based ComfyUI launches, Stability Matrix already applies several of these optimizations automatically in package code, including `MIOPEN_FIND_MODE=2`, `TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1`, `PYTORCH_TUNABLEOP_ENABLED=1`, and `COMFYUI_ENABLE_MIOPEN=1`. Linux installs do not currently get the same automatic overrides, so they will need to be enabled by the user.
146+
For some Windows ROCm-based ComfyUI launches, Stability Matrix already applies several of these optimizations automatically in package code, including:
147+
148+
`MIOPEN_FIND_MODE=2`
149+
`MIOPEN_SEARCH_CUTOFF=2` `TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1` (RDNA3 and newer only)
150+
`FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE` `COMFYUI_ENABLE_MIOPEN=1`
151+
`PYTORCH_ALLOC_CONF=max_split_size_mb:512,garbage_collection_threshold:0.8`
152+
153+
154+
Linux installs do not currently get the same automatic overrides, so they will need to be enabled by the user.
147155

148156
For a broader reference, see the [official ROCm environment variable documentation](https://rocm.docs.amd.com/en/latest/reference/env-variables.html) and the [official MIOpen environment variable documentation](https://rocm.docs.amd.com/projects/MIOpen/en/latest/reference/env_variables.html).
149157

0 commit comments

Comments
 (0)