Kustomize package for mpi-operator (v1alpha1) (#248)
Signed-off-by: Abhilash Pallerlamudi <stp.abhi@gmail.com>
This commit is contained in:
Abhilash Pallerlamudi
Kubernetes Prow Robot
parent
aa626c1f5d
commit
a648b70175
|
|
@ -0,0 +1,13 @@
|
|||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRoleBinding
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
roleRef:
|
||||
apiGroup: rbac.authorization.k8s.io
|
||||
kind: ClusterRole
|
||||
name: mpi-operator
|
||||
subjects:
|
||||
- kind: ServiceAccount
|
||||
name: mpi-operator
|
||||
|
|
@ -0,0 +1,83 @@
|
|||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRole
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
rules:
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- configmaps
|
||||
- serviceaccounts
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- pods
|
||||
verbs:
|
||||
- get
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- pods/exec
|
||||
verbs:
|
||||
- create
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- events
|
||||
verbs:
|
||||
- create
|
||||
- patch
|
||||
- apiGroups:
|
||||
- rbac.authorization.k8s.io
|
||||
resources:
|
||||
- roles
|
||||
- rolebindings
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- apps
|
||||
resources:
|
||||
- statefulsets
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- update
|
||||
- watch
|
||||
- apiGroups:
|
||||
- batch
|
||||
resources:
|
||||
- jobs
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- update
|
||||
- watch
|
||||
- apiGroups:
|
||||
- policy
|
||||
resources:
|
||||
- poddisruptionbudgets
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- apiextensions.k8s.io
|
||||
resources:
|
||||
- customresourcedefinitions
|
||||
verbs:
|
||||
- create
|
||||
- get
|
||||
- apiGroups:
|
||||
- kubeflow.org
|
||||
resources:
|
||||
- mpijobs
|
||||
verbs:
|
||||
- '*'
|
||||
|
|
@ -0,0 +1,99 @@
|
|||
apiVersion: apiextensions.k8s.io/v1beta1
|
||||
kind: CustomResourceDefinition
|
||||
metadata:
|
||||
name: mpijobs.kubeflow.org
|
||||
spec:
|
||||
group: kubeflow.org
|
||||
version: v1alpha1
|
||||
scope: Namespaced
|
||||
names:
|
||||
plural: mpijobs
|
||||
singular: mpijob
|
||||
kind: MPIJob
|
||||
shortNames:
|
||||
- mj
|
||||
- mpij
|
||||
validation:
|
||||
openAPIV3Schema:
|
||||
properties:
|
||||
spec:
|
||||
title: The MPIJob spec
|
||||
description: Only one of gpus, processingUnits, or replicas should be specified
|
||||
oneOf:
|
||||
- properties:
|
||||
gpus:
|
||||
title: Total number of GPUs
|
||||
description: Valid values are 1, 2, 4, or any multiple of 8
|
||||
oneOf:
|
||||
- type: integer
|
||||
enum:
|
||||
- 1
|
||||
- 2
|
||||
- 4
|
||||
- type: integer
|
||||
multipleOf: 8
|
||||
minimum: 8
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
gpusPerNode:
|
||||
title: The maximum number of GPUs available per node
|
||||
description: Defaults to the number of GPUs per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
required:
|
||||
- gpus
|
||||
- properties:
|
||||
processingUnits:
|
||||
title: Total number of processing units
|
||||
description: Valid values are 1, 2, 4, or any multiple of 8
|
||||
oneOf:
|
||||
- type: integer
|
||||
enum:
|
||||
- 1
|
||||
- 2
|
||||
- 4
|
||||
- type: integer
|
||||
multipleOf: 8
|
||||
minimum: 8
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingUnitsPerNode:
|
||||
title: The maximum number of processing units available per node
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingResourceType:
|
||||
title: The processing resource type, e.g. 'nvidia.com/gpu' or 'cpu'
|
||||
description: Defaults to 'nvidia.com/gpu'
|
||||
type: string
|
||||
enum:
|
||||
- nvidia.com/gpu
|
||||
- cpu
|
||||
required:
|
||||
- processingUnits
|
||||
- properties:
|
||||
replicas:
|
||||
title: Total number of replicas
|
||||
description: The processing resource limit should be specified for each replica
|
||||
type: integer
|
||||
minimum: 1
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingResourceType:
|
||||
title: The processing resource type, e.g. 'nvidia.com/gpu' or 'cpu'
|
||||
description: Defaults to 'nvidia.com/gpu'
|
||||
type: string
|
||||
enum:
|
||||
- nvidia.com/gpu
|
||||
- cpu
|
||||
required:
|
||||
- replicas
|
||||
|
|
@ -0,0 +1,24 @@
|
|||
apiVersion: apps/v1
|
||||
kind: Deployment
|
||||
metadata:
|
||||
name: mpi-operator
|
||||
spec:
|
||||
replicas: 1
|
||||
selector:
|
||||
matchLabels:
|
||||
app: mpi-operator
|
||||
template:
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
spec:
|
||||
containers:
|
||||
- args:
|
||||
- --gpus-per-node
|
||||
- "8"
|
||||
- --kubectl-delivery-image
|
||||
- $(kubectl-delivery-image)
|
||||
image: mpioperator/mpi-operator:0.1.0
|
||||
imagePullPolicy: Always
|
||||
name: mpi-operator
|
||||
serviceAccountName: mpi-operator
|
||||
|
|
@ -0,0 +1,28 @@
|
|||
apiVersion: kustomize.config.k8s.io/v1beta1
|
||||
kind: Kustomization
|
||||
namespace: kubeflow
|
||||
resources:
|
||||
- cluster-role-binding.yaml
|
||||
- cluster-role.yaml
|
||||
- crd.yaml
|
||||
- deployment.yaml
|
||||
- service-account.yaml
|
||||
commonLabels:
|
||||
kustomize.component: mpi-operator
|
||||
images:
|
||||
- name: mpioperator/mpi-operator
|
||||
newName: mpioperator/mpi-operator
|
||||
newTag: 0.1.0
|
||||
configMapGenerator:
|
||||
- name: mpi-operator-config
|
||||
env: params.env
|
||||
generatorOptions:
|
||||
disableNameSuffixHash: true
|
||||
vars:
|
||||
- name: kubectl-delivery-image
|
||||
objref:
|
||||
kind: ConfigMap
|
||||
name: mpi-operator-config
|
||||
apiVersion: v1
|
||||
fieldref:
|
||||
fieldpath: data.kubectl-delivery-image
|
||||
|
|
@ -0,0 +1 @@
|
|||
kubectl-delivery-image=mpioperator/kubectl-delivery:latest
|
||||
|
|
@ -0,0 +1,6 @@
|
|||
apiVersion: v1
|
||||
kind: ServiceAccount
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
apiVersion: app.k8s.io/v1beta1
|
||||
kind: Application
|
||||
metadata:
|
||||
name: mpi-operator
|
||||
spec:
|
||||
selector:
|
||||
matchLabels:
|
||||
app.kubernetes.io/name: mpi-operator
|
||||
app.kubernetes.io/instance: mpi-operator
|
||||
app.kubernetes.io/managed-by: kfctl
|
||||
app.kubernetes.io/component: mpijob
|
||||
app.kubernetes.io/part-of: kubeflow
|
||||
app.kubernetes.io/version: v0.7
|
||||
componentKinds:
|
||||
- group: apps
|
||||
kind: Deployment
|
||||
- group: core
|
||||
kind: ServiceAccount
|
||||
- group: kubeflow.org
|
||||
kind: MPIJob
|
||||
descriptor:
|
||||
type: "mpi-operator"
|
||||
version: "v1alpha1"
|
||||
description: "Mpi-operator allows users to create and manage the \"MPIJob\" custom resource."
|
||||
keywords:
|
||||
- "mpijob"
|
||||
- "mpi-operator"
|
||||
links:
|
||||
- description: About
|
||||
url: "https://github.com/kubeflow/mpi-operator"
|
||||
|
|
@ -0,0 +1,13 @@
|
|||
apiVersion: kustomize.config.k8s.io/v1beta1
|
||||
kind: Kustomization
|
||||
bases:
|
||||
- ../../base
|
||||
resources:
|
||||
- application.yaml
|
||||
commonLabels:
|
||||
app.kubernetes.io/name: mpi-operator
|
||||
app.kubernetes.io/instance: mpi-operator
|
||||
app.kubernetes.io/managed-by: kfctl
|
||||
app.kubernetes.io/component: mpijob
|
||||
app.kubernetes.io/part-of: kubeflow
|
||||
app.kubernetes.io/version: v0.7
|
||||
|
|
@ -0,0 +1,309 @@
|
|||
package tests_test
|
||||
|
||||
import (
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kunstruct"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/transformer"
|
||||
"sigs.k8s.io/kustomize/pkg/fs"
|
||||
"sigs.k8s.io/kustomize/pkg/loader"
|
||||
"sigs.k8s.io/kustomize/pkg/resmap"
|
||||
"sigs.k8s.io/kustomize/pkg/resource"
|
||||
"sigs.k8s.io/kustomize/pkg/target"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func writeMpiOperatorBase(th *KustTestHarness) {
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/cluster-role-binding.yaml", `
|
||||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRoleBinding
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
roleRef:
|
||||
apiGroup: rbac.authorization.k8s.io
|
||||
kind: ClusterRole
|
||||
name: mpi-operator
|
||||
subjects:
|
||||
- kind: ServiceAccount
|
||||
name: mpi-operator
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/cluster-role.yaml", `
|
||||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRole
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
rules:
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- configmaps
|
||||
- serviceaccounts
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- pods
|
||||
verbs:
|
||||
- get
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- pods/exec
|
||||
verbs:
|
||||
- create
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- events
|
||||
verbs:
|
||||
- create
|
||||
- patch
|
||||
- apiGroups:
|
||||
- rbac.authorization.k8s.io
|
||||
resources:
|
||||
- roles
|
||||
- rolebindings
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- apps
|
||||
resources:
|
||||
- statefulsets
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- update
|
||||
- watch
|
||||
- apiGroups:
|
||||
- batch
|
||||
resources:
|
||||
- jobs
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- update
|
||||
- watch
|
||||
- apiGroups:
|
||||
- policy
|
||||
resources:
|
||||
- poddisruptionbudgets
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- apiextensions.k8s.io
|
||||
resources:
|
||||
- customresourcedefinitions
|
||||
verbs:
|
||||
- create
|
||||
- get
|
||||
- apiGroups:
|
||||
- kubeflow.org
|
||||
resources:
|
||||
- mpijobs
|
||||
verbs:
|
||||
- '*'
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/crd.yaml", `
|
||||
apiVersion: apiextensions.k8s.io/v1beta1
|
||||
kind: CustomResourceDefinition
|
||||
metadata:
|
||||
name: mpijobs.kubeflow.org
|
||||
spec:
|
||||
group: kubeflow.org
|
||||
version: v1alpha1
|
||||
scope: Namespaced
|
||||
names:
|
||||
plural: mpijobs
|
||||
singular: mpijob
|
||||
kind: MPIJob
|
||||
shortNames:
|
||||
- mj
|
||||
- mpij
|
||||
validation:
|
||||
openAPIV3Schema:
|
||||
properties:
|
||||
spec:
|
||||
title: The MPIJob spec
|
||||
description: Only one of gpus, processingUnits, or replicas should be specified
|
||||
oneOf:
|
||||
- properties:
|
||||
gpus:
|
||||
title: Total number of GPUs
|
||||
description: Valid values are 1, 2, 4, or any multiple of 8
|
||||
oneOf:
|
||||
- type: integer
|
||||
enum:
|
||||
- 1
|
||||
- 2
|
||||
- 4
|
||||
- type: integer
|
||||
multipleOf: 8
|
||||
minimum: 8
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
gpusPerNode:
|
||||
title: The maximum number of GPUs available per node
|
||||
description: Defaults to the number of GPUs per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
required:
|
||||
- gpus
|
||||
- properties:
|
||||
processingUnits:
|
||||
title: Total number of processing units
|
||||
description: Valid values are 1, 2, 4, or any multiple of 8
|
||||
oneOf:
|
||||
- type: integer
|
||||
enum:
|
||||
- 1
|
||||
- 2
|
||||
- 4
|
||||
- type: integer
|
||||
multipleOf: 8
|
||||
minimum: 8
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingUnitsPerNode:
|
||||
title: The maximum number of processing units available per node
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingResourceType:
|
||||
title: The processing resource type, e.g. 'nvidia.com/gpu' or 'cpu'
|
||||
description: Defaults to 'nvidia.com/gpu'
|
||||
type: string
|
||||
enum:
|
||||
- nvidia.com/gpu
|
||||
- cpu
|
||||
required:
|
||||
- processingUnits
|
||||
- properties:
|
||||
replicas:
|
||||
title: Total number of replicas
|
||||
description: The processing resource limit should be specified for each replica
|
||||
type: integer
|
||||
minimum: 1
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingResourceType:
|
||||
title: The processing resource type, e.g. 'nvidia.com/gpu' or 'cpu'
|
||||
description: Defaults to 'nvidia.com/gpu'
|
||||
type: string
|
||||
enum:
|
||||
- nvidia.com/gpu
|
||||
- cpu
|
||||
required:
|
||||
- replicas
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/deployment.yaml", `
|
||||
apiVersion: apps/v1
|
||||
kind: Deployment
|
||||
metadata:
|
||||
name: mpi-operator
|
||||
spec:
|
||||
replicas: 1
|
||||
selector:
|
||||
matchLabels:
|
||||
app: mpi-operator
|
||||
template:
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
spec:
|
||||
containers:
|
||||
- args:
|
||||
- --gpus-per-node
|
||||
- "8"
|
||||
- --kubectl-delivery-image
|
||||
- $(kubectl-delivery-image)
|
||||
image: mpioperator/mpi-operator:0.1.0
|
||||
imagePullPolicy: Always
|
||||
name: mpi-operator
|
||||
serviceAccountName: mpi-operator
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/service-account.yaml", `
|
||||
apiVersion: v1
|
||||
kind: ServiceAccount
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/params.env", `
|
||||
kubectl-delivery-image=mpioperator/kubectl-delivery:latest
|
||||
`)
|
||||
th.writeK("/manifests/mpi-job/mpi-operator/base", `
|
||||
apiVersion: kustomize.config.k8s.io/v1beta1
|
||||
kind: Kustomization
|
||||
namespace: kubeflow
|
||||
resources:
|
||||
- cluster-role-binding.yaml
|
||||
- cluster-role.yaml
|
||||
- crd.yaml
|
||||
- deployment.yaml
|
||||
- service-account.yaml
|
||||
commonLabels:
|
||||
kustomize.component: mpi-operator
|
||||
images:
|
||||
- name: mpioperator/mpi-operator
|
||||
newName: mpioperator/mpi-operator
|
||||
newTag: 0.1.0
|
||||
configMapGenerator:
|
||||
- name: mpi-operator-config
|
||||
env: params.env
|
||||
generatorOptions:
|
||||
disableNameSuffixHash: true
|
||||
vars:
|
||||
- name: kubectl-delivery-image
|
||||
objref:
|
||||
kind: ConfigMap
|
||||
name: mpi-operator-config
|
||||
apiVersion: v1
|
||||
fieldref:
|
||||
fieldpath: data.kubectl-delivery-image
|
||||
`)
|
||||
}
|
||||
|
||||
func TestMpiOperatorBase(t *testing.T) {
|
||||
th := NewKustTestHarness(t, "/manifests/mpi-job/mpi-operator/base")
|
||||
writeMpiOperatorBase(th)
|
||||
m, err := th.makeKustTarget().MakeCustomizedResMap()
|
||||
if err != nil {
|
||||
t.Fatalf("Err: %v", err)
|
||||
}
|
||||
targetPath := "../mpi-job/mpi-operator/base"
|
||||
fsys := fs.MakeRealFS()
|
||||
_loader, loaderErr := loader.NewLoader(targetPath, fsys)
|
||||
if loaderErr != nil {
|
||||
t.Fatalf("could not load kustomize loader: %v", loaderErr)
|
||||
}
|
||||
rf := resmap.NewFactory(resource.NewFactory(kunstruct.NewKunstructuredFactoryImpl()))
|
||||
kt, err := target.NewKustTarget(_loader, rf, transformer.NewFactoryImpl())
|
||||
if err != nil {
|
||||
th.t.Fatalf("Unexpected construction error %v", err)
|
||||
}
|
||||
n, err := kt.MakeCustomizedResMap()
|
||||
if err != nil {
|
||||
t.Fatalf("Err: %v", err)
|
||||
}
|
||||
expected, err := n.EncodeAsYaml()
|
||||
th.assertActualEqualsExpected(m, string(expected))
|
||||
}
|
||||
|
|
@ -0,0 +1,356 @@
|
|||
package tests_test
|
||||
|
||||
import (
|
||||
"sigs.k8s.io/kustomize/k8sdeps/kunstruct"
|
||||
"sigs.k8s.io/kustomize/k8sdeps/transformer"
|
||||
"sigs.k8s.io/kustomize/pkg/fs"
|
||||
"sigs.k8s.io/kustomize/pkg/loader"
|
||||
"sigs.k8s.io/kustomize/pkg/resmap"
|
||||
"sigs.k8s.io/kustomize/pkg/resource"
|
||||
"sigs.k8s.io/kustomize/pkg/target"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func writeMpiOperatorOverlaysApplication(th *KustTestHarness) {
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/overlays/application/application.yaml", `
|
||||
apiVersion: app.k8s.io/v1beta1
|
||||
kind: Application
|
||||
metadata:
|
||||
name: mpi-operator
|
||||
spec:
|
||||
selector:
|
||||
matchLabels:
|
||||
app.kubernetes.io/name: mpi-operator
|
||||
app.kubernetes.io/instance: mpi-operator
|
||||
app.kubernetes.io/managed-by: kfctl
|
||||
app.kubernetes.io/component: mpijob
|
||||
app.kubernetes.io/part-of: kubeflow
|
||||
app.kubernetes.io/version: v0.7
|
||||
componentKinds:
|
||||
- group: apps
|
||||
kind: Deployment
|
||||
- group: core
|
||||
kind: ServiceAccount
|
||||
- group: kubeflow.org
|
||||
kind: MPIJob
|
||||
descriptor:
|
||||
type: "mpi-operator"
|
||||
version: "v1alpha1"
|
||||
description: "Mpi-operator allows users to create and manage the \"MPIJob\" custom resource."
|
||||
keywords:
|
||||
- "mpijob"
|
||||
- "mpi-operator"
|
||||
links:
|
||||
- description: About
|
||||
url: "https://github.com/kubeflow/mpi-operator"
|
||||
`)
|
||||
th.writeK("/manifests/mpi-job/mpi-operator/overlays/application", `
|
||||
apiVersion: kustomize.config.k8s.io/v1beta1
|
||||
kind: Kustomization
|
||||
bases:
|
||||
- ../../base
|
||||
resources:
|
||||
- application.yaml
|
||||
commonLabels:
|
||||
app.kubernetes.io/name: mpi-operator
|
||||
app.kubernetes.io/instance: mpi-operator
|
||||
app.kubernetes.io/managed-by: kfctl
|
||||
app.kubernetes.io/component: mpijob
|
||||
app.kubernetes.io/part-of: kubeflow
|
||||
app.kubernetes.io/version: v0.7
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/cluster-role-binding.yaml", `
|
||||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRoleBinding
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
roleRef:
|
||||
apiGroup: rbac.authorization.k8s.io
|
||||
kind: ClusterRole
|
||||
name: mpi-operator
|
||||
subjects:
|
||||
- kind: ServiceAccount
|
||||
name: mpi-operator
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/cluster-role.yaml", `
|
||||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRole
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
rules:
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- configmaps
|
||||
- serviceaccounts
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- pods
|
||||
verbs:
|
||||
- get
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- pods/exec
|
||||
verbs:
|
||||
- create
|
||||
- apiGroups:
|
||||
- ""
|
||||
resources:
|
||||
- events
|
||||
verbs:
|
||||
- create
|
||||
- patch
|
||||
- apiGroups:
|
||||
- rbac.authorization.k8s.io
|
||||
resources:
|
||||
- roles
|
||||
- rolebindings
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- apps
|
||||
resources:
|
||||
- statefulsets
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- update
|
||||
- watch
|
||||
- apiGroups:
|
||||
- batch
|
||||
resources:
|
||||
- jobs
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- update
|
||||
- watch
|
||||
- apiGroups:
|
||||
- policy
|
||||
resources:
|
||||
- poddisruptionbudgets
|
||||
verbs:
|
||||
- create
|
||||
- list
|
||||
- watch
|
||||
- apiGroups:
|
||||
- apiextensions.k8s.io
|
||||
resources:
|
||||
- customresourcedefinitions
|
||||
verbs:
|
||||
- create
|
||||
- get
|
||||
- apiGroups:
|
||||
- kubeflow.org
|
||||
resources:
|
||||
- mpijobs
|
||||
verbs:
|
||||
- '*'
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/crd.yaml", `
|
||||
apiVersion: apiextensions.k8s.io/v1beta1
|
||||
kind: CustomResourceDefinition
|
||||
metadata:
|
||||
name: mpijobs.kubeflow.org
|
||||
spec:
|
||||
group: kubeflow.org
|
||||
version: v1alpha1
|
||||
scope: Namespaced
|
||||
names:
|
||||
plural: mpijobs
|
||||
singular: mpijob
|
||||
kind: MPIJob
|
||||
shortNames:
|
||||
- mj
|
||||
- mpij
|
||||
validation:
|
||||
openAPIV3Schema:
|
||||
properties:
|
||||
spec:
|
||||
title: The MPIJob spec
|
||||
description: Only one of gpus, processingUnits, or replicas should be specified
|
||||
oneOf:
|
||||
- properties:
|
||||
gpus:
|
||||
title: Total number of GPUs
|
||||
description: Valid values are 1, 2, 4, or any multiple of 8
|
||||
oneOf:
|
||||
- type: integer
|
||||
enum:
|
||||
- 1
|
||||
- 2
|
||||
- 4
|
||||
- type: integer
|
||||
multipleOf: 8
|
||||
minimum: 8
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
gpusPerNode:
|
||||
title: The maximum number of GPUs available per node
|
||||
description: Defaults to the number of GPUs per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
required:
|
||||
- gpus
|
||||
- properties:
|
||||
processingUnits:
|
||||
title: Total number of processing units
|
||||
description: Valid values are 1, 2, 4, or any multiple of 8
|
||||
oneOf:
|
||||
- type: integer
|
||||
enum:
|
||||
- 1
|
||||
- 2
|
||||
- 4
|
||||
- type: integer
|
||||
multipleOf: 8
|
||||
minimum: 8
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingUnitsPerNode:
|
||||
title: The maximum number of processing units available per node
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingResourceType:
|
||||
title: The processing resource type, e.g. 'nvidia.com/gpu' or 'cpu'
|
||||
description: Defaults to 'nvidia.com/gpu'
|
||||
type: string
|
||||
enum:
|
||||
- nvidia.com/gpu
|
||||
- cpu
|
||||
required:
|
||||
- processingUnits
|
||||
- properties:
|
||||
replicas:
|
||||
title: Total number of replicas
|
||||
description: The processing resource limit should be specified for each replica
|
||||
type: integer
|
||||
minimum: 1
|
||||
slotsPerWorker:
|
||||
title: The number of slots per worker used in hostfile
|
||||
description: Defaults to the number of processing units per worker
|
||||
type: integer
|
||||
minimum: 1
|
||||
processingResourceType:
|
||||
title: The processing resource type, e.g. 'nvidia.com/gpu' or 'cpu'
|
||||
description: Defaults to 'nvidia.com/gpu'
|
||||
type: string
|
||||
enum:
|
||||
- nvidia.com/gpu
|
||||
- cpu
|
||||
required:
|
||||
- replicas
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/deployment.yaml", `
|
||||
apiVersion: apps/v1
|
||||
kind: Deployment
|
||||
metadata:
|
||||
name: mpi-operator
|
||||
spec:
|
||||
replicas: 1
|
||||
selector:
|
||||
matchLabels:
|
||||
app: mpi-operator
|
||||
template:
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
spec:
|
||||
containers:
|
||||
- args:
|
||||
- --gpus-per-node
|
||||
- "8"
|
||||
- --kubectl-delivery-image
|
||||
- $(kubectl-delivery-image)
|
||||
image: mpioperator/mpi-operator:0.1.0
|
||||
imagePullPolicy: Always
|
||||
name: mpi-operator
|
||||
serviceAccountName: mpi-operator
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/service-account.yaml", `
|
||||
apiVersion: v1
|
||||
kind: ServiceAccount
|
||||
metadata:
|
||||
labels:
|
||||
app: mpi-operator
|
||||
name: mpi-operator
|
||||
`)
|
||||
th.writeF("/manifests/mpi-job/mpi-operator/base/params.env", `
|
||||
kubectl-delivery-image=mpioperator/kubectl-delivery:latest
|
||||
`)
|
||||
th.writeK("/manifests/mpi-job/mpi-operator/base", `
|
||||
apiVersion: kustomize.config.k8s.io/v1beta1
|
||||
kind: Kustomization
|
||||
namespace: kubeflow
|
||||
resources:
|
||||
- cluster-role-binding.yaml
|
||||
- cluster-role.yaml
|
||||
- crd.yaml
|
||||
- deployment.yaml
|
||||
- service-account.yaml
|
||||
commonLabels:
|
||||
kustomize.component: mpi-operator
|
||||
images:
|
||||
- name: mpioperator/mpi-operator
|
||||
newName: mpioperator/mpi-operator
|
||||
newTag: 0.1.0
|
||||
configMapGenerator:
|
||||
- name: mpi-operator-config
|
||||
env: params.env
|
||||
generatorOptions:
|
||||
disableNameSuffixHash: true
|
||||
vars:
|
||||
- name: kubectl-delivery-image
|
||||
objref:
|
||||
kind: ConfigMap
|
||||
name: mpi-operator-config
|
||||
apiVersion: v1
|
||||
fieldref:
|
||||
fieldpath: data.kubectl-delivery-image
|
||||
`)
|
||||
}
|
||||
|
||||
func TestMpiOperatorOverlaysApplication(t *testing.T) {
|
||||
th := NewKustTestHarness(t, "/manifests/mpi-job/mpi-operator/overlays/application")
|
||||
writeMpiOperatorOverlaysApplication(th)
|
||||
m, err := th.makeKustTarget().MakeCustomizedResMap()
|
||||
if err != nil {
|
||||
t.Fatalf("Err: %v", err)
|
||||
}
|
||||
targetPath := "../mpi-job/mpi-operator/overlays/application"
|
||||
fsys := fs.MakeRealFS()
|
||||
_loader, loaderErr := loader.NewLoader(targetPath, fsys)
|
||||
if loaderErr != nil {
|
||||
t.Fatalf("could not load kustomize loader: %v", loaderErr)
|
||||
}
|
||||
rf := resmap.NewFactory(resource.NewFactory(kunstruct.NewKunstructuredFactoryImpl()))
|
||||
kt, err := target.NewKustTarget(_loader, rf, transformer.NewFactoryImpl())
|
||||
if err != nil {
|
||||
th.t.Fatalf("Unexpected construction error %v", err)
|
||||
}
|
||||
n, err := kt.MakeCustomizedResMap()
|
||||
if err != nil {
|
||||
t.Fatalf("Err: %v", err)
|
||||
}
|
||||
expected, err := n.EncodeAsYaml()
|
||||
th.assertActualEqualsExpected(m, string(expected))
|
||||
}
|
||||
Loading…
Reference in New Issue