Skip to content

Commit 9b6bf20

Browse files
committed
Set parallel count.
1 parent 2d751b0 commit 9b6bf20

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,8 @@ 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 --parallel --build_shared_lib \
349+
ARG COMMON_BUILD_ARGS="--config ${{ONNXRUNTIME_BUILD_CONFIG}} --skip_submodule_sync \
350+
--parallel $(( $(nproc --all ) + 2 )) --build_shared_lib \
350351
--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"
351352
""".format(
352353
cuda_archs

0 commit comments

Comments
 (0)