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 63f6607 commit 7fa4eabCopy full SHA for 7fa4eab
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