Skip to content

Add PriorityClasses to example Kueue configs#5614

Merged
Neelabh94 merged 1 commit into
GoogleCloudPlatform:developfrom
scaliby:kueue-examples
May 8, 2026
Merged

Add PriorityClasses to example Kueue configs#5614
Neelabh94 merged 1 commit into
GoogleCloudPlatform:developfrom
scaliby:kueue-examples

Conversation

@scaliby

@scaliby scaliby commented May 7, 2026

Copy link
Copy Markdown
Contributor

Objective

Align the Cluster Toolkit example Kueue configurations with advanced scheduling requirements (such as XPK's super-slicing setup) by including standard PriorityClasses.

Changes

  • Added very-low, low, medium, high, and very-high PriorityClasses to all example kueue-configuration.yaml.tftpl files in the examples/ directory.

This ensures users starting from our examples have a complete and robust priority setup out of the box.

@scaliby scaliby requested a review from a team as a code owner May 7, 2026 08:18
@gemini-code-assist

Copy link
Copy Markdown
Contributor

Summary of Changes

Hello, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request updates the example Kueue configuration templates to include a standard set of Kubernetes PriorityClasses. By providing these defaults, the changes improve the out-of-the-box experience for users and ensure that example setups are better prepared for advanced scheduling workloads.

Highlights

  • Standardized PriorityClasses: Added a comprehensive set of PriorityClasses (very-low, low, medium, high, very-high) to all example Kueue configuration templates.
  • Configuration Alignment: Ensured that example configurations align with advanced scheduling requirements, such as XPK's super-slicing setup.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize the Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counterproductive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for GitHub and other Google products, sign up here.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

@github-actions github-actions Bot added the external PR from external contributor label May 7, 2026

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces several PriorityClass definitions—ranging from very-low to very-high—across multiple Kueue configuration templates in various GKE examples. Feedback indicates that these definitions are duplicated across many files, reducing maintainability, and that the priority values assigned are inconsistent with existing templates in the repository. It is recommended to centralize these definitions to ensure consistency and follow best practices for code structure.

Comment thread examples/gke-a3-highgpu/kueue-configuration.yaml.tftpl
Comment thread examples/gke-a3-megagpu/kueue-configuration.yaml.tftpl
Comment thread examples/gke-a3-ultragpu/kueue-configuration.yaml.tftpl
Comment thread examples/gke-a4/kueue-configuration.yaml.tftpl
Comment thread examples/gke-a4x-max-bm/kueue-configuration.yaml.tftpl
Comment thread examples/gke-a4x/kueue-configuration.yaml.tftpl
Comment thread examples/gke-tpu-7x/kueue-configuration.yaml.tftpl
Comment thread examples/gke-tpu-v6e/kueue-configuration.yaml.tftpl
@scaliby

scaliby commented May 7, 2026

Copy link
Copy Markdown
Contributor Author

/assign @jamOne-

@Neelabh94

Copy link
Copy Markdown
Contributor

/gcbrun

@Neelabh94 Neelabh94 added enhancement New feature or request release-improvements Added to release notes under the "Improvements" heading. labels May 7, 2026

@SwarnaBharathiMantena SwarnaBharathiMantena left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you please update these files as well: examples/gke-consumption-options/dws-flex-start-queued-provisioning/dws-queues.yaml.tftpl and examples/gke-consumption-options/dws-flex-start-queued-provisioning/tpu-dws-queues.yaml.tftpl

Task: Add PriorityClasses to every example Kueue configs.
@scaliby

scaliby commented May 7, 2026

Copy link
Copy Markdown
Contributor Author

Can you please update these files as well: examples/gke-consumption-options/dws-flex-start-queued-provisioning/dws-queues.yaml.tftpl and examples/gke-consumption-options/dws-flex-start-queued-provisioning/tpu-dws-queues.yaml.tftpl

Done!

@SwarnaBharathiMantena

Copy link
Copy Markdown
Contributor

/gcbrun

@SwarnaBharathiMantena SwarnaBharathiMantena left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@Neelabh94 Neelabh94 merged commit 10778ef into GoogleCloudPlatform:develop May 8, 2026
13 of 82 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request external PR from external contributor release-improvements Added to release notes under the "Improvements" heading.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants