Current section

Files

Jump to
flyte lib flyteidl plugins kubeflow mpi.pb.ex
Raw

lib/flyteidl/plugins/kubeflow/mpi.pb.ex

defmodule Flyteidl.Plugins.Kubeflow.DistributedMPITrainingTask do
@moduledoc false
use Protobuf, syntax: :proto3, protoc_gen_elixir_version: "0.12.0"
field :worker_replicas, 1,
type: Flyteidl.Plugins.Kubeflow.DistributedMPITrainingReplicaSpec,
json_name: "workerReplicas"
field :launcher_replicas, 2,
type: Flyteidl.Plugins.Kubeflow.DistributedMPITrainingReplicaSpec,
json_name: "launcherReplicas"
field :run_policy, 3, type: Flyteidl.Plugins.Kubeflow.RunPolicy, json_name: "runPolicy"
field :slots, 4, type: :int32
end
defmodule Flyteidl.Plugins.Kubeflow.DistributedMPITrainingReplicaSpec do
@moduledoc false
use Protobuf, syntax: :proto3, protoc_gen_elixir_version: "0.12.0"
field :replicas, 1, type: :int32
field :image, 2, type: :string
field :resources, 3, type: Flyteidl.Core.Resources
field :restart_policy, 4,
type: Flyteidl.Plugins.Kubeflow.RestartPolicy,
json_name: "restartPolicy",
enum: true
field :command, 5, repeated: true, type: :string
end