saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-03 03:58:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add option to specify onnxruntime repo URL in tools/android_custom_build/build_custom_android_package.py. (#11250)

Browse source
This commit is contained in:
Edward Chen 2022-04-18 19:29:41 -07:00 committed by GitHub
parent efb0928e2b
commit 3dac66698b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tools/android_custom_build/build_custom_android_package.py
View file

@ -29,6 +29,9 @@ def parse_args():
"Supports branches and tags starting from 1.11 (branch rel-1.11.0 or tag v1.11.0). "
"If unspecified, builds the latest.")
parser.add_argument("--onnxruntime_repo_url",
help="The ONNX Runtime repo URL. If unspecified, uses the official repo.")
parser.add_argument("--include_ops_by_config", type=pathlib.Path,
help="The configuration file specifying which ops to include. "
"Such a configuration file is generated during ONNX to ORT format model conversion. "
@ -63,8 +66,11 @@ def parse_args():
def main():
args = parse_args()
docker_build_args = ["--build-arg", f"ONNXRUNTIME_BRANCH_OR_TAG={args.onnxruntime_branch_or_tag}"] \
if args.onnxruntime_branch_or_tag else []
docker_build_args = []
if args.onnxruntime_branch_or_tag:
docker_build_args += ["--build-arg", f"ONNXRUNTIME_BRANCH_OR_TAG={args.onnxruntime_branch_or_tag}"]
if args.onnxruntime_repo_url:
docker_build_args += ["--build-arg", f"ONNXRUNTIME_REPO={args.onnxruntime_repo_url}"]
docker_build_cmd = [args.docker_path, "build",
"--tag", args.docker_image_tag,