File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -85,7 +85,7 @@ def dockerfile_common():
8585 df = """
8686ARG BASE_IMAGE={}
8787ARG ONNXRUNTIME_VERSION={}
88- ARG ONNXRUNTIME_REPO=https://github.com/mc-nv /onnxruntime.git
88+ ARG ONNXRUNTIME_REPO=https://github.com/microsoft /onnxruntime
8989ARG 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):
280280ARG 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 != "" :
You can’t perform that action at this time.
0 commit comments