From b46f0aa51111a2ad91e88b5d286c96f7c575a599 Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Wed, 12 Jun 2024 10:58:41 -0700 Subject: [PATCH] Add WAR for CUDA 12.5 build issue (#257) --- tools/gen_ort_dockerfile.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index 20e1c12..4e3edd1 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -205,8 +205,11 @@ def dockerfile_for_linux(output_file): ARG ONNXRUNTIME_REPO ARG ONNXRUNTIME_BUILD_CONFIG +# [FIXME: DLIS-6856] WAR to cherry pick fix to build with CUDA 12.5, should be +# removed once we advance to an ORT release that contains the fix. RUN git clone -b rel-${ONNXRUNTIME_VERSION} --recursive ${ONNXRUNTIME_REPO} onnxruntime && \ - (cd onnxruntime && git submodule update --init --recursive) + (cd onnxruntime && git submodule update --init --recursive && \ + git cherry-pick -n 362a62390545234d28661307c7a203f2546280ae) """ if FLAGS.onnx_tensorrt_tag != "":