Skip to content

Commit 3c2f92d

Browse files
authored
Increase instance type size for graviton/arm64 cpu deep canaries (#4941)
* test deep canary * increase instance type size * Restore ['dlc_developer_config.toml'] dlc_developer_config.toml: ('Restore to ' 'https://raw.githubusercontent.com/aws/deep-learning-containers/master/dlc_developer_config.toml')
1 parent 5b76ea9 commit 3c2f92d

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/dlc_tests/ec2/pytorch/inference/test_pytorch_inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,13 @@
4444
)
4545

4646
PT_EC2_CPU_GRAVITON_INSTANCE_TYPE = get_ec2_instance_type(
47-
default="c6g.4xlarge", processor="cpu", arch_type="graviton"
47+
default="c6g.8xlarge", processor="cpu", arch_type="graviton"
4848
)
4949
PT_EC2_GPU_GRAVITON_INSTANCE_TYPE = get_ec2_instance_type(
5050
default="g5g.4xlarge", processor="gpu", arch_type="graviton"
5151
)
5252
PT_EC2_CPU_ARM64_INSTANCE_TYPE = get_ec2_instance_type(
53-
default="c6g.4xlarge", processor="cpu", arch_type="arm64"
53+
default="c6g.8xlarge", processor="cpu", arch_type="arm64"
5454
)
5555
PT_EC2_GPU_ARM64_INSTANCE_TYPE = get_ec2_instance_type(
5656
default="g5g.4xlarge", processor="gpu", arch_type="arm64"

test/dlc_tests/ec2/tensorflow/inference/test_tensorflow_inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,10 @@
3636
filter_function=ec2_utils.filter_only_single_gpu,
3737
)
3838
TF_EC2_GRAVITON_INSTANCE_TYPE = get_ec2_instance_type(
39-
default="c6g.4xlarge", processor="cpu", arch_type="graviton"
39+
default="c6g.8xlarge", processor="cpu", arch_type="graviton"
4040
)
4141
TF_EC2_ARM64_INSTANCE_TYPE = get_ec2_instance_type(
42-
default="c6g.4xlarge", processor="cpu", arch_type="arm64"
42+
default="c6g.8xlarge", processor="cpu", arch_type="arm64"
4343
)
4444

4545

0 commit comments

Comments
 (0)