Skip to content

Restore the rest of graviton/arm64 pipelines #4550

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sallyseok merged 1 commit into from
Feb 11, 2025
Merged

Restore the rest of graviton/arm64 pipelines #4550

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
176 changes: 88 additions & 88 deletions release_images_patches.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -132,34 +132,34 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
# 17:
# framework: "pytorch"
# version: "2.4.0"
# arch_type: "graviton"
# customer_type: "ec2"
# inference:
# device_types: [ "cpu", "gpu" ]
# python_versions: [ "py311" ]
# os_version: "ubuntu22.04"
# cuda_version: "cu124"
# example: False
# disable_sm_tag: False
# force_release: False
# 18:
# framework: "pytorch"
# version: "2.4.0"
# arch_type: "graviton"
# customer_type: "sagemaker"
# inference:
# device_types: [ "cpu" ]
# python_versions: [ "py311" ]
# os_version: "ubuntu22.04"
# example: False
# disable_sm_tag: False
# force_release: False
11:
framework: "pytorch"
version: "2.4.0"
arch_type: "graviton"
customer_type: "ec2"
inference:
device_types: [ "cpu", "gpu" ]
python_versions: [ "py311" ]
os_version: "ubuntu22.04"
cuda_version: "cu124"
example: False
disable_sm_tag: False
force_release: False
12:
framework: "pytorch"
version: "2.4.0"
arch_type: "graviton"
customer_type: "sagemaker"
inference:
device_types: [ "cpu" ]
python_versions: [ "py311" ]
os_version: "ubuntu22.04"
example: False
disable_sm_tag: False
force_release: False

# PyTorch 2.5 Images
11:
13:
framework: "pytorch"
version: "2.5.1"
customer_type: "ec2"
Expand All @@ -172,7 +172,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
12:
14:
framework: "pytorch"
version: "2.5.1"
customer_type: "sagemaker"
Expand All @@ -185,7 +185,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
13:
15:
framework: "pytorch"
version: "2.5.1"
arch_type: "x86"
Expand All @@ -198,7 +198,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
14:
16:
framework: "pytorch"
version: "2.5.1"
arch_type: "x86"
Expand All @@ -211,34 +211,34 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
# 23:
# framework: "pytorch"
# version: "2.5.1"
# arch_type: "arm64"
# customer_type: "ec2"
# inference:
# device_types: [ "cpu", "gpu" ]
# python_versions: [ "py311" ]
# os_version: "ubuntu22.04"
# cuda_version: "cu124"
# example: False
# disable_sm_tag: False
# force_release: False
# 24:
# framework: "pytorch"
# version: "2.5.1"
# arch_type: "arm64"
# customer_type: "sagemaker"
# inference:
# device_types: [ "cpu" ]
# python_versions: [ "py311" ]
# os_version: "ubuntu22.04"
# example: False
# disable_sm_tag: False
# force_release: False
17:
framework: "pytorch"
version: "2.5.1"
arch_type: "arm64"
customer_type: "ec2"
inference:
device_types: [ "cpu", "gpu" ]
python_versions: [ "py311" ]
os_version: "ubuntu22.04"
cuda_version: "cu124"
example: False
disable_sm_tag: False
force_release: False
18:
framework: "pytorch"
version: "2.5.1"
arch_type: "arm64"
customer_type: "sagemaker"
inference:
device_types: [ "cpu" ]
python_versions: [ "py311" ]
os_version: "ubuntu22.04"
example: False
disable_sm_tag: False
force_release: False

# TensorFlow 2.16 Images
15:
19:
framework: "tensorflow"
version: "2.16.2"
customer_type: "ec2"
Expand All @@ -251,7 +251,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
16:
20:
framework: "tensorflow"
version: "2.16.2"
customer_type: "sagemaker"
Expand All @@ -264,7 +264,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
17:
21:
framework: "tensorflow"
version: "2.16.1"
arch_type: "x86"
Expand All @@ -277,7 +277,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
18:
22:
framework: "tensorflow"
version: "2.16.1"
arch_type: "x86"
Expand All @@ -290,9 +290,35 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
23:
framework: "tensorflow"
version: "2.16.1"
arch_type: "graviton"
customer_type: "ec2"
inference:
device_types: [ "cpu" ]
python_versions: [ "py310" ]
os_version: "ubuntu20.04"
cuda_version: "cu122"
example: False
disable_sm_tag: False
force_release: False
24:
framework: "tensorflow"
version: "2.16.1"
arch_type: "graviton"
customer_type: "sagemaker"
inference:
device_types: [ "cpu" ]
python_versions: [ "py310" ]
os_version: "ubuntu20.04"
cuda_version: "cu122"
example: False
disable_sm_tag: False
force_release: False

# TensorFlow 2.18 Images
19:
25:
framework: "tensorflow"
version: "2.18.0"
customer_type: "ec2"
Expand All @@ -305,7 +331,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
20:
26:
framework: "tensorflow"
version: "2.18.0"
customer_type: "sagemaker"
Expand All @@ -318,7 +344,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
21:
27:
framework: "tensorflow"
version: "2.18.0"
arch_type: "x86"
Expand All @@ -331,7 +357,7 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
22:
28:
framework: "tensorflow"
version: "2.18.0"
arch_type: "x86"
Expand All @@ -344,29 +370,3 @@ release_images:
example: False
disable_sm_tag: False
force_release: False
# 29:
# framework: "tensorflow"
# version: "2.16.1"
# arch_type: "graviton"
# customer_type: "ec2"
# inference:
# device_types: [ "cpu" ]
# python_versions: [ "py310" ]
# os_version: "ubuntu20.04"
# cuda_version: "cu122"
# example: False
# disable_sm_tag: False
# force_release: False
# 30:
# framework: "tensorflow"
# version: "2.16.1"
# arch_type: "graviton"
# customer_type: "sagemaker"
# inference:
# device_types: [ "cpu" ]
# python_versions: [ "py310" ]
# os_version: "ubuntu20.04"
# cuda_version: "cu122"
# example: False
# disable_sm_tag: False
# force_release: False