o 1 i@sxddlmZddlmZddlmZedddeefddZedddeefd d Z edddefd d Z d S))Optional)TPUAcceleratorManager) PublicAPIalpha)Z stabilityreturncCst}|dkr d}|S)z Return the name of the TPU pod that the worker is a part of. Returns: The name of the TPU pod. Returns None if not part of a TPU pod. N)rZget_current_node_tpu_name)Ztpu_namere/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/ray/util/accelerators/tpu.pyget_current_pod_namesr cCtS)z Count the number of workers associated with the TPU pod that the worker belongs to. Returns: The total number of workers in the TPU pod. Returns None if the worker is not part of a TPU pod. )rZ"get_num_workers_in_current_tpu_podrrrr get_current_pod_worker_counts r cCr )z Return the number of TPU chips on the node. Returns: The total number of chips on the TPU node. Returns 0 if none are found. )rZ!get_current_node_num_acceleratorsrrrr get_num_tpu_chips_on_node!sr N) typingrZray._private.acceleratorsrZray.util.annotationsrstrr intr r rrrr s