Skip to content
This repository was archived by the owner on Nov 16, 2023. It is now read-only.

Commit 896761a

Browse files
authored
Update HiveD TF example (#55)
1 parent dbcb0d1 commit 896761a

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

example/framework/scenario/tensorflow/gpu/tensorflowdistributedtrainingwithhivedscheduledgpu.yaml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ spec:
4747
# 3 workers are specified with the same containerPort.
4848
# See https://kubernetes.io/docs/concepts/cluster-administration/networking
4949
hostNetwork: false
50+
priorityClassName: prod-priority
5051
containers:
5152
- name: tensorflow
5253
# Using official image to demonstrate this example.
@@ -159,6 +160,7 @@ spec:
159160
# [PREREQUISITE]
160161
# Same as ps TaskRole.
161162
hostNetwork: false
163+
priorityClassName: prod-priority
162164
containers:
163165
- name: tensorflow
164166
image: frameworkcontroller/tensorflow-examples:gpu
@@ -215,3 +217,13 @@ spec:
215217
#nfs:
216218
# server: {NFS Server Host}
217219
# path: {NFS Shared Directory}
220+
---
221+
# Post to {kubeApiServerAddress}/apis/scheduling.k8s.io/v1/priorityclasses/
222+
# PodPriority should be set proportional to above AffinityGroupPriority,
223+
# i.e. the hivedscheduler.microsoft.com/pod-scheduling-spec.priority,
224+
# otherwise, K8S may be incompatible with hived.
225+
apiVersion: scheduling.k8s.io/v1
226+
kind: PriorityClass
227+
metadata:
228+
name: prod-priority
229+
value: 1000

0 commit comments

Comments
 (0)