blob: 80661e5666cae75ef8b76502ad21cc0ef3ec9b6b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
unset(QUANTIZATION_VALUE_TEST)
unset(QUANTIZATION_VALUE_TEST_WITH_PARAM)
nnas_find_package(FlatBuffers QUIET)
if(NOT FlatBuffers_FOUND)
message(STATUS "Build pota-quantization-value-test: FAILED (missing FlatBuffers)")
return()
endif(NOT FlatBuffers_FOUND)
macro(addTest NAME GRANULARITY DTYPE)
list(APPEND QUANTIZATION_VALUE_TEST ${NAME})
list(APPEND QUANTIZATION_VALUE_TEST_WITH_PARAM ${NAME} ${GRANULARITY} ${DTYPE})
endmacro(addTest)
# Read "test.lst"
include("test.lst")
# Read "test.local.lst" if exists
include("test.local.lst" OPTIONAL)
unset(TEST_DEPS)
get_target_property(ARTIFACTS_BIN_PATH testDataGenerator BINARY_DIR)
get_target_property(SCHEMA_BIN_PATH mio_circle BINARY_DIR)
configure_file("${CMAKE_CURRENT_SOURCE_DIR}/gen_h5_explicit_inputs.py"
"${CMAKE_CURRENT_BINARY_DIR}/gen_h5_explicit_inputs.py" COPYONLY)
set(VIRTUALENV "${NNCC_OVERLAY_DIR}/venv_2_3_0")
###
### Generate test.config
###
set(TEST_CONFIG "${CMAKE_CURRENT_BINARY_DIR}/test.config")
add_custom_command(
OUTPUT ${TEST_CONFIG}
COMMAND ${CMAKE_COMMAND} -E remove -f ${TEST_CONFIG}
COMMAND ${CMAKE_COMMAND} -E echo 'RECORD_MINMAX_PATH=\"$<TARGET_FILE:record-minmax>\"' >> ${TEST_CONFIG}
COMMAND ${CMAKE_COMMAND} -E echo 'CIRCLE_QUANTIZER_PATH=\"$<TARGET_FILE:circle-quantizer>\"' >> ${TEST_CONFIG}
COMMAND ${CMAKE_COMMAND} -E echo 'CIRCLE_TENSORDUMP_PATH=\"$<TARGET_FILE:circle-tensordump>\"' >> ${TEST_CONFIG}
COMMAND ${CMAKE_COMMAND} -E echo 'VIRTUALENV=\"${VIRTUALENV}\"' >> ${TEST_CONFIG}
DEPENDS record-minmax
DEPENDS circle-quantizer
DEPENDS circle-tensordump
COMMENT "Generate test configuration"
)
###
### Generate python interface for circle schema
###
set(CIRCLE_SCHEMA_PYTHON_DIR "${CMAKE_CURRENT_BINARY_DIR}/circle")
add_custom_command(
OUTPUT ${CIRCLE_SCHEMA_PYTHON_DIR}
COMMAND ${CMAKE_COMMAND} -E remove_directory "${CIRCLE_SCHEMA_PYTHON_DIR}"
COMMAND "$<TARGET_FILE:flatbuffers::flatc>" --python
-o "${CMAKE_CURRENT_BINARY_DIR}" "${SCHEMA_BIN_PATH}/schema.fbs"
DEPENDS flatbuffers::flatc
COMMENT "Generate python interface for circle schema"
)
list(APPEND TEST_DEPS "${TEST_CONFIG}" "${CIRCLE_SCHEMA_PYTHON_DIR}")
# This enforces CMake to generate all the dependencies during "build" phase
add_custom_target(pota_quantization_value_test_deps ALL DEPENDS ${TEST_DEPS})
# Run tests
add_test(
NAME pota_fake_wquant_test
COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/test_fake_wquant.sh"
"${TEST_CONFIG}"
"${ARTIFACTS_BIN_PATH}"
${QUANTIZATION_VALUE_TEST_WITH_PARAM}
)
add_test(
NAME pota_record_minmax_test
COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/test_record_minmax.sh"
"${TEST_CONFIG}"
"${ARTIFACTS_BIN_PATH}"
${QUANTIZATION_VALUE_TEST_WITH_PARAM}
)
add_test(
NAME pota_quantization_test
COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/test_quantization.sh"
"${TEST_CONFIG}"
"${ARTIFACTS_BIN_PATH}"
${QUANTIZATION_VALUE_TEST_WITH_PARAM}
)
set_tests_properties(pota_record_minmax_test PROPERTIES DEPENDS pota_fake_wquant_test)
set_tests_properties(pota_quantization_test PROPERTIES DEPENDS pota_record_minmax_test)
|
