summaryrefslogtreecommitdiffstats
path: root/third_party/highway/hwy/targets_test.cc
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/highway/hwy/targets_test.cc
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/highway/hwy/targets_test.cc')
-rw-r--r--third_party/highway/hwy/targets_test.cc137
1 files changed, 137 insertions, 0 deletions
diff --git a/third_party/highway/hwy/targets_test.cc b/third_party/highway/hwy/targets_test.cc
new file mode 100644
index 0000000000..f00b24546d
--- /dev/null
+++ b/third_party/highway/hwy/targets_test.cc
@@ -0,0 +1,137 @@
+// Copyright 2020 Google LLC
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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 "hwy/targets.h"
+
+#include "hwy/tests/test_util-inl.h"
+
+namespace fake {
+
+#define DECLARE_FUNCTION(TGT) \
+ namespace N_##TGT { \
+ /* Function argument is just to ensure/demonstrate they are possible. */ \
+ int64_t FakeFunction(int) { return HWY_##TGT; } \
+ }
+
+DECLARE_FUNCTION(AVX3_DL)
+DECLARE_FUNCTION(AVX3)
+DECLARE_FUNCTION(AVX2)
+DECLARE_FUNCTION(SSE4)
+DECLARE_FUNCTION(SSSE3)
+DECLARE_FUNCTION(NEON)
+DECLARE_FUNCTION(SVE)
+DECLARE_FUNCTION(SVE2)
+DECLARE_FUNCTION(SVE_256)
+DECLARE_FUNCTION(SVE2_128)
+DECLARE_FUNCTION(PPC8)
+DECLARE_FUNCTION(WASM)
+DECLARE_FUNCTION(WASM_EMU256)
+DECLARE_FUNCTION(RVV)
+DECLARE_FUNCTION(SCALAR)
+DECLARE_FUNCTION(EMU128)
+
+HWY_EXPORT(FakeFunction);
+
+void CallFunctionForTarget(int64_t target, int line) {
+ if ((HWY_TARGETS & target) == 0) return;
+ hwy::SetSupportedTargetsForTest(target);
+
+ // Call Update() first to make &HWY_DYNAMIC_DISPATCH() return
+ // the pointer to the already cached function.
+ hwy::GetChosenTarget().Update(hwy::SupportedTargets());
+
+ EXPECT_EQ(target, HWY_DYNAMIC_DISPATCH(FakeFunction)(42)) << line;
+
+ // Calling DeInit() will test that the initializer function
+ // also calls the right function.
+ hwy::GetChosenTarget().DeInit();
+
+#if HWY_DISPATCH_WORKAROUND
+ EXPECT_EQ(HWY_STATIC_TARGET, HWY_DYNAMIC_DISPATCH(FakeFunction)(42)) << line;
+#else
+ EXPECT_EQ(target, HWY_DYNAMIC_DISPATCH(FakeFunction)(42)) << line;
+#endif
+
+ // Second call uses the cached value from the previous call.
+ EXPECT_EQ(target, HWY_DYNAMIC_DISPATCH(FakeFunction)(42)) << line;
+}
+
+void CheckFakeFunction() {
+ // When adding a target, also add to DECLARE_FUNCTION above.
+ CallFunctionForTarget(HWY_AVX3_DL, __LINE__);
+ CallFunctionForTarget(HWY_AVX3, __LINE__);
+ CallFunctionForTarget(HWY_AVX2, __LINE__);
+ CallFunctionForTarget(HWY_SSE4, __LINE__);
+ CallFunctionForTarget(HWY_SSSE3, __LINE__);
+ CallFunctionForTarget(HWY_NEON, __LINE__);
+ CallFunctionForTarget(HWY_SVE, __LINE__);
+ CallFunctionForTarget(HWY_SVE2, __LINE__);
+ CallFunctionForTarget(HWY_SVE_256, __LINE__);
+ CallFunctionForTarget(HWY_SVE2_128, __LINE__);
+ CallFunctionForTarget(HWY_PPC8, __LINE__);
+ CallFunctionForTarget(HWY_WASM, __LINE__);
+ CallFunctionForTarget(HWY_WASM_EMU256, __LINE__);
+ CallFunctionForTarget(HWY_RVV, __LINE__);
+ // The tables only have space for either HWY_SCALAR or HWY_EMU128; the former
+ // is opt-in only.
+#if defined(HWY_COMPILE_ONLY_SCALAR) || HWY_BROKEN_EMU128
+ CallFunctionForTarget(HWY_SCALAR, __LINE__);
+#else
+ CallFunctionForTarget(HWY_EMU128, __LINE__);
+#endif
+}
+
+} // namespace fake
+
+namespace hwy {
+
+class HwyTargetsTest : public testing::Test {
+ protected:
+ void TearDown() override {
+ SetSupportedTargetsForTest(0);
+ DisableTargets(0); // Reset the mask.
+ }
+};
+
+// Test that the order in the HWY_EXPORT static array matches the expected
+// value of the target bits. This is only checked for the targets that are
+// enabled in the current compilation.
+TEST_F(HwyTargetsTest, ChosenTargetOrderTest) { fake::CheckFakeFunction(); }
+
+TEST_F(HwyTargetsTest, DisabledTargetsTest) {
+ DisableTargets(~0LL);
+ // Check that disabling everything at least leaves the static target.
+ HWY_ASSERT(HWY_STATIC_TARGET == SupportedTargets());
+
+ DisableTargets(0); // Reset the mask.
+ const int64_t current_targets = SupportedTargets();
+ const int64_t enabled_baseline = static_cast<int64_t>(HWY_ENABLED_BASELINE);
+ // Exclude these two because they are always returned by SupportedTargets.
+ const int64_t fallback = HWY_SCALAR | HWY_EMU128;
+ if ((current_targets & ~enabled_baseline & ~fallback) == 0) {
+ // We can't test anything else if the only compiled target is the baseline.
+ return;
+ }
+
+ // Get the lowest bit in the mask (the best target) and disable that one.
+ const int64_t best_target = current_targets & (~current_targets + 1);
+ DisableTargets(best_target);
+
+ // Check that the other targets are still enabled.
+ HWY_ASSERT((best_target ^ current_targets) == SupportedTargets());
+ DisableTargets(0); // Reset the mask.
+}
+
+} // namespace hwy