Merge branch 'master' into dr-support-pip-cm

2026-05-06 02:07:59 +08:00 · 2025-10-03 10:23:40 +09:00
parent 28092933c1 4ffea0e864
commit 47436c59d7
11 changed files with 285 additions and 163 deletions
--- a/main.py
+++ b/main.py
@ -141,6 +141,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))