-
-
Notifications
You must be signed in to change notification settings - Fork 509
ComfyUI: Updated gfx110X Windows ROCm nightly index. Added Win-ROCm EnVars #1524
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
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.
Summary of ChangesHello @NeuralFault, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed! This pull request focuses on improving the robustness and compatibility of ComfyUI installations on Windows systems utilizing AMD ROCm. It updates the ROCm nightly build index to a more current and comprehensive source, and crucially, introduces specific environment variables designed to mitigate common GPU-related issues such as out-of-memory errors and ensure the correct functioning of MiOpen for accelerated computations. Highlights
🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console. Using Gemini Code AssistThe full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips. Invoking Gemini You can request assistance from Gemini at any point by creating a comment using either
Customization To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a Limitations & Feedback Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here. You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension. Footnotes
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Code Review
This pull request updates the ROCm nightly index URL for gfx110X architectures and adds two new environment variables for Windows ROCm users to improve GPU stability and performance in ComfyUI. The changes look good and align with the description. I've added a couple of suggestions to improve code readability and maintainability by refactoring long comments and suggesting the use of constants for magic numbers. Overall, a good improvement for AMD users.
| .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 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This line contains magic numbers (6144, 0.8) and a very long comment, which can impact readability and maintainability.
I recommend extracting the numeric values into named constants to make their purpose explicit and easier to modify in the future. For example:
private const int PytorchAllocMaxSplitSizeMb = 6144;
private const double PytorchAllocGcThreshold = 0.8;Additionally, the comment should be wrapped to fit within standard line length limits. I've provided a suggestion for reformatting the comment. Please consider applying the constants as well.
// Greatly helps prevent GPU OOM, instability, driver timeouts, and OS hard locks.
// It also decreases dependency on Tiled VAE at standard resolutions.
.SetItem("PYTORCH_ALLOC_CONF", "max_split_size_mb:6144,garbage_collection_threshold:0.8")| .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 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This comment is quite long and makes the line exceed typical length recommendations. For better readability, it's good practice to place comments on their own line(s) above the code they refer to, especially when they are descriptive.
// Re-enables "cudnn" in ComfyUI as it's needed for MIOpen to function properly.
.SetItem("COMFYUI_USE_MIOPEN", "1");
mohnjiles
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Sweet, thanks!
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.