Vpf0 XddlmZddlmZddlZddlmZGddejZdS)) annotations)clustersN find_specceZdZUdZded<dZded<eddZedd Zedd Z eddZ eddZ dS) Mpi4pyClustermpi4pystrnameTboolopt_in_only_methodreturnc$tdduS)Nr r)clss `/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/clusters/mpi4py_cluster.pyis_env_presentzMpi4pyCluster.is_env_presents X  d ** timeout_secs int | Nonecddlm}|j}|dkr0t j}t |dzdz}|d|}nd}||d}|S)NrMPIii:None)root)r r COMM_WORLDGet_ranksocket gethostnamehashbcast)rrrrhostnameport_ids rget_coordinator_addressz%Mpi4pyCluster.get_coordinator_address#sJ!!%''x..5(,=>**** q11H OrintcZddlm}t|jSNrr)r rr%rGet_sizerrs rget_process_countzMpi4pyCluster.get_process_countE0 s~&&(( ) ))rcZddlm}t|jSr')r rr%rrr)s rget_process_idzMpi4pyCluster.get_process_idJr+rcddlm}|j}||j}t |Sr')r rr Split_typeCOMM_TYPE_SHAREDr%r)rrrnew_comms rget_local_process_idz"Mpi4pyCluster.get_local_process_idOsQ J$$S%9::H x  "" # ##rN)rr )rrrr )rr%)rr) __name__ __module__ __qualname__r __annotations__r classmethodrr$r*r-r2rrrrs$!!!!!+++;+ ;B***;****;* $ $ $; $ $ $rr) __future__rjax._srcrrimportlib.utilr ClusterEnvrr8rrr=s#""""" $$$$$$F$F$F$F$F$H'F$F$F$F$F$r