Skip to content

Commit 5626515

Browse files
committed
Update gen script
1 parent 9b6bf20 commit 5626515

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -346,15 +346,14 @@ def dockerfile_for_linux(output_file):
346346

347347
df += """
348348
WORKDIR /workspace/onnxruntime
349-
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --skip_submodule_sync \
350-
--parallel $(( $(nproc --all ) + 2 )) --build_shared_lib \
349+
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --skip_submodule_sync --build_shared_lib \
351350
--compile_no_warning_as_error --build_dir /workspace/build --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES='{}' --cmake_extra_defines CMAKE_POLICY_VERSION_MINIMUM=3.5 --build_wheel"
352351
""".format(
353352
cuda_archs
354353
)
355354

356355
df += """
357-
RUN ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {}
356+
RUN ./build.sh ${{COMMON_BUILD_ARGS}} --parallel $(( $(nproc --all ) + 2 )) --update --build {}
358357
""".format(
359358
ep_flags
360359
)

0 commit comments

Comments
 (0)