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

* Add elastic config args to pytorch proto Signed-off-by: Fabio Graetz <[email protected]> * Add elastic config message type for torchrun training Signed-off-by: Fabio Graetz <[email protected]> --------- Signed-off-by: Fabio Graetz <[email protected]> Co-authored-by: Fabio Grätz <[email protected]> Co-authored-by: Ketan Umare <[email protected]> Signed-off-by: Eduardo Apolinario <[email protected]>
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