448e080c752a1b5f54c422401b41e824b0274a91,tensorflow/python/distribute/multi_process_lib.py,,_set_spawn_exe_path,#,89

Before Change

    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
        break
    if path_to_use is None:

After Change

    path = None
    // If all we have is a python module path, we"ll need to make a guess for
    // the actual executable path.
    if "bazel-out" in sys.argv[0]:
      // Guess the binary path under bazel. For target
      // //tensorflow/python/distribute:input_lib_test_multiworker_gpu, the
      // argv[0] is in the form of
      // /.../org_tensorflow/tensorflow/python/distribute/input_lib_test.py
      // and the binary is
      // /.../org_tensorflow/tensorflow/python/distribute/input_lib_test_multiworker_gpu
      org_tensorflow_path = sys.argv[0][:sys.argv[0].rfind("/tensorflow")]
      if org_tensorflow_path.endswith("/org_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)

Instances