change target

Author: mc-nv

tools/gen_ort_dockerfile.py

@@ -85,7 +85,7 @@ def dockerfile_common():
     df = """
 ARG BASE_IMAGE={}
 ARG ONNXRUNTIME_VERSION={}
-ARG ONNXRUNTIME_REPO=https://github.com/microsoft/onnxruntime
+ARG ONNXRUNTIME_REPO=https://github.com/mc-nv/onnxruntime.git
 ARG ONNXRUNTIME_BUILD_CONFIG={}
 """.format(
         FLAGS.triton_container, FLAGS.ort_version, FLAGS.ort_build_config
@@ -280,10 +280,7 @@ def dockerfile_for_linux(output_file):
 ARG ONNXRUNTIME_BUILD_CONFIG
 
 # Cherry-pick commit: https://github.com/microsoft/onnxruntime/commit/9dad9af9f9b48c05814d0c2d067d0565e8da6ce8
-RUN git clone -b main --recursive ${ONNXRUNTIME_REPO} onnxruntime \\
-    && cd onnxruntime \\
-    && git fetch origin refs/pull/25246/head:build-against-latest-upstream \\
-    && git checkout build-against-latest-upstream
+RUN git clone -b mc-nv/TPRD-1589/cccl-api-fix-preprocessor-vars --recursive ${ONNXRUNTIME_REPO} onnxruntime
         """
 
     if FLAGS.onnx_tensorrt_tag != "":