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