aws / aws/amazon-ec2-instance-selector

Filter for selecting Trainium Instances

Open
#350 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Go
Stars
932
Forks
110
PR merge metrics
No merged PRs in 30d

Description

Hi,

There seems to be no filter in the ec2-instance-selector to select the AWS Trainium Instances. It would be of great help if a filter for this can be included. Maybe some filter for `NeuronInfo` as in the describe instance types, similar to `--inference-accelerator` would be fine:

```
$ aws ec2 describe-instance-types --instance-type="trn1.32xlarge"
...
"NeuronInfo": {
"NeuronDevices": [
{
"Count": 16,
"Name": "Trainium",
"CoreInfo": {
"Count": 2,
"Version": 2
},
"MemoryInfo": {
"SizeInMiB": 32768
}
}
],
"TotalNeuronDeviceMemoryInMiB": 524288
},
```

Thanks in advance.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.