Onboard PT 2.6 ARM64 Inference EC2 to autopatch (#4621)

sallyseok · web-flow · commit 7901f6be2fe8 · 2025-03-10T11:59:46.000-07:00
* turn on autopatch and test

* revert temp changes
diff --git a/pytorch/inference/buildspec-arm64-2-6-ec2.yml b/pytorch/inference/buildspec-arm64-2-6-ec2.yml
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
 version: &VERSION 2.6.0
 short_version: &SHORT_VERSION "2.6"
 arch_type: arm64
-# autopatch_build: "True"
+autopatch_build: "True"
 
 repository_info:
   inference_repository: &INFERENCE_REPOSITORY
diff --git a/test/test_utils/__init__.py b/test/test_utils/__init__.py
@@ -58,6 +58,8 @@ def get_ami_id_boto3(region_name, ami_name_pattern, IncludeDeprecated=False):
         Owners=["amazon"],
         IncludeDeprecated=IncludeDeprecated,
     )
+    if not ami_list["Images"]:
+        raise RuntimeError(f"Unable to find AMI with pattern {ami_name_pattern}")
 
     # NOTE: Hotfix for fetching latest DLAMI before certain creation date.
     # replace `ami_list["Images"]` with `filtered_images` in max() if needed.

Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,8 @@ def get_ami_id_boto3(region_name, ami_name_pattern, IncludeDeprecated=False):`
`58`	`58`	`Owners=["amazon"],`
`59`	`59`	`IncludeDeprecated=IncludeDeprecated,`
`60`	`60`	`)`
	`61`	`+ if not ami_list["Images"]:`
	`62`	`+ raise RuntimeError(f"Unable to find AMI with pattern {ami_name_pattern}")`
`61`	`63`
`62`	`64`	`# NOTE: Hotfix for fetching latest DLAMI before certain creation date.`
`63`	`65`	# replace `ami_list["Images"]` with `filtered_images` in max() if needed.