From b20b0c9cb4c85bb9a3c901d5acaf421d84656850 Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Wed, 4 Oct 2023 14:17:55 +0100
Subject: Add initial TOSA MI generator support

Add support for dot-product MatMul - test set 0

Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ifd15b42570014b634f59c94a1fd1cd56bac79ea4
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
---
 .../src/generate/generate_dot_product_states.cc    | 119 +++++++++++++++++++++
 1 file changed, 119 insertions(+)
 create mode 100644 reference_model/src/generate/generate_dot_product_states.cc

(limited to 'reference_model/src/generate/generate_dot_product_states.cc')

diff --git a/reference_model/src/generate/generate_dot_product_states.cc b/reference_model/src/generate/generate_dot_product_states.cc
new file mode 100644
index 0000000..cd9ffba
--- /dev/null
+++ b/reference_model/src/generate/generate_dot_product_states.cc
@@ -0,0 +1,119 @@
+// Copyright (c) 2023, ARM Limited.
+//
+//    Licensed under the Apache License, Version 2.0 (the "License");
+//    you may not use this file except in compliance with the License.
+//    You may obtain a copy of the License at
+//
+//         http://www.apache.org/licenses/LICENSE-2.0
+//
+//    Unless required by applicable law or agreed to in writing, software
+//    distributed under the License is distributed on an "AS IS" BASIS,
+//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//    See the License for the specific language governing permissions and
+//    limitations under the License.
+
+#include "generate_dot_product.h"
+#include "generate_utils.h"
+
+#include <cstdint>
+
+namespace
+{
+
+// Input index global variables
+inline constexpr uint32_t P0 = 0;
+inline constexpr uint32_t P1 = 1;
+
+// Unused helper function
+template <typename... Args>
+inline void unused(Args&&...)
+{}
+
+// Primitive generator class
+//
+// Yields a new value on function operator access and increases the
+// index by one
+class PrimitiveGenerator
+{
+public:
+    PrimitiveGenerator(uint32_t S)
+        : _S(S)
+        , _m(0)
+        , _r(0)
+        , _index(0)
+    {
+        _m = (8 * _S + 1) * 0x705A5E75;
+        _r = _m + 1;
+    }
+
+    [[nodiscard]] float operator()()
+    {
+        _r           = _r * _m + 1;
+        float sign   = (_r >> 31) == 0 ? +1 : -1;
+        float pseudo = sign * (float)(_r & 0x7FFFFFFF) / (float)(0x7FFFFFFF);
+        ++_index;
+
+        return pseudo;
+    }
+
+    uint32_t index()
+    {
+        return _index;
+    }
+
+private:
+    uint32_t _S;
+    uint32_t _m;
+    uint32_t _r;
+    uint32_t _index;
+};
+
+//----------------------------------------------------------------------------//
+// State generators
+//----------------------------------------------------------------------------//
+
+// S0 generator
+class GeneratorS0 : public TosaReference::IDotProductGenerator
+{
+public:
+    GeneratorS0(uint32_t p)
+        : _p(p)
+        , _s0(0)    // set_data(2*S)
+        , _s1(1)    // set_data(2*S+1)
+    {}
+    float operator()(uint32_t k) override
+    {
+        unused(k);
+        const float s0 = _s0();
+        const float s1 = _s1();
+        if (_p == P0)
+            return s0 < 0.f ? 0.f : s1;
+        else
+            return s0 < 0.f ? s1 : 0.f;
+    }
+
+private:
+    uint32_t _p;
+    PrimitiveGenerator _s0;
+    PrimitiveGenerator _s1;
+};
+
+}    // namespace
+
+namespace TosaReference
+{
+
+std::unique_ptr<IDotProductGenerator> pickDotProductGenerator(const GenerateConfig& cfg)
+{
+    const DotProductInfo& dpinfo = cfg.dotProductInfo;
+    switch (dpinfo.s)
+    {
+        case 0:
+            return std::make_unique<GeneratorS0>(cfg.inputPos);
+        default:
+            return nullptr;
+    }
+    return nullptr;
+}
+
+}    // namespace TosaReference
\ No newline at end of file
-- 
cgit v1.2.1