Skip to content

Recreate v6e-1 TPU agents to use ci_v6e module #229

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
QiliangCui merged 5 commits into from
Dec 9, 2025
Merged

Recreate v6e-1 TPU agents to use ci_v6e module #229

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
c55ec0e
Recreate TPU v6e-1 agents to use ci_v6e module
CienetStingLin Dec 3, 2025
b92663a
Merge branch 'main' into sting/recreate_tpu_change_ci_v6e_module
CienetStingLin Dec 8, 2025
da75508
Fix module name
CienetStingLin Dec 8, 2025
84aaf28
Fix missing parameter
CienetStingLin Dec 8, 2025
7519153
Merge branch 'main' into sting/recreate_tpu_change_ci_v6e_module
CienetStingLin Dec 9, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 12 additions & 6 deletions terraform/gcp_old/tpu-inference/cloud-tpu-inference-test/main.tf
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,14 +18,20 @@ data "google_secret_manager_secret_version" "huggingface_token" {
version = "latest"
}

module "ci_v6" {
source = "../modules/ci_v6"
module "ci_v6e_1" {
source = "../modules/ci_v6e"
providers = {
google-beta.us-east5-b = google-beta.us-east5-b
google-beta = google-beta.us-east5-b
}
project_id = var.project_id
buildkite_token_value = data.google_secret_manager_secret_version.buildkite_agent_token_ci_cluster.secret_data
huggingface_token_value = data.google_secret_manager_secret_version.huggingface_token.secret_data

accelerator_type = "v6e-1"
reserved = false
instance_count = 24

@QiliangCui QiliangCui Dec 9, 2025

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

just to confirm the number.

@CienetStingLin CienetStingLin Dec 9, 2025

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

That's correct, we have 24 v6e-1 TPU agents in the cloud-tpu-inference-test project.

buildkite_queue_name = "tpu_v6e_queue"
project_id = var.project_id
project_short_name = var.project_short_name
buildkite_token_value = data.google_secret_manager_secret_version.buildkite_agent_token_ci_cluster.secret_data
huggingface_token_value = data.google_secret_manager_secret_version.huggingface_token.secret_data
}

module "ci_v6e_8" {
Expand Down