wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-05 11:40:53 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Turn on TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL by default. (#10168)

Browse Source
This commit is contained in:
comfyanonymous 2025-10-02 14:57:15 -07:00 committed by GitHub
parent f6e3e9a456
commit e9364ee279
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -115,6 +115,7 @@ if os.name == "nt":
os.environ['MIMALLOC_PURGE_DELAY'] = '0'
if __name__ == "__main__":
os.environ['TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL'] = '1'
if args.default_device is not None:
default_dev = args.default_device
devices = list(range(32))