in ray-on-gke/tpu/kuberay-tpu-webhook/main.go [189:208]
func getNumTPUChipsRequested(containers ...corev1.Container) int64 {
tpuLimit := int64(0)
tpuRequest := int64(0)
for _, container := range containers {
if l := container.Resources.Limits; l != nil {
if resource := l[tpuResourceName]; !resource.IsZero() {
tpuLimit = resource.Value()
}
}
if r := container.Resources.Requests; r != nil {
if resource := r[tpuResourceName]; !resource.IsZero() {
tpuRequest = resource.Value()
}
} else {
// default to limit if request is ommitted
tpuRequest = tpuLimit
}
}
return min(tpuLimit, tpuRequest)
}