Implements native k8s backend plugin for Kubeflow MPI operator. (flyt…

…eorg#314) * Implements native k8s backend plugin for Kubeflow MPI operator. Signed-off-by: Yuvraj <[email protected]>
eapolinario · Oct 19, 2021 · 64820e6 · 64820e6
1 parent 9a41c03
commit 64820e6
Show file tree

Hide file tree

Showing 3 changed files with 87 additions and 24 deletions.
diff --git a/flytepropeller/cmd/controller/main.go b/flytepropeller/cmd/controller/main.go
@@ -5,6 +5,7 @@ import (
 	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/array/k8s"
 	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/hive"
 	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/container"
+	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/kfoperators/mpi"
 	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch"
 	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/sagemaker"
 	_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/sidecar"

diff --git a/flytepropeller/go.mod b/flytepropeller/go.mod
@@ -6,8 +6,8 @@ require (
 	github.com/DiSiqueira/GoTree v1.0.1-0.20180907134536-53a8e837f295
 	github.com/benlaurie/objecthash v0.0.0-20180202135721-d1e3d6079fc1
 	github.com/fatih/color v1.10.0
-	github.com/flyteorg/flyteidl v0.21.1
-	github.com/flyteorg/flyteplugins v0.7.0
+	github.com/flyteorg/flyteidl v0.21.4
+	github.com/flyteorg/flyteplugins v0.7.1
 	github.com/flyteorg/flytestdlib v0.3.36
 	github.com/ghodss/yaml v1.0.0
 	github.com/go-redis/redis v6.15.7+incompatible
@@ -18,7 +18,7 @@ require (
 	github.com/magiconair/properties v1.8.4
 	github.com/mitchellh/mapstructure v1.4.1
 	github.com/pkg/errors v0.9.1
-	github.com/prometheus/client_golang v1.9.0
+	github.com/prometheus/client_golang v1.10.0
 	github.com/sirupsen/logrus v1.7.0
 	github.com/spf13/cobra v1.1.1
 	github.com/spf13/pflag v1.0.5