Feat: Add ElasticConfig message type for torch elastic training (#394)

* Add elastic config args to pytorch proto Signed-off-by: Fabio Graetz <fabiograetz@googlemail.com> * Add elastic config message type for torchrun training Signed-off-by: Fabio Graetz <fabiograetz@googlemail.com> --------- Signed-off-by: Fabio Graetz <fabiograetz@googlemail.com> Co-authored-by: Fabio Grätz <fabiogratz@googlemail.com> Co-authored-by: Ketan Umare <ketan.umare@gmail.com> Signed-off-by: Eduardo Apolinario <eapolinario@users.noreply.github.com>
flyteorg · Sep 13, 2023 · 3d4c196 · 3d4c196
1 parent 8b38649
commit 3d4c196
Show file tree

Hide file tree

Showing 9 changed files with 2,278 additions and 41 deletions.
diff --git a/flyteidl/gen/pb-cpp/flyteidl/plugins/pytorch.pb.cc b/flyteidl/gen/pb-cpp/flyteidl/plugins/pytorch.pb.cc
diff --git a/flyteidl/gen/pb-cpp/flyteidl/plugins/pytorch.pb.h b/flyteidl/gen/pb-cpp/flyteidl/plugins/pytorch.pb.h
diff --git a/flyteidl/gen/pb-go/flyteidl/plugins/pytorch.pb.go b/flyteidl/gen/pb-go/flyteidl/plugins/pytorch.pb.go
diff --git a/flyteidl/gen/pb-go/flyteidl/plugins/pytorch.pb.validate.go b/flyteidl/gen/pb-go/flyteidl/plugins/pytorch.pb.validate.go