Skip to content
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

Jump to bottom

Ray TPU Webhook Reliability Improvements #723

Merged
merged 61 commits into from
Jul 25, 2024
Merged

Ray TPU Webhook Reliability Improvements #723

merged 61 commits into from
Jul 25, 2024

Conversation

ryanaoleary
Copy link
Collaborator

@ryanaoleary ryanaoleary commented Jul 4, 2024 •
edited

This PR depends on changes from #740 which should be merged first (c4cdf31 marks the start of the changes in this PR).

This PR improves the reliability of the webhook by making it stateless in between calls, fixing issues related to the sliceToWorkers mapping being cleared upon webhook restart. These changes rely on adding a k8s PodInformer to the webhook that watches Pods in the GKE cluster with the ray.io/node-type=worker label. We can then determine the next replicaIndex and TPU_WORKER_ID using the PodInformer cache. These changes remove the need to intercept Pod deletion requests.

This PR has been tested as follows:

  • Unit Tests
  • Manual Tests using single-host, multi-host, multi-slice, and an autoscaling RayCluster with a TPU worker group added, intercepting additional Pods after webhook restart, and multiple webhook replicas with a Kuberay operator restart

@ryanaoleary ryanaoleary self-assigned this Jul 4, 2024
@ryanaoleary ryanaoleary changed the title Ray TPU Webhook Auto-scaling Support and Reliability Improvements Ray TPU Webhook Autoscaling Support and Reliability Improvements Jul 4, 2024
@ryanaoleary ryanaoleary deleted the autoscaling-changes branch July 15, 2024 22:25
@ryanaoleary ryanaoleary restored the autoscaling-changes branch July 15, 2024 23:22
@ryanaoleary ryanaoleary reopened this Jul 15, 2024
@ryanaoleary ryanaoleary changed the title Ray TPU Webhook Autoscaling Support and Reliability Improvements Ray TPU Webhook Reliability Improvements Jul 15, 2024
andrewsykim
andrewsykim reviewed Jul 17, 2024
View reviewed changes
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
andrewsykim
andrewsykim reviewed Jul 18, 2024
View reviewed changes
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
@andrewsykim andrewsykim mentioned this pull request Jul 19, 2024
Merged
2 tasks
@ryanaoleary
Generate hostnames at Pod creation
f22a4a6 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Update tests and add error checking
274ac21 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Make webhook stateless in between mutate calls
f20e6f3 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Formatting changes
02986e7 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Fix bug causing incorrect IDs
c628ca8
@ryanaoleary
Add cluster role and log formatting changes
d9880eb 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Filter pods by Ray worker group label
1cb96ce 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Vulnerability fixes
c2ce875 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Better names and add ServiceAccount
df09e9d 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Change version back to v1.1
7829c6f 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
andrewsykim
andrewsykim reviewed Jul 24, 2024
View reviewed changes
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
ray-on-gke/tpu/kuberay-tpu-webhook/main.go Outdated Show resolved Hide resolved
@ryanaoleary
Fix nits and go vet output
af1993f 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Initial cloudbuil commit
b639368 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Fix vet command
843e7b7 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Update cloudbuild
0d4d0c0 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Fix cloudbuild errors
54c97f1 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Add dir
4149cc5 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Remove arg
7c69b01 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Change to bash command
351051a 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
increase timeout time
adcd5f6 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Fix validateRayCluster test
6614355 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Fix nits for cloudbuild
746f898 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Break early in validateRayCluster
4164563 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Remove unnecessary args from validateRayCluster test
54b8a71 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Change break to continue
fdf4e6c 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
@ryanaoleary
Remove unused vars from webhook tests and add edge cases
78e15a8 
Signed-off-by: Ryan O'Leary <ryanaoleary@google.com>
ryanaoleary and others added 2 commits July 25, 2024 00:56
andrewsykim
andrewsykim approved these changes Jul 25, 2024
View reviewed changes
Copy link
Collaborator

@andrewsykim andrewsykim left a comment

Choose a reason for hiding this comment

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

LGTM!

@ryanaoleary ryanaoleary merged commit 0ae82b1 into GoogleCloudPlatform:main Jul 25, 2024
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@spencer-p spencer-p spencer-p left review comments

@andrewsykim andrewsykim andrewsykim approved these changes

Assignees

@ryanaoleary ryanaoleary

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
3 participants
@ryanaoleary @spencer-p @andrewsykim