Turn on TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL by default.

2026-03-12 20:57:47 +08:00 · 2025-10-02 17:53:01 -04:00 · 2025-10-02 17:53:01 -04:00 · 42508ab789
commit 42508ab789
parent f6e3e9a456
1 changed files with 1 additions and 0 deletions
--- a/main.py
+++ b/main.py
@ -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))