diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-02 17:16:25 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-10 16:35:17 +0000 |
commit | aee62afba99a74f772b97356fd4c18f3fdf37073 (patch) | |
tree | 47c7f0bc619ae44332f58c786b52460b40bc1a97 /reference_model/test | |
parent | bfc53031803338d9f0866f88f1d2deffd4928bcc (diff) | |
download | reference_model-aee62afba99a74f772b97356fd4c18f3fdf37073.tar.gz |
Main Compliance testing for FULLY_CONNECTED
Updated shapes to meet MIN_DOT_PRODUCTS.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I82297917c009b3120306f8a9bb965209d109ef8d
Diffstat (limited to 'reference_model/test')
-rw-r--r-- | reference_model/test/generate_tests.cpp | 170 |
1 files changed, 170 insertions, 0 deletions
diff --git a/reference_model/test/generate_tests.cpp b/reference_model/test/generate_tests.cpp index 40dd59f..9249f55 100644 --- a/reference_model/test/generate_tests.cpp +++ b/reference_model/test/generate_tests.cpp @@ -509,6 +509,7 @@ TEST_CASE("positive - pseudo random") check_not_output<float>(bufferP0, bufferP1); } } + void reduce_sum_test_FP32(const std::string tosaName, const size_t tosaElements, const std::string templateJsonCfg, @@ -580,4 +581,173 @@ TEST_CASE("positive - FP32 reduce_sum dot product (values 6,7 & 8)") reduce_sum_test_FP32(tosaName, tosaElements, templateJsonCfg, "5", expected); } } + +void fully_connected_test_FP32(const std::string tosaName[3], + const size_t tosaElements[3], + const std::string templateJsonCfg, + const std::string setStr, + int32_t param, + const std::vector<uint32_t> lastExpected) +{ + std::string jsonCfg = templateJsonCfg; + update_json_template(jsonCfg, "_SET_", setStr); + + std::vector<float> buffer(tosaElements[param]); + REQUIRE(tgd_generate_data(jsonCfg.c_str(), tosaName[param].c_str(), (void*)buffer.data(), tosaElements[param] * 4)); + if (param != 2) + { + // Get values at positions -8, -7 and -6 from the end + std::vector<float> last_three_ish(buffer.end() - 8, buffer.end() - 5); + check_output<float>(last_three_ish, lastExpected); + } + else + { + // Use last three as this buffer is too small + std::vector<float> last_three(buffer.end() - std::min<int>(3, buffer.size()), buffer.end()); + check_output<float>(last_three, lastExpected); + } +} +TEST_CASE("positive - FP32 fully_connected dot product (values -8, -7 & -6 from the end)") +{ + std::string templateJsonCfg = R"({ + "tensors" : { + "input" : { + "generator": "DOT_PRODUCT", + "data_type": "FP32", + "input_type": "VARIABLE", + "shape" : [ 6, 9 ], + "input_pos": 0, + "op" : "FULLY_CONNECTED", + "dot_product_info": { + "s": _SET_, + "ks": 9, + "acc_type": "FP32" + } + }, + "weight" : { + "generator": "DOT_PRODUCT", + "data_type": "FP32", + "input_type": "CONSTANT", + "shape" : [ 4, 9 ], + "input_pos": 1, + "op" : "FULLY_CONNECTED", + "dot_product_info": { + "s": _SET_, + "ks": 9, + "acc_type": "FP32" + } + }, + "bias" : { + "generator": "DOT_PRODUCT", + "data_type": "FP32", + "input_type": "CONSTANT", + "shape" : [ 4 ], + "input_pos": 2, + "op" : "FULLY_CONNECTED", + "dot_product_info": { + "s": _SET_, + "ks": 9, + "acc_type": "FP32" + } + } + + } + })"; + + const std::string tosaName[3] = { "input", "weight", "bias" }; + const size_t tosaElements[3] = { (6 * 9), (4 * 9), 4 }; + + SUBCASE("fully_connected, set 0, param 0") + { + std::vector<uint32_t> lastExpected = { 0x3f13876f, 0x0, 0x0 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "0", 0, lastExpected); + } + SUBCASE("fully_connected, set 0, param 1") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x0, 0x3f648dfd }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "0", 1, lastExpected); + } + SUBCASE("fully_connected, set 0, param 2") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x0, 0x0 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "0", 2, lastExpected); + } + SUBCASE("fully_connected, set 1, param 0") + { + // NOTE: Python test script produced 0x5e6cc8d7 - so off by 1 + std::vector<uint32_t> lastExpected = { 0x5e531bbf, 0x5e6cc8d8, 0x5e4e2539 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "1", 0, lastExpected); + } + SUBCASE("fully_connected, set 1, param 1") + { + std::vector<uint32_t> lastExpected = { 0x5e9870df, 0x5e9824c5, 0x5e9a898f }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "1", 1, lastExpected); + } + SUBCASE("fully_connected, set 1, param 2") + { + // NOTE: Python test script produced 0x7dc95352 - so off by 1 + std::vector<uint32_t> lastExpected = { 0x7d9a212a, 0x7dc95351, 0x7db7c1f2 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "1", 2, lastExpected); + } + SUBCASE("fully_connected, set 2, param 0") + { + std::vector<uint32_t> lastExpected = { 0xbcc1e987, 0xbe68efd7, 0x3db90130 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "2", 0, lastExpected); + } + SUBCASE("fully_connected, set 2, param 1") + { + std::vector<uint32_t> lastExpected = { 0x3e069935, 0x3de3a507, 0xbe6a0c0c }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "2", 1, lastExpected); + } + SUBCASE("fully_connected, set 2, param 2") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x0, 0x0 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "2", 2, lastExpected); + } + SUBCASE("fully_connected, set 3, param 0") + { + std::vector<uint32_t> lastExpected = { 0x3e57454e, 0x3b48e294, 0x3e889ece }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "3", 0, lastExpected); + } + SUBCASE("fully_connected, set 3, param 1") + { + std::vector<uint32_t> lastExpected = { 0xbd20e608, 0x3f91e619, 0x3e9ac66b }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "3", 1, lastExpected); + } + SUBCASE("fully_connected, set 3, param 2") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x0, 0x0 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "3", 2, lastExpected); + } + SUBCASE("fully_connected, set 4, param 0") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x5e29ad6d, 0x5e959eac }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "4", 0, lastExpected); + } + SUBCASE("fully_connected, set 4, param 1") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x5e6736d7, 0x5e44d571 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "4", 1, lastExpected); + } + SUBCASE("fully_connected, set 4, param 2") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x0, 0x0 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "4", 2, lastExpected); + } + SUBCASE("fully_connected, set 5, param 0") + { + std::vector<uint32_t> lastExpected = { 0xde8b0d70, 0xdd51465a, 0x5e57c772 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "5", 0, lastExpected); + } + SUBCASE("fully_connected, set 5, param 1") + { + std::vector<uint32_t> lastExpected = { 0xddde72f1, 0xde7e31ff, 0x5e0bdb32 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "5", 1, lastExpected); + } + SUBCASE("fully_connected, set 5, param 2") + { + std::vector<uint32_t> lastExpected = { 0x0, 0x0, 0x0 }; + fully_connected_test_FP32(tosaName, tosaElements, templateJsonCfg, "5", 2, lastExpected); + } +} TEST_SUITE_END(); // generate |