def parse_arguments(): parser = argparse.ArgumentParser() parser.add_argument("--repo", help="Repository root directory") parser.add_argument("command", help="Command to run (show|import|export|check|init)") args = parser.parse_args() # Default repo path. if args.repo is None: args.repo = utils.find_git_toplevel() return args
def parse_arguments(): parser = argparse.ArgumentParser() parser.add_argument("--repo", help="Repository root directory") parser.add_argument("--llvm_path", help="Path to the LLVM sources " "(defaults to third_party/llvm-project)", default=None) parser.add_argument("--tensorflow_path", help="Path to the tensorflow sources " "(default to third_party/tensorflow)", default=None) parser.add_argument( "--tensorflow_rev", "--tf_rev", "--tensorflow_commit", "--tf_commit", help=("Update TensorFlow to this rev, or a named option:" f" {COMMIT_OPTIONS}"), default=LATEST_MATCHING_COMMIT_OPTION) parser.add_argument("--mlir_hlo_path", help="Path to the tensorflow sources " "(default to third_party/tensorflow)", default=None) parser.add_argument("--mlir_hlo_rev", "--mlir_hlo_commit", help=("Update mlir-hlo to this rev, or a named option:" f" {COMMIT_OPTIONS}"), default=LATEST_MATCHING_COMMIT_OPTION) parser.add_argument( "--validate", help="Validate that the selected commits all match the LLVM commit", type=utils.str2bool, nargs="?", default=True, ) args = parser.parse_args() # Default repo path. if args.repo is None: args.repo = utils.find_git_toplevel() # Set some defaults. if not args.llvm_path: args.llvm_path = os.path.join(args.repo, "third_party", "llvm-project") if not args.tensorflow_path: args.tensorflow_path = os.path.join(args.repo, "third_party", "tensorflow") if not args.mlir_hlo_path: args.mlir_hlo_path = os.path.join(args.repo, "third_party", "mlir-hlo") return args
def parse_arguments(): parser = argparse.ArgumentParser() parser.add_argument("--repo", help="Repository root directory") parser.add_argument("--llvm", help="Path to the LLVM sources " "(defaults to third_party/llvm-project)", default=None) parser.add_argument("--llvm_bazel", help="Path to the LLVM Bazel BUILD files" "(defaults to third_party/llvm-bazel)", default=None) parser.add_argument( "--llvm_bazel_commit", help=( "Update llvm-bazel to this commit, or a named option:" f" {COMMIT_OPTIONS}." f" {LATEST_MATCHING_COMMIT} and {INTEGRATE_COMMIT} are equivalent" " for this repository."), default=LATEST_MATCHING_COMMIT) parser.add_argument("--tensorflow", help="Path to the tensorflow sources " "(default to third_party/tensorflow)", default=None) parser.add_argument( "--tensorflow_commit", "--tf_commit", help=("Update TensorFlow to this commit, or a named option:" f" {COMMIT_OPTIONS}"), default=LATEST_MATCHING_COMMIT) parser.add_argument( "--validate", help="Validate that the selected commits all match the LLVM commit", type=utils.str2bool, nargs="?", default=True, ) args = parser.parse_args() # Default repo path. if args.repo is None: args.repo = utils.find_git_toplevel() # Set some defaults. if not args.tensorflow: args.tensorflow = os.path.join(args.repo, "third_party", "tensorflow") if not args.llvm: args.llvm = os.path.join(args.repo, "third_party", "llvm-project") if not args.llvm_bazel: args.llvm_bazel = os.path.join(args.repo, "third_party", "llvm-bazel") return args
def parse_arguments(): parser = argparse.ArgumentParser() parser.add_argument("--repo", help="Repository root directory") parser.add_argument("--tensorflow", help="Path to the tensorflow sources " "(default to third_party/tensorflow)", default=None) parser.add_argument("--llvm", help="Path to the LLVM sources " "(defaults to third_party/llvm-project)", default=None) parser.add_argument( "--tensorflow_commit", "--tf_commit", help= f"Update TensorFlow to this commit, or a named option: {COMMIT_OPTIONS}", default=LATEST_MATCHING_COMMIT) parser.add_argument( "--validate", help="Validate that the selected commits all match the LLVM commit", type=utils.str2bool, nargs="?", default=True, ) parser.add_argument( "--update_build_files", help="Updates the IREE LLVM build files from TensorFlow.", type=utils.str2bool, nargs="?", default=True) args = parser.parse_args() # Default repo path. if args.repo is None: args.repo = utils.find_git_toplevel() # Set some defaults. if not args.tensorflow: args.tensorflow = os.path.join(args.repo, "third_party", "tensorflow") if not args.llvm: args.llvm = os.path.join(args.repo, "third_party", "llvm-project") return args
def parse_arguments(): parser = argparse.ArgumentParser() parser.add_argument("--repo", help="Repository root directory") parser.add_argument("--tensorflow", help="Path to the tensorflow sources " "(default to third_party/tensorflow)", default=None) parser.add_argument("--llvm", help="Path to the LLVM sources " "(defaults to third_party/llvm-project)", default=None) parser.add_argument( "--tensorflow_commit", help="Update TensorFlow to this commit (or 'KEEP', 'REMOTE')", default="REMOTE") parser.add_argument( "--llvm_commit", help="Update LLVM to this commit (or 'KEEP', 'REMOTE', 'TENSORFLOW')", default="TENSORFLOW") parser.add_argument( "--update_build_files", help=("Updates the IREE LLVM build files from TensorFlow." "Defaults to True iff llvm_commit==TENSORFLOW"), type=utils.str2bool, nargs="?", default=None) args = parser.parse_args() # Default repo path. if args.repo is None: args.repo = utils.find_git_toplevel() # Set some defaults. if not args.tensorflow: args.tensorflow = os.path.join(args.repo, "third_party", "tensorflow") if not args.llvm: args.llvm = os.path.join(args.repo, "third_party", "llvm-project") return args