From 42508ab789ad75ada46c9583b949431e00c09a0a Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 2 Oct 2025 17:53:01 -0400
Subject: [PATCH] Turn on TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL by default.

---
 main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/main.py b/main.py
index 70696fcc3..35857dba8 100644
--- 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))