in ray-on-gke/tpu/kuberay-tpu-webhook/main.go [171:185]
func containerRequestingTPUs(containers ...corev1.Container) bool {
for _, container := range containers {
if l := container.Resources.Limits; l != nil {
if resource := l[tpuResourceName]; !resource.IsZero() {
return true
}
}
if r := container.Resources.Requests; r != nil {
if resource := r[tpuResourceName]; !resource.IsZero() {
return true
}
}
}
return false
}