o ?e@sdZddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddZ dd Z Gd d d Z e rVGd d d e ejjZGdddejjZeZejZnGdddeZdaddZddZddZddZdS)z"Library for multi-process testing.N)app)logging)testcCsttjdko dtjdvS)z*Returns whether the test is run under OSS.Zbazelr)lensysargvr r o/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/distribute/multi_process_lib.pyis_osssr cCsBddtjD}tr|rdStjdkrtdkrdStjdkS)NcSsg|] }|dr|qS)z--tpu) startswith).0argr r r $sz_is_enabled..F)Linuxwin32)rrr version_infoplatformsystem)Ztpu_argsr r r _is_enabled!s   rcs(eZdZdZfddZddZZS) _AbslProcessz'A process that runs using absl.app.run.cs.tt|j|i|t|d|_|j|_dS)Nrun)superr__init__getattr _run_impl_run_with_abslrselfargskwargs __class__r r r/s  z_AbslProcess.__init__cstfdddS)NcsS)N)r)_r r r 6sz-_AbslProcess._run_with_absl..)rrr&r r&r r5sz_AbslProcess._run_with_absl)__name__ __module__ __qualname____doc__rr __classcell__r r r#r r,s rc@seZdZdZdS)AbslForkServerProcessz^An absl-compatible Forkserver process. Note: Forkserver is not available in windows. N)r(r)r*r+r r r r r-;sr-c@seZdZdZeZdS)AbslForkServerContextZabsl_forkserverN)r(r)r*_namer-Processr r r r r.Bsr.c@seZdZdZddZdS)r0z7A process that skips test (until windows is supported).cOs~~td)NzBTODO(b/150264776): Windows is not supported in MultiProcessRunner.)unittestZSkipTestrr r r rNszProcess.__init__N)r(r)r*r+rr r r r r0Ks r0FcCsttjddr.dd}|d}|s|d}|dur)tdtjdtjtd |tjd<t tjddS) 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.pycSsdtjdvrD|tjdvrFtjddtjd|}tjdddddd}tj|||}t d|t |tj rB|SdSdSdS) Nz bazel-outrZ TEST_TARGET:/rzGuessed test binary path: %s) rrrfindosenvironreplacepathjoinrinfoaccessX_OK)Z package_rootZpackage_root_basebinaryZ possible_pathr r r guess_pathbs  z'_set_spawn_exe_path..guess_pathZorg_tensorflowZ org_kerasNz:Cannot determine binary path. sys.argv[0]=%s os.environ=%szCannot determine binary path) rrendswithrerrorr6r7 RuntimeErrormultiprocessingZ get_contextZset_executable)r?r9r r r _set_spawn_exe_pathWs   rDcCsrdtjddvotjtjddd}|sdStjtjdd}tjddt_t|tddS)zDIf spawned process, run requested spawn task and exit. Else a no-op.z-crNzfrom multiprocessing.r)rrindexr execexit)Z is_spawnedcmdr r r _if_spawn_run_and_exitsrIcCs,dadtjd<trtttdS)z
s2      *