aws / aws/sagemaker-training-toolkit
Add 'ml.p5.48xlarge' as a supported instance for SM_EFA_NCCL_INSTANCES.
- Dominant language
- Python
- Stars
- 530
- Forks
- 140
- Avg merge
- 1h 12m
- Merged PRs (30d)
- 2
Description
**Describe the feature you'd like**
Add "ml.p5.48xlarge" to the list of SM_EFA_NCCL_INSTANCES. This means SageMaker Training Toolkit sets FI_PROVIDER=efa automatically for use in training jobs to ensure EFA is used.
**How would this feature be used? Please describe.**
Automatically setting EFA for use in SageMaker Training Jobs that leverage the SageMaker Training Toolkit
**Describe alternatives you've considered**
Unsure. Currently you would manually set the environment variable.
**Additional context**
NA
Contributor guide
Research direction
Search the repository for SM_EFA_NCCL_INSTANCES and inspect the surrounding SageMaker Training Toolkit configuration. Confirm how the list controls automatic FI_PROVIDER=efa behavior, then verify that ml.p5.48xlarge is included and that the existing tests or checks pass.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- aws, python
- Domain
- cloud, machine-learning
- Issue type
- Feature
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 48/100