From 36d72d7788482edcef91c9184d033e8ab7f7a003 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 5 Dec 2025 16:10:39 -0800 Subject: [PATCH 1/2] fix: Update ONNX Runtime container version. --- build.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/build.py b/build.py index cb535550d3..3ae0878c14 100755 --- a/build.py +++ b/build.py @@ -711,7 +711,7 @@ def onnxruntime_cmake_args(images, library_paths): "onnxruntime", "TRITON_BUILD_CONTAINER_VERSION", None, - FLAGS.triton_container_version, + FLAGS.upstream_container_version, ) ) From f785ba527354705ee501116f86a65d7ecc0055aa Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 5 Dec 2025 16:20:50 -0800 Subject: [PATCH 2/2] fix: update validation logic --- build.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/build.py b/build.py index 3ae0878c14..bd2bb8a7ed 100755 --- a/build.py +++ b/build.py @@ -2880,12 +2880,11 @@ def enable_all(): # backends, repo-agents, and caches if a repo-tag is not given # explicitly. For release branches we use the release branch as # the default, otherwise we use 'main'. - default_repo_tag = "main" - cver = FLAGS.upstream_container_version - if cver is None: - cver = FLAGS.triton_container_version - if not cver.endswith("dev"): - default_repo_tag = "r" + cver + default_repo_tag = ( + "main" + if FLAGS.triton_container_version.endswith("dev") + else "r" + FLAGS.triton_container_version + ) log("default repo-tag: {}".format(default_repo_tag)) # For other versions use the TRITON_VERSION_MAP unless explicitly