blob: 53233cdd022398196967eb271ba212b1b6859b18 (
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
if(NOT ENABLE_TEST)
return()
endif(NOT ENABLE_TEST)
unset(QUANTIZATION_VALUE_TEST)
unset(QUANTIZATION_VALUE_TEST_WITH_PARAM)
unset(QUANTIZATION_CONFIG_VALUE_TEST)
unset(QUANTIZATION_CONFIG_VALUE_TEST_WITH_PARAM)
macro(addTest NAME GRANULARITY DTYPE)
list(APPEND QUANTIZATION_VALUE_TEST ${NAME})
list(APPEND QUANTIZATION_VALUE_TEST_WITH_PARAM ${NAME} ${GRANULARITY} ${DTYPE})
endmacro(addTest)
macro(addQConfTest NAME GRANULARITY DTYPE)
list(APPEND QUANTIZATION_CONFIG_VALUE_TEST ${NAME})
list(APPEND QUANTIZATION_CONFIG_VALUE_TEST_WITH_PARAM ${NAME} ${GRANULARITY} ${DTYPE})
endmacro(addQConfTest)
# 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)
configure_file("${CMAKE_CURRENT_SOURCE_DIR}/gen_h5_explicit_inputs.py"
"${CMAKE_CURRENT_BINARY_DIR}/gen_h5_explicit_inputs.py" COPYONLY)
# TODO Run both 2.8.0 and 2.10.1 test for jammy
if(ONE_UBUNTU_CODENAME_JAMMY)
set(VIRTUALENV "${NNCC_OVERLAY_DIR}/venv_2_10_1")
else(ONE_UBUNTU_CODENAME_JAMMY)
set(VIRTUALENV "${NNCC_OVERLAY_DIR}/venv_2_8_0")
endif(ONE_UBUNTU_CODENAME_JAMMY)
###
### 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"
)
# Import pics module
get_target_property(PICS_BIN_PATH pics BINARY_DIR)
add_custom_command(OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/circle
COMMAND ${CMAKE_COMMAND} -E create_symlink
${PICS_BIN_PATH}/circle ${CMAKE_CURRENT_BINARY_DIR}/circle)
list(APPEND TEST_DEPS "${TEST_CONFIG}" "${CMAKE_CURRENT_BINARY_DIR}/circle")
# 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}
)
add_test(
NAME pota_fake_wquant_test_with_config
COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/test_fake_wquant_with_config.sh"
"${TEST_CONFIG}"
"${ARTIFACTS_BIN_PATH}"
${QUANTIZATION_CONFIG_VALUE_TEST_WITH_PARAM}
)
add_test(
NAME pota_parallel_record_minmax_test
COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/test_parallel_record_minmax.sh"
"${TEST_CONFIG}"
"${ARTIFACTS_BIN_PATH}"
${QUANTIZATION_VALUE_TEST_WITH_PARAM}
)
add_test(
NAME pota_quantization_test_with_config
COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/test_quantization_with_config.sh"
"${TEST_CONFIG}"
"${ARTIFACTS_BIN_PATH}"
${QUANTIZATION_CONFIG_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)
set_tests_properties(pota_parallel_record_minmax_test PROPERTIES DEPENDS pota_record_minmax_test)
set_tests_properties(pota_quantization_test_with_config PROPERTIES DEPENDS pota_fake_wquant_test_with_config)
|
