[AArch64] Pass args.instance_type to start_instance

pytorch · Mar 24, 2023 · a007fbc · a007fbc
1 parent 85075c1
commit a007fbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aarch64_linux/build_aarch64_wheel.py b/aarch64_linux/build_aarch64_wheel.py
@@ -743,7 +743,7 @@ def parse_arguments():
             check `~/.ssh/` folder or manually set SSH_KEY_PATH environment variable.""")
 
     # Starting the instance
-    inst = start_instance(key_name, ami=ami)
+    inst = start_instance(key_name, ami=ami, instance_type=args.instance_type)
     instance_name = f'{args.key_name}-{args.os}'
     if args.python_version is not None:
         instance_name += f'-py{args.python_version}'