Skip to content

Add integration test files for TPU v6e#4906

Merged
agrawalkhushi18 merged 1 commit into
GoogleCloudPlatform:developfrom
agrawalkhushi18:integrationTest-v6e
Dec 3, 2025
Merged

Add integration test files for TPU v6e#4906
agrawalkhushi18 merged 1 commit into
GoogleCloudPlatform:developfrom
agrawalkhushi18:integrationTest-v6e

Conversation

@agrawalkhushi18

Copy link
Copy Markdown
Contributor

This PR adds a build file and a test that verifies TPU v6e cluster provisioning, runs a simple JAX job, and tears down the cluster. The test uses a single-host, single-slice (2x2) topology and invokes the jax command to exercise the device count. I validated the change by running the test locally — it completed successfully.

Submission Checklist

NOTE: Community submissions can take up to 2 weeks to be reviewed.

Please take the following actions before submitting this pull request.

  • Fork your PR branch from the Toolkit "develop" branch (not main)
  • Test all changes with pre-commit in a local branch #
  • Confirm that "make tests" passes all tests
  • Add or modify unit tests to cover code changes
  • Ensure that unit test coverage remains above 80%
  • Update all applicable documentation
  • Follow Cluster Toolkit Contribution guidelines #

@agrawalkhushi18 agrawalkhushi18 added release-key-new-features Added to release notes under the "Key New Features" heading. release-improvements Added to release notes under the "Improvements" heading. labels Nov 28, 2025
@agrawalkhushi18 agrawalkhushi18 force-pushed the integrationTest-v6e branch 2 times, most recently from 623e48e to 8513b65 Compare November 28, 2025 09:58
@agrawalkhushi18 agrawalkhushi18 marked this pull request as ready for review November 28, 2025 10:32
@agrawalkhushi18 agrawalkhushi18 requested review from a team and samskillman as code owners November 28, 2025 10:32
LAVEEN
LAVEEN previously approved these changes Nov 28, 2025
@agrawalkhushi18 agrawalkhushi18 marked this pull request as draft November 30, 2025 13:03
@agrawalkhushi18 agrawalkhushi18 marked this pull request as ready for review November 30, 2025 14:09
@agrawalkhushi18 agrawalkhushi18 force-pushed the integrationTest-v6e branch 2 times, most recently from fc3b032 to f4a85fc Compare December 2, 2025 09:30
Comment thread tools/cloud-build/daily-tests/tests/gke-tpu-v6e.yml

@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

@agrawalkhushi18 agrawalkhushi18 merged commit a02d488 into GoogleCloudPlatform:develop Dec 3, 2025
12 of 68 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

release-improvements Added to release notes under the "Improvements" heading. release-key-new-features Added to release notes under the "Key New Features" heading.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants