lib/jxl/simd_util_test.cc

   1 // Copyright (c) the JPEG XL Project Authors. All rights reserved.
   2 //
   3 // Use of this source code is governed by a BSD-style
   4 // license that can be found in the LICENSE file.
   5
   6 #include "lib/jxl/testing.h"
   7
   8 #undef HWY_TARGET_INCLUDE
   9 #define HWY_TARGET_INCLUDE "lib/jxl/simd_util_test.cc"
  10 #include <hwy/foreach_target.h>
  11
  12 #include "lib/jxl/simd_util-inl.h"
  13
  14 // Test utils
  15 #include <hwy/highway.h>
  16 #include <hwy/tests/hwy_gtest.h>
  17 HWY_BEFORE_NAMESPACE();
  18 namespace jxl {
  19 namespace HWY_NAMESPACE {
  20 namespace {
  21
  22 HWY_NOINLINE void TestInterleave2() {
  23   HWY_FULL(float) d;
  24   auto vec1 = Iota(d, 0 * 128.0);
  25   auto vec2 = Iota(d, 1 * 128.0);
  26   HWY_ALIGN float mem[MaxLanes(d) * 2];
  27   StoreInterleaved(d, vec1, vec2, mem);
  28   for (size_t i = 0; i < Lanes(d); i++) {
  29     for (size_t j = 0; j < 2; j++) {
  30       EXPECT_EQ(mem[2 * i + j], j * 128 + i) << "i: " << i << " j: " << j;
  31     }
  32   }
  33 }
  34 HWY_NOINLINE void TestInterleave4() {
  35   HWY_FULL(float) d;
  36   auto vec1 = Iota(d, 0 * 128.0);
  37   auto vec2 = Iota(d, 1 * 128.0);
  38   auto vec3 = Iota(d, 2 * 128.0);
  39   auto vec4 = Iota(d, 3 * 128.0);
  40   HWY_ALIGN float mem[MaxLanes(d) * 4];
  41   StoreInterleaved(d, vec1, vec2, vec3, vec4, mem);
  42   for (size_t i = 0; i < Lanes(d); i++) {
  43     for (size_t j = 0; j < 4; j++) {
  44       EXPECT_EQ(mem[4 * i + j], j * 128 + i) << "i: " << i << " j: " << j;
  45     }
  46   }
  47 }
  48 HWY_NOINLINE void TestInterleave8() {
  49   HWY_FULL(float) d;
  50   auto vec1 = Iota(d, 0 * 128.0);
  51   auto vec2 = Iota(d, 1 * 128.0);
  52   auto vec3 = Iota(d, 2 * 128.0);
  53   auto vec4 = Iota(d, 3 * 128.0);
  54   auto vec5 = Iota(d, 4 * 128.0);
  55   auto vec6 = Iota(d, 5 * 128.0);
  56   auto vec7 = Iota(d, 6 * 128.0);
  57   auto vec8 = Iota(d, 7 * 128.0);
  58   HWY_ALIGN float mem[MaxLanes(d) * 8];
  59   StoreInterleaved(d, vec1, vec2, vec3, vec4, vec5, vec6, vec7, vec8, mem);
  60   for (size_t i = 0; i < Lanes(d); i++) {
  61     for (size_t j = 0; j < 8; j++) {
  62       EXPECT_EQ(mem[8 * i + j], j * 128 + i) << "i: " << i << " j: " << j;
  63     }
  64   }
  65 }
  66
  67 }  // namespace
  68 // NOLINTNEXTLINE(google-readability-namespace-comments)
  69 }  // namespace HWY_NAMESPACE
  70 }  // namespace jxl
  71 HWY_AFTER_NAMESPACE();
  72
  73 #if HWY_ONCE
  74 namespace jxl {
  75
  76 class SimdUtilTargetTest : public hwy::TestWithParamTarget {};
  77 HWY_TARGET_INSTANTIATE_TEST_SUITE_P(SimdUtilTargetTest);
  78
  79 HWY_EXPORT_AND_TEST_P(SimdUtilTargetTest, TestInterleave2);
  80 HWY_EXPORT_AND_TEST_P(SimdUtilTargetTest, TestInterleave4);
  81 HWY_EXPORT_AND_TEST_P(SimdUtilTargetTest, TestInterleave8);
  82
  83 }  // namespace jxl
  84 #endif  // HWY_ONCE