We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f6e3e9a commit e9364eeCopy full SHA for e9364ee
main.py
@@ -115,6 +115,7 @@ def execute_script(script_path):
115
os.environ['MIMALLOC_PURGE_DELAY'] = '0'
116
117
if __name__ == "__main__":
118
+ os.environ['TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL'] = '1'
119
if args.default_device is not None:
120
default_dev = args.default_device
121
devices = list(range(32))
0 commit comments