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

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