aws / aws/sagemaker-training-toolkit

Add 'ml.p5.48xlarge' as a supported instance for SM_EFA_NCCL_INSTANCES.

Open
#219 1 comment 0 reactions 0 assignees View on GitHub
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.