8349849e1c
* Use FindPython3.cmake
* Fixed compilation on macOS 14 with new core development tools
* Try to use Python3_SOABI instead of PYTHON_MODULE_EXTENSION
* Use Development.Module
* Keep specifying only Python3_EXECUTABLE
* Print PYTHON_MODULE_EXTENSION
* Added check for minimal cmake version for python API
* Returned Python3_INCLUDE_DIR for cross-compilation case
* Try to allow cmake older than 3.18
* Use build python interpreter to check cython dependency
* revert changes in .ci/openvino-onnx/Dockerfile
* removed unused code
* Fixed issue with variables scope
* Experiment: remove include dirs
* Corrected docs
* Use pybind11 function to set extension
* Revert "Experiment: remove include dirs"
This reverts commit 6f7f90211c.
* Refactor ConvolutionBackpropDataLayerTest, ConvolutionLayerTest, DeformableConvolutionLayerTest (#19810)
* Refactor ConvolutionBackpropDataLayerTest
* Refactor ConvolutionLayerTest
* Refactor DeformableConvolutionLayerTest
* Apply comments
* Apply comments
* Fix
* Updated minimum cmake version for Windows
* Simplified check
* Removed useless message status
* Use puiblic option
---------
Co-authored-by: Oleg Pipikin <oleg.pipikin@intel.com>
110 lines
4.5 KiB
CMake
110 lines
4.5 KiB
CMake
# Copyright (C) 2018-2023 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
set_property(GLOBAL PROPERTY JOB_POOLS four_jobs=4)
|
|
|
|
if(ENABLE_OV_ONNX_FRONTEND)
|
|
# if requirements are not installed automatically, we need to checks whether they are here
|
|
ov_check_pip_packages(REQUIREMENTS_FILE "${OpenVINO_SOURCE_DIR}/src/frontends/onnx/tests/requirements.txt"
|
|
RESULT_VAR onnx_FOUND
|
|
WARNING_MESSAGE "ONNX testing models weren't generated, some tests will fail due .onnx models not found"
|
|
MESSAGE_MODE WARNING)
|
|
endif()
|
|
|
|
function(ov_model_convert SRC DST OUT)
|
|
set(onnx_gen_script ${OpenVINO_SOURCE_DIR}/src/frontends/onnx/tests/onnx_prototxt_converter.py)
|
|
|
|
file(GLOB_RECURSE xml_models RELATIVE "${SRC}" "${SRC}/*.xml")
|
|
file(GLOB_RECURSE bin_models RELATIVE "${SRC}" "${SRC}/*.bin")
|
|
file(GLOB_RECURSE onnx_models RELATIVE "${SRC}" "${SRC}/*.onnx")
|
|
file(GLOB_RECURSE data_models RELATIVE "${SRC}" "${SRC}/*.data")
|
|
|
|
if(onnx_FOUND)
|
|
file(GLOB_RECURSE prototxt_models RELATIVE "${SRC}" "${SRC}/*.prototxt")
|
|
find_host_package(Python3 REQUIRED COMPONENTS Interpreter)
|
|
endif()
|
|
|
|
foreach(in_file IN LISTS prototxt_models xml_models bin_models onnx_models data_models)
|
|
get_filename_component(ext "${in_file}" EXT)
|
|
get_filename_component(rel_dir "${in_file}" DIRECTORY)
|
|
get_filename_component(name_we "${in_file}" NAME_WE)
|
|
set(model_source_dir "${SRC}/${rel_dir}")
|
|
|
|
if(ext STREQUAL ".prototxt")
|
|
# convert model
|
|
set(rel_out_name "${name_we}.onnx")
|
|
if(rel_dir)
|
|
set(rel_out_name "${rel_dir}/${rel_out_name}")
|
|
endif()
|
|
else()
|
|
# copy as is
|
|
set(rel_out_name "${in_file}")
|
|
endif()
|
|
|
|
set(full_out_name "${DST}/${rel_out_name}")
|
|
|
|
if(ext STREQUAL ".prototxt")
|
|
# convert .prototxt models to .onnx binary
|
|
add_custom_command(OUTPUT ${full_out_name}
|
|
COMMAND ${CMAKE_COMMAND} -E make_directory
|
|
"${DST}/${rel_dir}"
|
|
COMMAND ${Python3_EXECUTABLE} ${onnx_gen_script}
|
|
"${SRC}/${in_file}" ${full_out_name}
|
|
DEPENDS ${onnx_gen_script} "${SRC}/${in_file}"
|
|
COMMENT "Generate ${rel_out_name}"
|
|
JOB_POOL four_jobs
|
|
WORKING_DIRECTORY "${model_source_dir}")
|
|
else()
|
|
add_custom_command(OUTPUT ${full_out_name}
|
|
COMMAND ${CMAKE_COMMAND} -E make_directory
|
|
"${DST}/${rel_dir}"
|
|
COMMAND "${CMAKE_COMMAND}" -E copy_if_different
|
|
"${SRC}/${in_file}" ${full_out_name}
|
|
DEPENDS ${onnx_gen_script} "${SRC}/${in_file}"
|
|
COMMENT "Copy ${rel_out_name}"
|
|
JOB_POOL four_jobs
|
|
WORKING_DIRECTORY "${model_source_dir}")
|
|
endif()
|
|
list(APPEND files "${full_out_name}")
|
|
endforeach()
|
|
|
|
set(${OUT} ${files} PARENT_SCOPE)
|
|
endfunction()
|
|
|
|
if(OV_GENERATOR_MULTI_CONFIG AND CMAKE_VERSION VERSION_GREATER_EQUAL 3.20)
|
|
set(TEST_MODEL_ZOO_OUTPUT_DIR "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/$<CONFIG>/test_model_zoo" CACHE PATH "")
|
|
else()
|
|
set(TEST_MODEL_ZOO_OUTPUT_DIR "${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/test_model_zoo" CACHE PATH "")
|
|
endif()
|
|
|
|
ov_model_convert("${CMAKE_CURRENT_SOURCE_DIR}/src/core/tests"
|
|
"${TEST_MODEL_ZOO_OUTPUT_DIR}/core"
|
|
core_tests_out_files)
|
|
|
|
set(rel_path "src/tests/functional/plugin/shared/models")
|
|
ov_model_convert("${OpenVINO_SOURCE_DIR}/${rel_path}"
|
|
"${TEST_MODEL_ZOO_OUTPUT_DIR}/func_tests/models"
|
|
ft_out_files)
|
|
|
|
set(rel_path "src/frontends/onnx/tests/models")
|
|
ov_model_convert("${OpenVINO_SOURCE_DIR}/${rel_path}"
|
|
"${TEST_MODEL_ZOO_OUTPUT_DIR}/onnx"
|
|
onnx_fe_out_files)
|
|
|
|
if(ENABLE_TESTS)
|
|
add_custom_target(test_model_zoo DEPENDS ${core_tests_out_files}
|
|
${ft_out_files}
|
|
${onnx_fe_out_files})
|
|
|
|
# TODO Reenable PDPD after paddlepaddle==2.5.0 with compliant protobuf is released (ticket 95904)
|
|
#if (ENABLE_OV_PADDLE_FRONTEND)
|
|
# add_dependencies(test_model_zoo paddle_test_models)
|
|
#endif()
|
|
|
|
install(DIRECTORY "${TEST_MODEL_ZOO_OUTPUT_DIR}"
|
|
DESTINATION tests COMPONENT tests EXCLUDE_FROM_ALL)
|
|
|
|
set(TEST_MODEL_ZOO "./test_model_zoo" CACHE PATH "Path to test model zoo")
|
|
endif()
|