Skip to content

Commit a41062d

Browse files
authored
Adding terraform trigger for cleanup script and spacing out onspot tests (GoogleCloudPlatform#5149)
1 parent 5bbdfc1 commit a41062d

3 files changed

Lines changed: 36 additions & 1 deletion

File tree

tools/cloud-build/provision/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ When prompted for project, use integration test project.
3636
|------|--------|---------|
3737
| <a name="module_daily_image_test_runner_schedule"></a> [daily\_image\_test\_runner\_schedule](#module\_daily\_image\_test\_runner\_schedule) | ./trigger-schedule | n/a |
3838
| <a name="module_daily_project_cleanup_filestore_schedule"></a> [daily\_project\_cleanup\_filestore\_schedule](#module\_daily\_project\_cleanup\_filestore\_schedule) | ./trigger-schedule | n/a |
39+
| <a name="module_daily_project_cleanup_schedule"></a> [daily\_project\_cleanup\_schedule](#module\_daily\_project\_cleanup\_schedule) | ./trigger-schedule | n/a |
3940
| <a name="module_daily_project_cleanup_slurm_schedule"></a> [daily\_project\_cleanup\_slurm\_schedule](#module\_daily\_project\_cleanup\_slurm\_schedule) | ./trigger-schedule | n/a |
4041
| <a name="module_daily_test_schedule"></a> [daily\_test\_schedule](#module\_daily\_test\_schedule) | ./trigger-schedule | n/a |
4142
| <a name="module_weekly_build_dependency_check_schedule"></a> [weekly\_build\_dependency\_check\_schedule](#module\_weekly\_build\_dependency\_check\_schedule) | ./trigger-schedule | n/a |
@@ -44,6 +45,7 @@ When prompted for project, use integration test project.
4445

4546
| Name | Type |
4647
|------|------|
48+
| [google_cloudbuild_trigger.daily_project_cleanup](https://registry.terraform.io/providers/hashicorp/google/latest/docs/resources/cloudbuild_trigger) | resource |
4749
| [google_cloudbuild_trigger.daily_project_cleanup_filestore](https://registry.terraform.io/providers/hashicorp/google/latest/docs/resources/cloudbuild_trigger) | resource |
4850
| [google_cloudbuild_trigger.daily_project_cleanup_slurm](https://registry.terraform.io/providers/hashicorp/google/latest/docs/resources/cloudbuild_trigger) | resource |
4951
| [google_cloudbuild_trigger.daily_test](https://registry.terraform.io/providers/hashicorp/google/latest/docs/resources/cloudbuild_trigger) | resource |

tools/cloud-build/provision/daily-cleanup.tf

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,3 +63,29 @@ module "daily_project_cleanup_slurm_schedule" {
6363
schedule = "0 0 * * MON-FRI"
6464
retry_count = 4
6565
}
66+
67+
resource "google_cloudbuild_trigger" "daily_project_cleanup" {
68+
name = "DAILY-project-cleanup"
69+
description = "A resource cleanup script to run periodically"
70+
tags = [local.notify_chat_tag]
71+
72+
git_file_source {
73+
path = "tools/cloud-build/project-cleanup.yaml"
74+
revision = local.ref_develop
75+
uri = var.repo_uri
76+
repo_type = "GITHUB"
77+
}
78+
79+
source_to_build {
80+
uri = var.repo_uri
81+
ref = local.ref_develop
82+
repo_type = "GITHUB"
83+
}
84+
}
85+
86+
module "daily_project_cleanup_schedule" {
87+
source = "./trigger-schedule"
88+
trigger = google_cloudbuild_trigger.daily_project_cleanup
89+
schedule = "0 22 * * *"
90+
retry_count = 4
91+
}

tools/cloud-build/provision/list_tests.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,12 @@
5454
# (set_of_tests, min_distance)
5555
((
5656
"ml-a4-highgpu-slurm",
57-
"ml-a4-highgpu-onspot-slurm",
5857
"gke-a4"
5958
), 2*60),
59+
((
60+
"ml-a4-highgpu-onspot-slurm",
61+
"gke-a4-onspot"
62+
), 2*60),
6063
((
6164
"ml-a3-ultragpu-onspot-slurm",
6265
"ml-a3-ultragpu-onspot-jbvms",
@@ -72,6 +75,10 @@
7275
"ml-a3-highgpu-slurm",
7376
"gke-a3-highgpu"
7477
), 1*60),
78+
((
79+
"ml-a3-highgpu-onspot-slurm",
80+
"gke-a3-highgpu-onspot"
81+
), 1*60),
7582
]
7683
# TODO:
7784
# * Consider defining constraints (e.g. reservations used) as a tags within tests yamls

0 commit comments

Comments
 (0)