Skip to content

Commit e9364ee

Browse files
Turn on TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL by default. (#10168)
1 parent f6e3e9a commit e9364ee

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ def execute_script(script_path):
115115
os.environ['MIMALLOC_PURGE_DELAY'] = '0'
116116

117117
if __name__ == "__main__":
118+
os.environ['TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL'] = '1'
118119
if args.default_device is not None:
119120
default_dev = args.default_device
120121
devices = list(range(32))

0 commit comments

Comments
 (0)