Set max thread count to 64 to prevent OOM in CI

mc-nv · mc-nv · commit d2de8cff2223 · 2025-10-08T15:49:11.000-07:00
diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py
@@ -353,7 +353,8 @@ def dockerfile_for_linux(output_file):
     )
 
     df += """
-RUN ./build.sh ${{COMMON_BUILD_ARGS}} --parallel  $( nproc --all ) --update --build {}
+RUN N_PARALLEL_JOBS=$( [[ $(nproc --all) < 64 ]] &&  echo $(nproc --all)  || echo 64 ) && \\
+    ./build.sh ${{COMMON_BUILD_ARGS}} --parallel  ${N_PARALLEL_JOBS} --update --build {}
 """.format(
         ep_flags
     )

Original file line number	Diff line number	Diff line change
`@@ -353,7 +353,8 @@ def dockerfile_for_linux(output_file):`
`353`	`353`	`)`
`354`	`354`
`355`	`355`	`df += """`
`356`		`-RUN ./build.sh ${{COMMON_BUILD_ARGS}} --parallel $( nproc --all ) --update --build {}`
	`356`	`+RUN N_PARALLEL_JOBS=$( [[ $(nproc --all) < 64 ]] && echo $(nproc --all) \|\| echo 64 ) && \\`
	`357`	`+ ./build.sh ${{COMMON_BUILD_ARGS}} --parallel ${N_PARALLEL_JOBS} --update --build {}`
`357`	`358`	`""".format(`
`358`	`359`	`ep_flags`
`359`	`360`	`)`