Turn on TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL by default. (#10168)

2025-10-02 14:57:15 -07:00 · 2025-10-02 14:57:15 -07:00 · e9364ee279
parent f6e3e9a456
commit e9364ee279
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))