Skip to content

Commit a831347

Browse files
committed
Revert "change target"
This reverts commit 4c20af8.
1 parent 4c20af8 commit a831347

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def dockerfile_common():
8585
df = """
8686
ARG BASE_IMAGE={}
8787
ARG ONNXRUNTIME_VERSION={}
88-
ARG ONNXRUNTIME_REPO=https://github.com/mc-nv/onnxruntime.git
88+
ARG ONNXRUNTIME_REPO=https://github.com/microsoft/onnxruntime
8989
ARG ONNXRUNTIME_BUILD_CONFIG={}
9090
""".format(
9191
FLAGS.triton_container, FLAGS.ort_version, FLAGS.ort_build_config
@@ -280,7 +280,10 @@ def dockerfile_for_linux(output_file):
280280
ARG ONNXRUNTIME_BUILD_CONFIG
281281
282282
# Cherry-pick commit: https://github.com/microsoft/onnxruntime/commit/9dad9af9f9b48c05814d0c2d067d0565e8da6ce8
283-
RUN git clone -b mc-nv/TPRD-1589/cccl-api-fix-preprocessor-vars --recursive ${ONNXRUNTIME_REPO} onnxruntime
283+
RUN git clone -b main --recursive ${ONNXRUNTIME_REPO} onnxruntime \\
284+
&& cd onnxruntime \\
285+
&& git fetch origin refs/pull/25246/head:build-against-latest-upstream \\
286+
&& git checkout build-against-latest-upstream
284287
"""
285288

286289
if FLAGS.onnx_tensorrt_tag != "":

0 commit comments

Comments
 (0)