de336139770c6e9e035c0e402375efda48d75301,tensorflow/python/distribute/multi_process_lib.py,,_set_spawn_exe_path,#,89

Before Change

    // -> tensorflow/python/some/my_test
    // -> tensorflow/python/my_test
    path_to_use = None
    guess_path = sys.argv[0][:-3]
    guess_path = guess_path.split(os.sep)
    for path_reduction in range(-1, -len(guess_path), -1):
      possible_path = os.sep.join(guess_path[:path_reduction] +
                                  [guess_path[-1]])
      if os.access(possible_path, os.X_OK):
        path_to_use = possible_path
        break
      // The binary can possibly have _gpu suffix.
      possible_path += "_gpu"
      if os.access(possible_path, os.X_OK):
        path_to_use = possible_path

After Change

      // /.../tensorflow/python/distribute/input_lib_test.py
      // and the binary is
      // /.../tensorflow/python/distribute/input_lib_test_multiworker_gpu
      org_tensorflow_path = sys.argv[0][:sys.argv[0].rfind("/tensorflow")]
      binary = os.environ["TEST_TARGET"][2:].replace(":", "/", 1)
      possible_path = os.path.join(org_tensorflow_path, binary)
      logging.info("Guessed test binary path: %s", possible_path)
      if os.access(possible_path, os.X_OK):
        path = possible_path
    if path is None:
      logging.error(
          "Cannot determine binary path. sys.argv[0]=%s os.environ=%s",
          sys.argv[0], os.environ)
      raise RuntimeError("Cannot determine binary path")
    sys.argv[0] = path
  // Note that this sets the executable for *all* contexts.
  multiprocessing.get_context().set_executable(sys.argv[0])

Instances