BVhZdZddlZddlZddlZddlZddlZddlmZddlmZddl m Z dZ dZ Gdd Z e rSGd d e ejjZGd d ejj"ZeZej&Zn GddeZdadZdZdZdZy)z"Library for multi-process testing.N)app)logging)testchttjdk\xrdtjdvS)z*Returns whether the test is run under OSS.bazelr)lensysargv^/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/distribute/multi_process_lib.pyis_ossrs' SXX!  6388A; 66r ctjDcgc]}|jds|}}tr|rytjdk(rt j dk(rytj dk7Scc}w)Nz--tpuF)Linuxwin32)r r startswithr version_infoplatformsystem)argtpu_argss r _is_enabledr!sd!XX Ac)@c A( A X( HOO$5$@    Bs A;A;c(eZdZdZfdZdZxZS) _AbslProcessz'A process that runs using absl.app.run.cptt| |i|t|d|_|j |_y)Nrun)superr__init__getattr _run_impl_run_with_abslr)selfargskwargs __class__s rr!z_AbslProcess.__init__/s2 ,&77T5)DN""DHr c6tjfdy)Nc$jS)N)r#)_r%s rz-_AbslProcess._run_with_absl..6sdnn&r )rr)r%s`rr$z_AbslProcess._run_with_absl5sGG &'r )__name__ __module__ __qualname____doc__r!r$ __classcell__)r(s@rrr,s/# (r rceZdZdZy)AbslForkServerProcessz^An absl-compatible Forkserver process. Note: Forkserver is not available in windows. N)r-r.r/r0r r rr3r3;sr r3ceZdZdZeZy)AbslForkServerContextabsl_forkserverN)r-r.r/_namer3Processr r rr5r5Bs E#Gr r5ceZdZdZdZy)r8z7A process that skips test (until windows is supported).c0~~tjd)NzBTODO(b/150264776): Windows is not supported in MultiProcessRunner.)unittestSkipTest)r%r&r's rr!zProcess.__init__Ns     N PPr N)r-r.r/r0r!r r rr8r8Ks APr r8Fctjdjdrkd}|d}|s|d}|Atjdtjdt j td|tjd<tjjtjdy) aSet the path to the executable for spawned processes. This utility searches for the binary the parent process is using, and sets the executable of multiprocessing's context accordingly. Raises: RuntimeError: If the binary path cannot be determined. rz.pycdtjdvr|tjdvrtjddtjdj|}tjdddj ddd}tj j|||}tjd|tj|tjr|Syyy) Nz bazel-outr TEST_TARGET:/rzGuessed test binary path: %s) r r rfindosenvironreplacepathjoinrinfoaccessX_OK) package_rootpackage_root_basebinary possible_paths r guess_pathz'_set_spawn_exe_path..guess_pathbs  #  (C HHQK(H!):):<)HIM*12.66sCC %6 %+-  3]C 99]BGG , )D #r org_tensorflow org_kerasNz:Cannot determine binary path. sys.argv[0]=%s os.environ=%szCannot determine binary path) r r endswithrerrorrDrE RuntimeErrormultiprocessing get_contextset_executable)rPrGs r_set_spawn_exe_pathrYWs XXa[% $ & 'D   $d | mm F ((1+rzz# 7 88CHHQK..sxx{;r cdtjddvxrBtjtjjddzjd}|sytjtjjddz}tjddt_t |tj dy)zDIf spawned process, run requested spawn task and exit. Else a no-op.z-crNzfrom multiprocessing.r)r r indexrexecexit) is_spawnedcmds r_if_spawn_run_and_exitr`s $@-&J'>?    %)*#XXa]#( s)((1+r cdadtjd<trt t t jy)z
ros)  (7 ! ( (=l-55GG$o55GG$*+/  # #'PP'