Skip to content

Conversation

@kangclzjc
Copy link
Contributor

@kangclzjc kangclzjc commented Nov 28, 2025

What type of PR is this?

Kubernete will support workload API to support gang schedule. We also need to support workload API not only PodGang

What this PR does / why we need it:

Which issue(s) this PR fixes:

Fixes #275

Special notes for your reviewer:

Does this PR introduce a API change?

Yes, we will introduce Workload API based on kubernetes/kubernetes#134564, you can check the API here(https://github.com/kubernetes/kubernetes/blob/61c629cc57649bf0ce2378903027c83939d84109/staging/src/k8s.io/api/scheduling/v1alpha1/generated.proto).


Additional documentation e.g., enhancement proposals, usage docs, etc.:


Signed-off-by: kangclzjc <[email protected]>
@kangclzjc kangclzjc marked this pull request as draft November 28, 2025 05:59
Signed-off-by: kangclzjc <[email protected]>
Signed-off-by: kangclzjc <[email protected]>
Signed-off-by: kangclzjc <[email protected]>
Signed-off-by: kangclzjc <[email protected]>
@kangclzjc
Copy link
Contributor Author

Currently, I can generate Workload like below. These can support basic gang schedule for pcsg, pclq level. However, Workload has a limitation. For example, I defined frontend (pclq), prefill (pcsg-1), and decode (pcsg-2) in PCS. In KAI, we will place these three types into the same pod group, but in different sub-pod groups. In this case, we can ensure that these three types are scheduled together as a gang. However, in Kubernetes 1.35 Workload, sub-pod groups are not yet available—though they will be in the future. So for now, we have this limitation, but as Kubernetes versions evolve, we will be able to place them into sub-pod groups accordingly.
img_v3_02sf_e15b9479-7f63-4711-b192-16f5dd4a2d7g

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add Native Support for Kubernetes Workload API to Enable Gang Scheduling

1 participant