Skip to content

Commit eadc939

Browse files
authored
Merge branch 'master' into master
2 parents 70ab7e6 + 72c40f4 commit eadc939

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

dlc_developer_config.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,4 +187,4 @@ dlc-pr-tensorflow-2-eia-inference = ""
187187
dlc-pr-vllm = ""
188188

189189
# sglang
190-
dlc-pr-sglang = ""
190+
dlc-pr-sglang = ""

test/test_utils/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,10 @@ def get_ami_id_ssm(region_name, parameter_path):
140140
ami_name_pattern="Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.2.? (Ubuntu 20.04) ????????",
141141
IncludeDeprecated=True,
142142
)
143+
AML2_CPU_ARM64_US_EAST_1 = get_ami_id_boto3(
144+
region_name="us-east-1",
145+
ami_name_pattern="Deep Learning ARM64 Base OSS Nvidia Driver GPU AMI (Amazon Linux 2) ????????",
146+
)
143147
PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_EAST_1 = "ami-0673bb31cc62485dd"
144148
PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_WEST_2 = "ami-02d9a47bc61a31d43"
145149

0 commit comments

Comments
 (0)