Join Slack
Powered by
<#394 Feat: Add `ElasticConfig` message type for t...
# flyte-github
a
acoustic-carpenter-78188
04/23/2023, 10:55 AM
#394 Feat: Add `ElasticConfig` message type for torch elastic training
Pull request opened by
fg91
TL;DR
This PR adds a custom proto for a torch elastic config for distributed training using
torchrun
. The elastic configuration set by the user in the task decorator is converted to the
`ElasticPolicy` in kubeflow's `PytorchJob`
. See
flyteorg/flyte#3614
issue for more details.
Type
☐ Bug Fix ☑︎ Feature ☐ Plugin
Are all requirements met?
☑︎ Code completed ☑︎ Smoke tested ☐ Unit tests added ☐ Code documentation added ☑︎ Any pending items have an associated Issue
Complete description
Tracking Issue
flyteorg/flyte#3614
Follow-up issue
flyteorg/flyteidl
✅ All checks have passed 13/13 successful checks
acoustic-carpenter-78188
04/23/2023, 10:55 AM
#394 Feat: Add `ElasticConfig` message type for torch elastic training
Pull request ready for review by
fg91
flyteorg/flyteidl
acoustic-carpenter-78188
04/24/2023, 2:42 AM
#394 Feat: Add `ElasticConfig` message type for torch elastic training
Pull request merged by
eapolinario
flyteorg/flyteidl
Open in Slack
Previous
Next