O-RAN E Maintenance Release contribution for ODULOW

[o-du/phy.git] / fhi_lib / test / test_xran / mod_compression_unit_test.cc

diff --git a/fhi_lib/test/test_xran/mod_compression_unit_test.cc b/fhi_lib/test/test_xran/mod_compression_unit_test.cc
new file mode 100644 (file)
index 0000000..3706a7a
--- /dev/null
+++ b/fhi_lib/test/test_xran/mod_compression_unit_test.cc
@@ -0,0 +1,85 @@
+/******************************************************************************
+*
+*   Copyright (c) 2020 Intel.
+*
+*   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 "common.hpp"
+#include "xran_fh_o_du.h"
+#include "xran_mod_compression.h"
+
+#include <stdint.h>
+#include <random>
+#include <algorithm>
+#include <iterator>
+#include <iostream>
+#include <cstring>
+
+const std::string module_name = "mod_compression";
+
+extern int _may_i_use_cpu_feature(unsigned __int64);
+
+int16_t loc_ModCompIn[273*12*14*2*16*2];
+int8_t loc_ModCompOut[273*12*14*2*16];
+
+class Mod_CompressionPerf : public KernelTests
+{
+protected:
+    struct xranlib_5gnr_mod_compression_request  mod_com_req;
+    struct xranlib_5gnr_mod_compression_response mod_com_rsp;
+
+    void SetUp() override {
+        init_test("mod_compression_performace");
+        // Create random number generator
+        std::random_device rd;
+        std::mt19937 gen(rd()); //Standard mersenne_twister_engine seeded with rd()
+        std::uniform_int_distribution<int16_t> randInt16(-32768, 32767);
+        std::uniform_int_distribution<int> randExpShift(0, 4);
+        std::memset(&loc_ModCompOut[0], 0, 273*12);
+
+        std::memset(&mod_com_req, 0, sizeof(struct xranlib_5gnr_mod_compression_request));
+        std::memset(&mod_com_rsp, 0, sizeof(struct xranlib_5gnr_mod_compression_response));
+        mod_com_req.unit = get_input_parameter<int16_t>("unit");
+        mod_com_req.modulation = get_input_parameter<xran_modulation_order>("modulation");
+        mod_com_req.num_symbols = get_input_parameter<int32_t>("num_symbols");
+
+        for (int m = 0; m < 2*mod_com_req.num_symbols; ++m) {
+            loc_ModCompIn[m] = int16_t(randInt16(gen));
+        }
+
+        mod_com_req.data_in    = (int16_t *)loc_ModCompIn;
+        mod_com_rsp.data_out   = (int8_t *)(loc_ModCompOut);
+    }
+
+    /* It's called after an execution of the each test case.*/
+    void TearDown() override {
+
+    }
+};
+
+TEST_P(Mod_CompressionPerf, AVX512_Mod_Comp)
+{
+     performance("AVX512", module_name, xranlib_5gnr_mod_compression_avx512, &mod_com_req, &mod_com_rsp);
+}
+
+
+TEST_P(Mod_CompressionPerf,  AVXSNC_Mod_Comp)
+{
+    if(_may_i_use_cpu_feature(_FEATURE_AVX512IFMA52))
+         performance("AVXSNC", module_name, xranlib_5gnr_mod_compression_snc, &mod_com_req, &mod_com_rsp);
+}
+
+INSTANTIATE_TEST_CASE_P(UnitTest, Mod_CompressionPerf,
+                        testing::ValuesIn(get_sequence(Mod_CompressionPerf::get_number_of_cases("mod_compression_performace"))));