IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use only 4 jobs to convert prototxt to onnx (#7597)

Browse Source
This commit is contained in:
Ilya Lavrenov 2021-09-22 18:02:07 +03:00 committed by GitHub
parent 0b4ae86001
commit 3b4c46ea80
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/test_model_zoo.cmake
View File

@ -2,6 +2,8 @@
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
# #
set_property(GLOBAL PROPERTY JOB_POOLS four_jobs=4)
function(ov_model_convert SRC DST OUT) function(ov_model_convert SRC DST OUT)
set(onnx_gen_script ${OpenVINO_SOURCE_DIR}/ngraph/test/models/onnx/onnx_prototxt_converter.py) set(onnx_gen_script ${OpenVINO_SOURCE_DIR}/ngraph/test/models/onnx/onnx_prototxt_converter.py)
@ -43,6 +45,7 @@ function(ov_model_convert SRC DST OUT)
"${SRC}/${in_file}" ${full_out_name} "${SRC}/${in_file}" ${full_out_name}
DEPENDS ${onnx_gen_script} "${SRC}/${in_file}" DEPENDS ${onnx_gen_script} "${SRC}/${in_file}"
COMMENT "Generate ${rel_out_name}" COMMENT "Generate ${rel_out_name}"
JOB_POOL four_jobs
WORKING_DIRECTORY "${model_source_dir}") WORKING_DIRECTORY "${model_source_dir}")
else() else()
add_custom_command(OUTPUT ${full_out_name} add_custom_command(OUTPUT ${full_out_name}
@ -50,6 +53,7 @@ function(ov_model_convert SRC DST OUT)
"${SRC}/${in_file}" ${full_out_name} "${SRC}/${in_file}" ${full_out_name}
DEPENDS ${onnx_gen_script} "${SRC}/${in_file}" DEPENDS ${onnx_gen_script} "${SRC}/${in_file}"
COMMENT "Copy ${rel_out_name}" COMMENT "Copy ${rel_out_name}"
JOB_POOL four_jobs
WORKING_DIRECTORY "${model_source_dir}") WORKING_DIRECTORY "${model_source_dir}")
endif() endif()
list(APPEND files "${full_out_name}") list(APPEND files "${full_out_name}")