func getNumTPUChipsRequested()

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)
}