Skip to content

Commit

Permalink
feat: Add support for c7i-flex, g6e, hpc6a.48xlarge, `hpc6id.32…
Browse files Browse the repository at this point in the history 
…xlarge`, `hpc7a`, and `p5e.48xlarge` instance types
  • Loading branch information
@bryantbiggs
bryantbiggs committed Sep 11, 2024
1 parent 1dc8321 commit 45af2c7
Show file tree
Hide file tree
Showing 2 changed files with 181 additions and 1 deletion.
180 changes: 180 additions & 0 deletions cookiecluster/src/inputs/instance.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1225,6 +1225,51 @@ pub const INSTANCE_TYPES: &[InstanceInfo] = &[
cbr_supported: false,
cpu_arch: "arm64",
},
InstanceInfo {
instance_type: "c7i-flex.2xlarge",
instance_storage_supported: false,
efa_supported: false,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "c7i-flex.4xlarge",
instance_storage_supported: false,
efa_supported: false,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "c7i-flex.8xlarge",
instance_storage_supported: false,
efa_supported: false,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "c7i-flex.large",
instance_storage_supported: false,
efa_supported: false,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "c7i-flex.xlarge",
instance_storage_supported: false,
efa_supported: false,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "c7i.12xlarge",
instance_storage_supported: false,
Expand Down Expand Up @@ -1693,6 +1738,78 @@ pub const INSTANCE_TYPES: &[InstanceInfo] = &[
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.12xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.16xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.24xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.2xlarge",
instance_storage_supported: true,
efa_supported: false,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.48xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.4xlarge",
instance_storage_supported: true,
efa_supported: false,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.8xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "g6e.xlarge",
instance_storage_supported: true,
efa_supported: false,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "gr6.4xlarge",
instance_storage_supported: true,
Expand Down Expand Up @@ -1747,6 +1864,60 @@ pub const INSTANCE_TYPES: &[InstanceInfo] = &[
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc6a.48xlarge",
instance_storage_supported: false,
efa_supported: true,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc6id.32xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc7a.12xlarge",
instance_storage_supported: false,
efa_supported: true,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc7a.24xlarge",
instance_storage_supported: false,
efa_supported: true,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc7a.48xlarge",
instance_storage_supported: false,
efa_supported: true,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc7a.96xlarge",
instance_storage_supported: false,
efa_supported: true,
nvidia_gpu_supported: false,
neuron_supported: false,
cbr_supported: false,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "hpc7g.16xlarge",
instance_storage_supported: false,
Expand Down Expand Up @@ -3682,6 +3853,15 @@ pub const INSTANCE_TYPES: &[InstanceInfo] = &[
cbr_supported: true,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "p5e.48xlarge",
instance_storage_supported: true,
efa_supported: true,
nvidia_gpu_supported: true,
neuron_supported: false,
cbr_supported: true,
cpu_arch: "x86-64",
},
InstanceInfo {
instance_type: "r5.12xlarge",
instance_storage_supported: false,
Expand Down
2 changes: 1 addition & 1 deletion xtask/src/main.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ async fn get_ec2_instances(

/// Collect the EC2 instance details and populate the instances.rs file
async fn update_ec2_instances() -> Result<()> {
let regions = vec!["us-east-1", "us-west-2"];
let regions = vec!["us-east-1", "us-east-2", "us-west-2"];
let mut instances: BTreeMap<String, InstanceInfo> = BTreeMap::new();

// Not GA and requires allow listing
Expand Down

0 comments on commit 45af2c7

Please sign in to comment.