o `۷i @s^ddlmZmZddlmZddlmZddddddZd d Zd e d e d ee fddZ dS))DecimalInvalidOperation)reduce)Optionalv4v5ev5pv6e)ztpu-v4-podsliceztpu-v5-lite-deviceztpu-v5-lite-podslicez tpu-v5p-slicez tpu-v6e-slicec Csbt|tttfr t|Sdddddddddd d }t|}|}d }t|dkr@|dd kr@|d|vr?|d d}|dd }nt|dkrX|d|vrX|d d}|dd }zt|}Wntyltd |w|d urs|S| d r{d}nt|dkrd}ntd||dkrtd||d|vrtd|t||d}|||S)aParse kubernetes canonical form quantity like 200Mi to a decimal number. Supported SI suffixes: base1024: Ki | Mi | Gi | Ti | Pi | Ei base1000: n | u | m | "" | k | M | G | T | P | E See https://github.com/kubernetes/apimachinery/blob/master/pkg/api/resource/quantity.go Args: quantity: string. kubernetes canonical form quantity Returns: Decimal: The parsed quantity as a decimal number Raises: ValueError: On invalid or unknown input ) numKkMGTPENizInvalid number format: {}iiz{} has unknown suffixkir) isinstanceintfloatrstrlenr ValueErrorformatendswith)quantity exponentsnumbersuffixbaseexponentr-[/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/autoscaler/_private/kuberay/utils.pyparse_quantitysT            r/topology acceleratorreturncCsd|r0|r0t|}dd|dD}tdd|}d}|dks#|dkr%d }||}|d |Sd S) aConvert Kubernetes gke-tpu nodeSelectors to TPU accelerator_type for a kuberay TPU worker group. Args: topology: value of the cloud.google.com/gke-tpu-topology Kubernetes nodeSelector, describes the physical topology of the TPU podslice. accelerator: value of the cloud.google.com/gke-tpu-accelerator nodeSelector, the name of the TPU accelerator, e.g. tpu-v4-podslice Returns: A string, accelerator_type, e.g. "v4-8". cSsg|]}t|qSr-)r ).0 chip_countr-r-r. gsz.tpu_node_selectors_to_type..xcSs||S)Nr-)r6yr-r-r.hsz,tpu_node_selectors_to_type..r rrr-N)!gke_tpu_accelerator_to_generationsplitr)r0r1 generationchip_dimensions num_chipsdefault_num_cores_per_chip num_coresr-r-r.tpu_node_selectors_to_typeYs rAN) decimalrr functoolsrtypingrr:r/r"rAr-r-r-r.s   G