MultiProcessRunner: Add a convenient _is_enabled() function in multi_process_lib.

This is to make other possible skipping easier.

PiperOrigin-RevId: 337573035
Change-Id: Ifdb489398be87f7459ca17de550afe7b1dc070ed
This commit is contained in:
Rick Chao 2020-10-16 14:00:26 -07:00 committed by TensorFlower Gardener
parent ba58a5c410
commit f701ab5c69

View File

@ -26,6 +26,10 @@ from absl import app
from tensorflow.python.eager import test
def _is_enabled():
return sys.platform != 'win32'
class _AbslProcess:
"""A process that runs using absl.app.run."""
@ -39,7 +43,7 @@ class _AbslProcess:
app.run(lambda _: self._run_impl())
if sys.platform != 'win32':
if _is_enabled():
class AbslForkServerProcess(_AbslProcess,
multiprocessing.context.ForkServerProcess):
@ -142,7 +146,7 @@ def test_main():
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
if sys.platform != 'win32':
if _is_enabled():
_set_spawn_exe_path()
_if_spawn_run_and_exit()