text: /*
   * Copyright (C) 2018-2022 Intel Corporation
   *
   * SPDX-License-Identifier: MIT
   *
   */
  
  #include "shared/source/helpers/aligned_memory.h"
  #include "shared/source/helpers/uint16_sse4.h"
  
  #include "gtest/gtest.h"
  
  using namespace NEO;
  
  TEST(Uint16Sse4, GivenMaskWhenCastingToBoolThenTrueIsReturned) {
      EXPECT_TRUE(static_cast<bool>(NEO::uint16x8_t::mask()));
  }
  
  TEST(Uint16Sse4, GivenZeroWhenCastingToBoolThenFalseIsReturned) {
      EXPECT_FALSE(static_cast<bool>(NEO::uint16x8_t::zero()));
  }
  
  TEST(Uint16Sse4, WhenConjoiningMaskAndZeroThenBooleanResultIsCorrect) {
      EXPECT_TRUE(NEO::uint16x8_t::mask() && NEO::uint16x8_t::mask());
      EXPECT_FALSE(NEO::uint16x8_t::mask() && NEO::uint16x8_t::zero());
      EXPECT_FALSE(NEO::uint16x8_t::zero() && NEO::uint16x8_t::mask());
      EXPECT_FALSE(NEO::uint16x8_t::zero() && NEO::uint16x8_t::zero());
  }
  
  TEST(Uint16Sse4, GivenOneWhenCreatingThenInstancesAreSame) {
      auto one = NEO::uint16x8_t::one();
      NEO::uint16x8_t alsoOne(one.value);
      EXPECT_EQ(0, memcmp(&alsoOne, &one, sizeof(NEO::uint16x8_t)));
  }
  
  TEST(Uint16Sse4, GivenValueWhenCreatingThenConstructorIsReplicated) {
      NEO::uint16x8_t allSevens(7u);
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(7u, allSevens.get(i));
      }
  }
  
  ALIGNAS(32)
  static const uint16_t laneValues[] = {
      0, 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};
  
  TEST(Uint16Sse4, GivenArrayWhenCreatingThenConstructorIsReplicated) {
      NEO::uint16x8_t lanes(laneValues);
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i), lanes.get(i));
      }
  }
  
  TEST(Uint16Sse4, WhenLoadingThenValuesAreSetCorrectly) {
      NEO::uint16x8_t lanes;
      lanes.load(laneValues);
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i), lanes.get(i));
      }
  }
  
  TEST(Uint16Sse4, WhenLoadingUnalignedThenValuesAreSetCorrectly) {
      NEO::uint16x8_t lanes;
      lanes.loadUnaligned(laneValues + 1);
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i + 1), lanes.get(i));
      }
  }
  
  TEST(Uint16Sse4, WhenStoringThenValuesAreSetCorrectly) {
      uint16_t *alignedMemory = reinterpret_cast<uint16_t *>(alignedMalloc(1024, 32));
  
      NEO::uint16x8_t lanes(laneValues);
      lanes.store(alignedMemory);
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i), alignedMemory[i]);
      }
  
      alignedFree(alignedMemory);
  }
  
  TEST(Uint16Sse4, WhenStoringUnalignedThenValuesAreSetCorrectly) {
      uint16_t *alignedMemory = reinterpret_cast<uint16_t *>(alignedMalloc(1024, 32));
  
      NEO::uint16x8_t lanes(laneValues);
      lanes.storeUnaligned(alignedMemory + 1);
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i), (alignedMemory + 1)[i]);
      }
  
      alignedFree(alignedMemory);
  }
  
  TEST(Uint16Sse4, WhenDecrementingThenValuesAreSetCorrectly) {
      NEO::uint16x8_t result(laneValues);
      result -= NEO::uint16x8_t::one();
  
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i - 1), result.get(i));
      }
  }
  
  TEST(Uint16Sse4, WhenIncrementingThenValuesAreSetCorrectly) {
      NEO::uint16x8_t result(laneValues);
      result += NEO::uint16x8_t::one();
  
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(static_cast<uint16_t>(i + 1), result.get(i));
      }
  }
  
  TEST(Uint16Sse4, WhenBlendingThenValuesAreSetCorrectly) {
      NEO::uint16x8_t a(NEO::uint16x8_t::one());
      NEO::uint16x8_t b(NEO::uint16x8_t::zero());
      NEO::uint16x8_t c;
  
      // c = mask ? a : b
      c = blend(a, b, NEO::uint16x8_t::mask());
  
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(a.get(i), c.get(i));
      }
  
      // c = mask ? a : b
      c = blend(a, b, NEO::uint16x8_t::zero());
  
      for (int i = 0; i < NEO::uint16x8_t::numChannels; ++i) {
          EXPECT_EQ(b.get(i), c.get(i));
      }
  }
  