зеркало из https://github.com/mozilla/gecko-dev.git
Bug 1786069 - Update libjxl and highway r=tnikkel
Differential Revision: https://phabricator.services.mozilla.com/D155272
This commit is contained in:
Родитель
ddc22eb2ef
Коммит
ca5565dea6
|
@ -11,6 +11,7 @@ LOCAL_INCLUDES += [
|
||||||
SOURCES += [
|
SOURCES += [
|
||||||
"/third_party/highway/hwy/aligned_allocator.cc",
|
"/third_party/highway/hwy/aligned_allocator.cc",
|
||||||
"/third_party/highway/hwy/contrib/image/image.cc",
|
"/third_party/highway/hwy/contrib/image/image.cc",
|
||||||
|
"/third_party/highway/hwy/per_target.cc",
|
||||||
"/third_party/highway/hwy/targets.cc",
|
"/third_party/highway/hwy/targets.cc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
@ -29,6 +30,7 @@ EXPORTS.hwy += [
|
||||||
EXPORTS.hwy.ops += [
|
EXPORTS.hwy.ops += [
|
||||||
"/third_party/highway/hwy/ops/arm_neon-inl.h",
|
"/third_party/highway/hwy/ops/arm_neon-inl.h",
|
||||||
"/third_party/highway/hwy/ops/arm_sve-inl.h",
|
"/third_party/highway/hwy/ops/arm_sve-inl.h",
|
||||||
|
"/third_party/highway/hwy/ops/emu128-inl.h",
|
||||||
"/third_party/highway/hwy/ops/generic_ops-inl.h",
|
"/third_party/highway/hwy/ops/generic_ops-inl.h",
|
||||||
"/third_party/highway/hwy/ops/rvv-inl.h",
|
"/third_party/highway/hwy/ops/rvv-inl.h",
|
||||||
"/third_party/highway/hwy/ops/scalar-inl.h",
|
"/third_party/highway/hwy/ops/scalar-inl.h",
|
||||||
|
|
|
@ -20,11 +20,11 @@ origin:
|
||||||
|
|
||||||
# Human-readable identifier for this version/release
|
# Human-readable identifier for this version/release
|
||||||
# Generally "version NNN", "tag SSS", "bookmark SSS"
|
# Generally "version NNN", "tag SSS", "bookmark SSS"
|
||||||
release: commit f13e3b956eb226561ac79427893ec0afd66f91a8 (2022-02-15T18:19:21Z).
|
release: 7f2e26854086fba4255220fd6c77e9141f1f87cc
|
||||||
|
|
||||||
# Revision to pull in
|
# Revision to pull in
|
||||||
# Must be a long or short commit SHA (long preferred)
|
# Must be a long or short commit SHA (long preferred)
|
||||||
revision: f13e3b956eb226561ac79427893ec0afd66f91a8
|
revision: 7f2e26854086fba4255220fd6c77e9141f1f87cc
|
||||||
|
|
||||||
# The package's license, where possible using the mnemonic from
|
# The package's license, where possible using the mnemonic from
|
||||||
# https://spdx.org/licenses/
|
# https://spdx.org/licenses/
|
||||||
|
|
|
@ -49,7 +49,6 @@ SOURCES += [
|
||||||
"/third_party/jpeg-xl/lib/jxl/enc_bit_writer.cc",
|
"/third_party/jpeg-xl/lib/jxl/enc_bit_writer.cc",
|
||||||
"/third_party/jpeg-xl/lib/jxl/entropy_coder.cc",
|
"/third_party/jpeg-xl/lib/jxl/entropy_coder.cc",
|
||||||
"/third_party/jpeg-xl/lib/jxl/epf.cc",
|
"/third_party/jpeg-xl/lib/jxl/epf.cc",
|
||||||
"/third_party/jpeg-xl/lib/jxl/exif.cc",
|
|
||||||
"/third_party/jpeg-xl/lib/jxl/fast_dct.cc",
|
"/third_party/jpeg-xl/lib/jxl/fast_dct.cc",
|
||||||
"/third_party/jpeg-xl/lib/jxl/fields.cc",
|
"/third_party/jpeg-xl/lib/jxl/fields.cc",
|
||||||
"/third_party/jpeg-xl/lib/jxl/frame_header.cc",
|
"/third_party/jpeg-xl/lib/jxl/frame_header.cc",
|
||||||
|
|
|
@ -10,9 +10,9 @@ origin:
|
||||||
|
|
||||||
url: https://github.com/libjxl/libjxl
|
url: https://github.com/libjxl/libjxl
|
||||||
|
|
||||||
release: 59f7d19e454bc5e1edd692187e1178f5926fdfd9 (2022-07-28T11:13:01Z).
|
release: bb8eac5d6acec223e44cf8cc72ae02f0816de311
|
||||||
|
|
||||||
revision: 59f7d19e454bc5e1edd692187e1178f5926fdfd9
|
revision: bb8eac5d6acec223e44cf8cc72ae02f0816de311
|
||||||
|
|
||||||
license: Apache-2.0
|
license: Apache-2.0
|
||||||
|
|
||||||
|
|
|
@ -1,7 +1,9 @@
|
||||||
load("@bazel_skylib//lib:selects.bzl", "selects")
|
load("@bazel_skylib//lib:selects.bzl", "selects")
|
||||||
|
|
||||||
load("@rules_cc//cc:defs.bzl", "cc_test")
|
load("@rules_cc//cc:defs.bzl", "cc_test")
|
||||||
package(default_visibility = ["//visibility:public"])
|
package(
|
||||||
|
default_visibility = ["//visibility:public"],
|
||||||
|
)
|
||||||
|
|
||||||
licenses(["notice"])
|
licenses(["notice"])
|
||||||
|
|
||||||
|
@ -14,10 +16,33 @@ config_setting(
|
||||||
)
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "compiler_msvc",
|
name = "compiler_clangcl",
|
||||||
|
flag_values = {"@bazel_tools//tools/cpp:compiler": "lexan"},
|
||||||
|
)
|
||||||
|
|
||||||
|
config_setting(
|
||||||
|
name = "compiler_msvc_actual",
|
||||||
flag_values = {"@bazel_tools//tools/cpp:compiler": "msvc"},
|
flag_values = {"@bazel_tools//tools/cpp:compiler": "msvc"},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# The above is insufficient for Bazel on Windows, which does not seem to
|
||||||
|
# detect/set a compiler flag. This workaround prevents compile errors due to
|
||||||
|
# passing clang-only warning flags to MSVC.
|
||||||
|
config_setting(
|
||||||
|
name = "compiler_msvc_cpu",
|
||||||
|
values = {
|
||||||
|
"cpu": "x64_windows",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
selects.config_setting_group(
|
||||||
|
name = "compiler_msvc",
|
||||||
|
match_any = [
|
||||||
|
":compiler_msvc_actual",
|
||||||
|
":compiler_msvc_cpu",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
config_setting(
|
config_setting(
|
||||||
name = "compiler_emscripten",
|
name = "compiler_emscripten",
|
||||||
values = {"cpu": "wasm32"},
|
values = {"cpu": "wasm32"},
|
||||||
|
@ -54,8 +79,8 @@ CLANG_GCC_COPTS = [
|
||||||
"-Wunreachable-code",
|
"-Wunreachable-code",
|
||||||
]
|
]
|
||||||
|
|
||||||
# Additional warnings only supported by Clang
|
# Warnings supported by Clang and Clang-cl
|
||||||
CLANG_ONLY_COPTS = [
|
CLANG_OR_CLANGCL_OPTS = CLANG_GCC_COPTS + [
|
||||||
"-Wfloat-overflow-conversion",
|
"-Wfloat-overflow-conversion",
|
||||||
"-Wfloat-zero-conversion",
|
"-Wfloat-zero-conversion",
|
||||||
"-Wfor-loop-analysis",
|
"-Wfor-loop-analysis",
|
||||||
|
@ -73,11 +98,19 @@ CLANG_ONLY_COPTS = [
|
||||||
"-Wunused-comparison",
|
"-Wunused-comparison",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# Warnings only supported by Clang, but not Clang-cl
|
||||||
|
CLANG_ONLY_COPTS = CLANG_OR_CLANGCL_OPTS + [
|
||||||
|
# Do not treat the third_party headers as system headers when building
|
||||||
|
# highway - the errors are pertinent.
|
||||||
|
"--no-system-header-prefix=third_party/highway",
|
||||||
|
]
|
||||||
|
|
||||||
COPTS = select({
|
COPTS = select({
|
||||||
":compiler_msvc": [],
|
":compiler_msvc": [],
|
||||||
":compiler_gcc": CLANG_GCC_COPTS,
|
":compiler_gcc": CLANG_GCC_COPTS,
|
||||||
|
":compiler_clangcl": CLANG_OR_CLANGCL_OPTS,
|
||||||
# Default to clang because compiler detection only works in Bazel
|
# Default to clang because compiler detection only works in Bazel
|
||||||
"//conditions:default": CLANG_GCC_COPTS + CLANG_ONLY_COPTS,
|
"//conditions:default": CLANG_ONLY_COPTS,
|
||||||
}) + select({
|
}) + select({
|
||||||
"@platforms//cpu:riscv64": [
|
"@platforms//cpu:riscv64": [
|
||||||
"-march=rv64gcv1p0",
|
"-march=rv64gcv1p0",
|
||||||
|
@ -87,6 +120,12 @@ COPTS = select({
|
||||||
],
|
],
|
||||||
})
|
})
|
||||||
|
|
||||||
|
DEFINES = select({
|
||||||
|
":compiler_msvc": ["HWY_SHARED_DEFINE"],
|
||||||
|
":compiler_clangcl": ["HWY_SHARED_DEFINE"],
|
||||||
|
"//conditions:default": [],
|
||||||
|
})
|
||||||
|
|
||||||
# Unused on Bazel builds, where this is not defined/known; Copybara replaces
|
# Unused on Bazel builds, where this is not defined/known; Copybara replaces
|
||||||
# usages with an empty list.
|
# usages with an empty list.
|
||||||
COMPAT = [
|
COMPAT = [
|
||||||
|
@ -102,6 +141,8 @@ cc_library(
|
||||||
name = "hwy",
|
name = "hwy",
|
||||||
srcs = [
|
srcs = [
|
||||||
"hwy/aligned_allocator.cc",
|
"hwy/aligned_allocator.cc",
|
||||||
|
"hwy/per_target.cc",
|
||||||
|
"hwy/print.cc",
|
||||||
"hwy/targets.cc",
|
"hwy/targets.cc",
|
||||||
],
|
],
|
||||||
# Normal headers with include guards
|
# Normal headers with include guards
|
||||||
|
@ -110,10 +151,12 @@ cc_library(
|
||||||
"hwy/base.h",
|
"hwy/base.h",
|
||||||
"hwy/cache_control.h",
|
"hwy/cache_control.h",
|
||||||
"hwy/detect_compiler_arch.h", # private
|
"hwy/detect_compiler_arch.h", # private
|
||||||
"hwy/highway_export.h",
|
"hwy/print.h",
|
||||||
],
|
],
|
||||||
compatible_with = [],
|
compatible_with = [],
|
||||||
copts = COPTS,
|
copts = COPTS,
|
||||||
|
defines = DEFINES,
|
||||||
|
local_defines = ["hwy_EXPORTS"],
|
||||||
textual_hdrs = [
|
textual_hdrs = [
|
||||||
# These are textual because config macros influence them:
|
# These are textual because config macros influence them:
|
||||||
"hwy/detect_targets.h", # private
|
"hwy/detect_targets.h", # private
|
||||||
|
@ -121,8 +164,12 @@ cc_library(
|
||||||
# End of list
|
# End of list
|
||||||
"hwy/highway.h", # public
|
"hwy/highway.h", # public
|
||||||
"hwy/foreach_target.h", # public
|
"hwy/foreach_target.h", # public
|
||||||
|
"hwy/per_target.h", # public
|
||||||
|
"hwy/print-inl.h", # public
|
||||||
|
"hwy/highway_export.h", # public
|
||||||
"hwy/ops/arm_neon-inl.h",
|
"hwy/ops/arm_neon-inl.h",
|
||||||
"hwy/ops/arm_sve-inl.h",
|
"hwy/ops/arm_sve-inl.h",
|
||||||
|
"hwy/ops/emu128-inl.h",
|
||||||
"hwy/ops/generic_ops-inl.h",
|
"hwy/ops/generic_ops-inl.h",
|
||||||
"hwy/ops/scalar-inl.h",
|
"hwy/ops/scalar-inl.h",
|
||||||
"hwy/ops/set_macros-inl.h",
|
"hwy/ops/set_macros-inl.h",
|
||||||
|
@ -140,9 +187,24 @@ cc_library(
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
cc_library(
|
||||||
|
name = "algo",
|
||||||
|
compatible_with = [],
|
||||||
|
copts = COPTS,
|
||||||
|
textual_hdrs = [
|
||||||
|
"hwy/contrib/algo/copy-inl.h",
|
||||||
|
"hwy/contrib/algo/find-inl.h",
|
||||||
|
"hwy/contrib/algo/transform-inl.h",
|
||||||
|
],
|
||||||
|
deps = [
|
||||||
|
":hwy",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
cc_library(
|
cc_library(
|
||||||
name = "dot",
|
name = "dot",
|
||||||
compatible_with = [],
|
compatible_with = [],
|
||||||
|
copts = COPTS,
|
||||||
textual_hdrs = [
|
textual_hdrs = [
|
||||||
"hwy/contrib/dot/dot-inl.h",
|
"hwy/contrib/dot/dot-inl.h",
|
||||||
],
|
],
|
||||||
|
@ -160,6 +222,8 @@ cc_library(
|
||||||
"hwy/contrib/image/image.h",
|
"hwy/contrib/image/image.h",
|
||||||
],
|
],
|
||||||
compatible_with = [],
|
compatible_with = [],
|
||||||
|
copts = COPTS,
|
||||||
|
local_defines = ["hwy_contrib_EXPORTS"],
|
||||||
deps = [
|
deps = [
|
||||||
":hwy",
|
":hwy",
|
||||||
],
|
],
|
||||||
|
@ -168,6 +232,7 @@ cc_library(
|
||||||
cc_library(
|
cc_library(
|
||||||
name = "math",
|
name = "math",
|
||||||
compatible_with = [],
|
compatible_with = [],
|
||||||
|
copts = COPTS,
|
||||||
textual_hdrs = [
|
textual_hdrs = [
|
||||||
"hwy/contrib/math/math-inl.h",
|
"hwy/contrib/math/math-inl.h",
|
||||||
],
|
],
|
||||||
|
@ -181,6 +246,9 @@ cc_library(
|
||||||
name = "hwy_test_util",
|
name = "hwy_test_util",
|
||||||
srcs = ["hwy/tests/test_util.cc"],
|
srcs = ["hwy/tests/test_util.cc"],
|
||||||
hdrs = ["hwy/tests/test_util.h"],
|
hdrs = ["hwy/tests/test_util.h"],
|
||||||
|
compatible_with = [],
|
||||||
|
copts = COPTS,
|
||||||
|
local_defines = ["hwy_test_EXPORTS"],
|
||||||
textual_hdrs = [
|
textual_hdrs = [
|
||||||
"hwy/tests/test_util-inl.h",
|
"hwy/tests/test_util-inl.h",
|
||||||
"hwy/tests/hwy_gtest.h",
|
"hwy/tests/hwy_gtest.h",
|
||||||
|
@ -196,12 +264,16 @@ cc_library(
|
||||||
name = "nanobenchmark",
|
name = "nanobenchmark",
|
||||||
srcs = ["hwy/nanobenchmark.cc"],
|
srcs = ["hwy/nanobenchmark.cc"],
|
||||||
hdrs = ["hwy/nanobenchmark.h"],
|
hdrs = ["hwy/nanobenchmark.h"],
|
||||||
|
compatible_with = [],
|
||||||
|
copts = COPTS,
|
||||||
|
local_defines = ["hwy_EXPORTS"],
|
||||||
deps = [":hwy"],
|
deps = [":hwy"],
|
||||||
)
|
)
|
||||||
|
|
||||||
cc_binary(
|
cc_binary(
|
||||||
name = "benchmark",
|
name = "benchmark",
|
||||||
srcs = ["hwy/examples/benchmark.cc"],
|
srcs = ["hwy/examples/benchmark.cc"],
|
||||||
|
copts = COPTS,
|
||||||
deps = [
|
deps = [
|
||||||
":hwy",
|
":hwy",
|
||||||
":nanobenchmark",
|
":nanobenchmark",
|
||||||
|
@ -212,6 +284,8 @@ cc_library(
|
||||||
name = "skeleton",
|
name = "skeleton",
|
||||||
srcs = ["hwy/examples/skeleton.cc"],
|
srcs = ["hwy/examples/skeleton.cc"],
|
||||||
hdrs = ["hwy/examples/skeleton.h"],
|
hdrs = ["hwy/examples/skeleton.h"],
|
||||||
|
copts = COPTS,
|
||||||
|
local_defines = ["hwy_EXPORTS"],
|
||||||
textual_hdrs = ["hwy/examples/skeleton-inl.h"],
|
textual_hdrs = ["hwy/examples/skeleton-inl.h"],
|
||||||
deps = [
|
deps = [
|
||||||
":hwy",
|
":hwy",
|
||||||
|
@ -226,6 +300,9 @@ cc_binary(
|
||||||
|
|
||||||
# path, name
|
# path, name
|
||||||
HWY_TESTS = [
|
HWY_TESTS = [
|
||||||
|
("hwy/contrib/algo/", "copy_test"),
|
||||||
|
("hwy/contrib/algo/", "find_test"),
|
||||||
|
("hwy/contrib/algo/", "transform_test"),
|
||||||
("hwy/contrib/dot/", "dot_test"),
|
("hwy/contrib/dot/", "dot_test"),
|
||||||
("hwy/contrib/image/", "image_test"),
|
("hwy/contrib/image/", "image_test"),
|
||||||
("hwy/contrib/math/", "math_test"),
|
("hwy/contrib/math/", "math_test"),
|
||||||
|
@ -238,20 +315,40 @@ HWY_TESTS = [
|
||||||
("hwy/", "targets_test"),
|
("hwy/", "targets_test"),
|
||||||
("hwy/tests/", "arithmetic_test"),
|
("hwy/tests/", "arithmetic_test"),
|
||||||
("hwy/tests/", "blockwise_test"),
|
("hwy/tests/", "blockwise_test"),
|
||||||
|
("hwy/tests/", "blockwise_shift_test"),
|
||||||
("hwy/tests/", "combine_test"),
|
("hwy/tests/", "combine_test"),
|
||||||
("hwy/tests/", "compare_test"),
|
("hwy/tests/", "compare_test"),
|
||||||
|
("hwy/tests/", "compress_test"),
|
||||||
("hwy/tests/", "convert_test"),
|
("hwy/tests/", "convert_test"),
|
||||||
("hwy/tests/", "crypto_test"),
|
("hwy/tests/", "crypto_test"),
|
||||||
("hwy/tests/", "demote_test"),
|
("hwy/tests/", "demote_test"),
|
||||||
|
("hwy/tests/", "float_test"),
|
||||||
|
("hwy/tests/", "if_test"),
|
||||||
|
("hwy/tests/", "interleaved_test"),
|
||||||
("hwy/tests/", "logical_test"),
|
("hwy/tests/", "logical_test"),
|
||||||
("hwy/tests/", "mask_test"),
|
("hwy/tests/", "mask_test"),
|
||||||
|
("hwy/tests/", "mask_mem_test"),
|
||||||
("hwy/tests/", "memory_test"),
|
("hwy/tests/", "memory_test"),
|
||||||
|
("hwy/tests/", "mul_test"),
|
||||||
|
("hwy/tests/", "reduction_test"),
|
||||||
|
("hwy/tests/", "reverse_test"),
|
||||||
("hwy/tests/", "shift_test"),
|
("hwy/tests/", "shift_test"),
|
||||||
("hwy/tests/", "swizzle_test"),
|
("hwy/tests/", "swizzle_test"),
|
||||||
("hwy/tests/", "test_util_test"),
|
("hwy/tests/", "test_util_test"),
|
||||||
]
|
]
|
||||||
|
|
||||||
|
HWY_TEST_COPTS = select({
|
||||||
|
":compiler_msvc": [],
|
||||||
|
"//conditions:default": [
|
||||||
|
# gTest triggers this warning (which is enabled by the
|
||||||
|
# extra-semi in COPTS), so we need to disable it here,
|
||||||
|
# but it's still enabled for :hwy.
|
||||||
|
"-Wno-c++98-compat-extra-semi",
|
||||||
|
],
|
||||||
|
})
|
||||||
|
|
||||||
HWY_TEST_DEPS = [
|
HWY_TEST_DEPS = [
|
||||||
|
":algo",
|
||||||
":dot",
|
":dot",
|
||||||
":hwy",
|
":hwy",
|
||||||
":hwy_test_util",
|
":hwy_test_util",
|
||||||
|
@ -272,12 +369,7 @@ HWY_TEST_DEPS = [
|
||||||
srcs = [
|
srcs = [
|
||||||
subdir + test + ".cc",
|
subdir + test + ".cc",
|
||||||
],
|
],
|
||||||
copts = COPTS + [
|
copts = COPTS + HWY_TEST_COPTS,
|
||||||
# gTest triggers this warning (which is enabled by the
|
|
||||||
# extra-semi in COPTS), so we need to disable it here,
|
|
||||||
# but it's still enabled for :hwy.
|
|
||||||
"-Wno-c++98-compat-extra-semi",
|
|
||||||
],
|
|
||||||
features = select({
|
features = select({
|
||||||
"@platforms//cpu:riscv64": ["fully_static_link"],
|
"@platforms//cpu:riscv64": ["fully_static_link"],
|
||||||
"//conditions:default": [],
|
"//conditions:default": [],
|
||||||
|
|
|
@ -19,7 +19,7 @@ if(POLICY CMP0083)
|
||||||
cmake_policy(SET CMP0083 NEW)
|
cmake_policy(SET CMP0083 NEW)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
project(hwy VERSION 0.16.0) # Keep in sync with highway.h version
|
project(hwy VERSION 1.0.0) # Keep in sync with highway.h version
|
||||||
|
|
||||||
# Directly define the ABI version from the cmake project() version values:
|
# Directly define the ABI version from the cmake project() version values:
|
||||||
set(LIBRARY_VERSION "${hwy_VERSION}")
|
set(LIBRARY_VERSION "${hwy_VERSION}")
|
||||||
|
@ -48,6 +48,7 @@ set(HWY_CMAKE_ARM7 OFF CACHE BOOL "Set copts for ARMv7 with NEON (requires vfpv4
|
||||||
# arise due to compiler/platform changes. Enable this in CI/tests.
|
# arise due to compiler/platform changes. Enable this in CI/tests.
|
||||||
set(HWY_WARNINGS_ARE_ERRORS OFF CACHE BOOL "Add -Werror flag?")
|
set(HWY_WARNINGS_ARE_ERRORS OFF CACHE BOOL "Add -Werror flag?")
|
||||||
|
|
||||||
|
set(HWY_ENABLE_CONTRIB ON CACHE BOOL "Include contrib/")
|
||||||
set(HWY_ENABLE_EXAMPLES ON CACHE BOOL "Build examples")
|
set(HWY_ENABLE_EXAMPLES ON CACHE BOOL "Build examples")
|
||||||
set(HWY_ENABLE_INSTALL ON CACHE BOOL "Install library")
|
set(HWY_ENABLE_INSTALL ON CACHE BOOL "Install library")
|
||||||
|
|
||||||
|
@ -62,12 +63,25 @@ check_cxx_source_compiles(
|
||||||
HWY_EMSCRIPTEN
|
HWY_EMSCRIPTEN
|
||||||
)
|
)
|
||||||
|
|
||||||
set(HWY_CONTRIB_SOURCES
|
check_cxx_source_compiles(
|
||||||
|
"int main() {
|
||||||
|
#if !defined(__riscv)
|
||||||
|
static_assert(false, \"__riscv is not defined\");
|
||||||
|
#endif
|
||||||
|
return 0;
|
||||||
|
}"
|
||||||
|
HWY_RISCV
|
||||||
|
)
|
||||||
|
|
||||||
|
if (HWY_ENABLE_CONTRIB)
|
||||||
|
# Glob all the traits so we don't need to modify this file when adding
|
||||||
|
# additional special cases.
|
||||||
|
file(GLOB HWY_CONTRIB_SOURCES "hwy/contrib/sort/vqsort_*.cc")
|
||||||
|
list(APPEND HWY_CONTRIB_SOURCES
|
||||||
hwy/contrib/dot/dot-inl.h
|
hwy/contrib/dot/dot-inl.h
|
||||||
hwy/contrib/image/image.cc
|
hwy/contrib/image/image.cc
|
||||||
hwy/contrib/image/image.h
|
hwy/contrib/image/image.h
|
||||||
hwy/contrib/math/math-inl.h
|
hwy/contrib/math/math-inl.h
|
||||||
hwy/contrib/sort/disabled_targets.h
|
|
||||||
hwy/contrib/sort/shared-inl.h
|
hwy/contrib/sort/shared-inl.h
|
||||||
hwy/contrib/sort/sorting_networks-inl.h
|
hwy/contrib/sort/sorting_networks-inl.h
|
||||||
hwy/contrib/sort/traits-inl.h
|
hwy/contrib/sort/traits-inl.h
|
||||||
|
@ -75,25 +89,8 @@ set(HWY_CONTRIB_SOURCES
|
||||||
hwy/contrib/sort/vqsort-inl.h
|
hwy/contrib/sort/vqsort-inl.h
|
||||||
hwy/contrib/sort/vqsort.cc
|
hwy/contrib/sort/vqsort.cc
|
||||||
hwy/contrib/sort/vqsort.h
|
hwy/contrib/sort/vqsort.h
|
||||||
hwy/contrib/sort/vqsort_128a.cc
|
|
||||||
hwy/contrib/sort/vqsort_128d.cc
|
|
||||||
hwy/contrib/sort/vqsort_f32a.cc
|
|
||||||
hwy/contrib/sort/vqsort_f32d.cc
|
|
||||||
hwy/contrib/sort/vqsort_f64a.cc
|
|
||||||
hwy/contrib/sort/vqsort_f64d.cc
|
|
||||||
hwy/contrib/sort/vqsort_i16a.cc
|
|
||||||
hwy/contrib/sort/vqsort_i16d.cc
|
|
||||||
hwy/contrib/sort/vqsort_i32a.cc
|
|
||||||
hwy/contrib/sort/vqsort_i32d.cc
|
|
||||||
hwy/contrib/sort/vqsort_i64a.cc
|
|
||||||
hwy/contrib/sort/vqsort_i64d.cc
|
|
||||||
hwy/contrib/sort/vqsort_u16a.cc
|
|
||||||
hwy/contrib/sort/vqsort_u16d.cc
|
|
||||||
hwy/contrib/sort/vqsort_u32a.cc
|
|
||||||
hwy/contrib/sort/vqsort_u32d.cc
|
|
||||||
hwy/contrib/sort/vqsort_u64a.cc
|
|
||||||
hwy/contrib/sort/vqsort_u64d.cc
|
|
||||||
)
|
)
|
||||||
|
endif() # HWY_ENABLE_CONTRIB
|
||||||
|
|
||||||
set(HWY_SOURCES
|
set(HWY_SOURCES
|
||||||
hwy/aligned_allocator.cc
|
hwy/aligned_allocator.cc
|
||||||
|
@ -109,6 +106,7 @@ set(HWY_SOURCES
|
||||||
hwy/nanobenchmark.h
|
hwy/nanobenchmark.h
|
||||||
hwy/ops/arm_neon-inl.h
|
hwy/ops/arm_neon-inl.h
|
||||||
hwy/ops/arm_sve-inl.h
|
hwy/ops/arm_sve-inl.h
|
||||||
|
hwy/ops/emu128-inl.h
|
||||||
hwy/ops/generic_ops-inl.h
|
hwy/ops/generic_ops-inl.h
|
||||||
hwy/ops/scalar-inl.h
|
hwy/ops/scalar-inl.h
|
||||||
hwy/ops/set_macros-inl.h
|
hwy/ops/set_macros-inl.h
|
||||||
|
@ -117,6 +115,11 @@ set(HWY_SOURCES
|
||||||
hwy/ops/x86_128-inl.h
|
hwy/ops/x86_128-inl.h
|
||||||
hwy/ops/x86_256-inl.h
|
hwy/ops/x86_256-inl.h
|
||||||
hwy/ops/x86_512-inl.h
|
hwy/ops/x86_512-inl.h
|
||||||
|
hwy/per_target.cc
|
||||||
|
hwy/per_target.h
|
||||||
|
hwy/print-inl.h
|
||||||
|
hwy/print.cc
|
||||||
|
hwy/print.h
|
||||||
hwy/targets.cc
|
hwy/targets.cc
|
||||||
hwy/targets.h
|
hwy/targets.h
|
||||||
)
|
)
|
||||||
|
@ -129,7 +132,10 @@ set(HWY_TEST_SOURCES
|
||||||
)
|
)
|
||||||
|
|
||||||
if (MSVC)
|
if (MSVC)
|
||||||
# TODO(janwas): add flags
|
set(HWY_FLAGS
|
||||||
|
# fix build error C1128 in blockwise*_test & arithmetic_test
|
||||||
|
/bigobj
|
||||||
|
)
|
||||||
else()
|
else()
|
||||||
set(HWY_FLAGS
|
set(HWY_FLAGS
|
||||||
# Avoid changing binaries based on the current time and date.
|
# Avoid changing binaries based on the current time and date.
|
||||||
|
@ -215,10 +221,23 @@ else()
|
||||||
)
|
)
|
||||||
endif() # HWY_CMAKE_ARM7
|
endif() # HWY_CMAKE_ARM7
|
||||||
|
|
||||||
|
if(HWY_RISCV)
|
||||||
|
list(APPEND HWY_FLAGS -march=rv64gcv1p0)
|
||||||
|
if(${CMAKE_CXX_COMPILER_ID} MATCHES "Clang")
|
||||||
|
list(APPEND HWY_FLAGS -menable-experimental-extensions)
|
||||||
|
endif()
|
||||||
|
endif()
|
||||||
|
|
||||||
if (HWY_WARNINGS_ARE_ERRORS)
|
if (HWY_WARNINGS_ARE_ERRORS)
|
||||||
list(APPEND HWY_FLAGS -Werror)
|
list(APPEND HWY_FLAGS -Werror)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
# Prevent "wasm-ld: error: --shared-memory is disallowed by targets.cc.o
|
||||||
|
# because it was not compiled with 'atomics' or 'bulk-memory' features."
|
||||||
|
if (HWY_EMSCRIPTEN)
|
||||||
|
list(APPEND HWY_FLAGS -matomics)
|
||||||
|
endif()
|
||||||
|
|
||||||
endif() # !MSVC
|
endif() # !MSVC
|
||||||
|
|
||||||
# By default prefer STATIC build (legacy behavior)
|
# By default prefer STATIC build (legacy behavior)
|
||||||
|
@ -249,31 +268,57 @@ target_compile_definitions(hwy PUBLIC "${DLLEXPORT_TO_DEFINE}")
|
||||||
target_compile_options(hwy PRIVATE ${HWY_FLAGS})
|
target_compile_options(hwy PRIVATE ${HWY_FLAGS})
|
||||||
set_property(TARGET hwy PROPERTY POSITION_INDEPENDENT_CODE ON)
|
set_property(TARGET hwy PROPERTY POSITION_INDEPENDENT_CODE ON)
|
||||||
set_target_properties(hwy PROPERTIES VERSION ${LIBRARY_VERSION} SOVERSION ${LIBRARY_SOVERSION})
|
set_target_properties(hwy PROPERTIES VERSION ${LIBRARY_VERSION} SOVERSION ${LIBRARY_SOVERSION})
|
||||||
target_include_directories(hwy PUBLIC ${CMAKE_CURRENT_LIST_DIR})
|
target_include_directories(hwy PUBLIC
|
||||||
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}>
|
||||||
|
$<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>)
|
||||||
target_compile_features(hwy PUBLIC cxx_std_11)
|
target_compile_features(hwy PUBLIC cxx_std_11)
|
||||||
set_target_properties(hwy PROPERTIES
|
set_target_properties(hwy PROPERTIES
|
||||||
LINK_DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version)
|
LINK_DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version)
|
||||||
# not supported by MSVC/Clang, safe to skip (we use DLLEXPORT annotations)
|
|
||||||
if(UNIX AND NOT APPLE)
|
if(UNIX AND NOT APPLE)
|
||||||
|
if(NOT HWY_EMSCRIPTEN)
|
||||||
|
# For GCC __atomic_store_8, see #887
|
||||||
|
target_link_libraries(hwy atomic)
|
||||||
|
endif()
|
||||||
|
# not supported by MSVC/Clang, safe to skip (we use DLLEXPORT annotations)
|
||||||
set_property(TARGET hwy APPEND_STRING PROPERTY
|
set_property(TARGET hwy APPEND_STRING PROPERTY
|
||||||
LINK_FLAGS " -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version")
|
LINK_FLAGS " -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version")
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
if (HWY_ENABLE_CONTRIB)
|
||||||
add_library(hwy_contrib ${HWY_LIBRARY_TYPE} ${HWY_CONTRIB_SOURCES})
|
add_library(hwy_contrib ${HWY_LIBRARY_TYPE} ${HWY_CONTRIB_SOURCES})
|
||||||
target_link_libraries(hwy_contrib hwy)
|
target_link_libraries(hwy_contrib hwy)
|
||||||
target_compile_options(hwy_contrib PRIVATE ${HWY_FLAGS})
|
target_compile_options(hwy_contrib PRIVATE ${HWY_FLAGS})
|
||||||
set_property(TARGET hwy_contrib PROPERTY POSITION_INDEPENDENT_CODE ON)
|
set_property(TARGET hwy_contrib PROPERTY POSITION_INDEPENDENT_CODE ON)
|
||||||
set_target_properties(hwy_contrib PROPERTIES VERSION ${LIBRARY_VERSION} SOVERSION ${LIBRARY_SOVERSION})
|
set_target_properties(hwy_contrib PROPERTIES VERSION ${LIBRARY_VERSION} SOVERSION ${LIBRARY_SOVERSION})
|
||||||
target_include_directories(hwy_contrib PUBLIC ${CMAKE_CURRENT_LIST_DIR})
|
target_include_directories(hwy_contrib PUBLIC
|
||||||
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}>
|
||||||
|
$<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>)
|
||||||
target_compile_features(hwy_contrib PUBLIC cxx_std_11)
|
target_compile_features(hwy_contrib PUBLIC cxx_std_11)
|
||||||
|
set_target_properties(hwy_contrib PROPERTIES
|
||||||
|
LINK_DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version)
|
||||||
|
# not supported by MSVC/Clang, safe to skip (we use DLLEXPORT annotations)
|
||||||
|
if(UNIX AND NOT APPLE)
|
||||||
|
set_property(TARGET hwy_contrib APPEND_STRING PROPERTY
|
||||||
|
LINK_FLAGS " -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version")
|
||||||
|
endif()
|
||||||
|
endif() # HWY_ENABLE_CONTRIB
|
||||||
|
|
||||||
add_library(hwy_test ${HWY_LIBRARY_TYPE} ${HWY_TEST_SOURCES})
|
add_library(hwy_test ${HWY_LIBRARY_TYPE} ${HWY_TEST_SOURCES})
|
||||||
target_link_libraries(hwy_test hwy)
|
target_link_libraries(hwy_test hwy)
|
||||||
target_compile_options(hwy_test PRIVATE ${HWY_FLAGS})
|
target_compile_options(hwy_test PRIVATE ${HWY_FLAGS})
|
||||||
set_property(TARGET hwy_test PROPERTY POSITION_INDEPENDENT_CODE ON)
|
set_property(TARGET hwy_test PROPERTY POSITION_INDEPENDENT_CODE ON)
|
||||||
set_target_properties(hwy_test PROPERTIES VERSION ${LIBRARY_VERSION} SOVERSION ${LIBRARY_SOVERSION})
|
set_target_properties(hwy_test PROPERTIES VERSION ${LIBRARY_VERSION} SOVERSION ${LIBRARY_SOVERSION})
|
||||||
target_include_directories(hwy_test PUBLIC ${CMAKE_CURRENT_LIST_DIR})
|
target_include_directories(hwy_test PUBLIC
|
||||||
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}>
|
||||||
|
$<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>)
|
||||||
target_compile_features(hwy_test PUBLIC cxx_std_11)
|
target_compile_features(hwy_test PUBLIC cxx_std_11)
|
||||||
|
set_target_properties(hwy_test PROPERTIES
|
||||||
|
LINK_DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version)
|
||||||
|
# not supported by MSVC/Clang, safe to skip (we use DLLEXPORT annotations)
|
||||||
|
if(UNIX AND NOT APPLE)
|
||||||
|
set_property(TARGET hwy_test APPEND_STRING PROPERTY
|
||||||
|
LINK_FLAGS " -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/hwy/hwy.version")
|
||||||
|
endif()
|
||||||
|
|
||||||
# -------------------------------------------------------- hwy_list_targets
|
# -------------------------------------------------------- hwy_list_targets
|
||||||
# Generate a tool to print the compiled-in targets as defined by the current
|
# Generate a tool to print the compiled-in targets as defined by the current
|
||||||
|
@ -313,6 +358,7 @@ foreach (source ${HWY_SOURCES})
|
||||||
endif()
|
endif()
|
||||||
endforeach()
|
endforeach()
|
||||||
|
|
||||||
|
if (HWY_ENABLE_CONTRIB)
|
||||||
install(TARGETS hwy_contrib
|
install(TARGETS hwy_contrib
|
||||||
LIBRARY DESTINATION "${CMAKE_INSTALL_LIBDIR}"
|
LIBRARY DESTINATION "${CMAKE_INSTALL_LIBDIR}"
|
||||||
ARCHIVE DESTINATION "${CMAKE_INSTALL_LIBDIR}"
|
ARCHIVE DESTINATION "${CMAKE_INSTALL_LIBDIR}"
|
||||||
|
@ -326,6 +372,7 @@ foreach (source ${HWY_CONTRIB_SOURCES})
|
||||||
DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/${dirname}")
|
DESTINATION "${CMAKE_INSTALL_INCLUDEDIR}/${dirname}")
|
||||||
endif()
|
endif()
|
||||||
endforeach()
|
endforeach()
|
||||||
|
endif() # HWY_ENABLE_CONTRIB
|
||||||
|
|
||||||
install(TARGETS hwy_test
|
install(TARGETS hwy_test
|
||||||
LIBRARY DESTINATION "${CMAKE_INSTALL_LIBDIR}"
|
LIBRARY DESTINATION "${CMAKE_INSTALL_LIBDIR}"
|
||||||
|
@ -343,7 +390,11 @@ endforeach()
|
||||||
|
|
||||||
# Add a pkg-config file for libhwy and the contrib/test libraries.
|
# Add a pkg-config file for libhwy and the contrib/test libraries.
|
||||||
set(HWY_LIBRARY_VERSION "${CMAKE_PROJECT_VERSION}")
|
set(HWY_LIBRARY_VERSION "${CMAKE_PROJECT_VERSION}")
|
||||||
foreach (pc libhwy.pc libhwy-contrib.pc libhwy-test.pc)
|
set(HWY_PC_FILES libhwy.pc libhwy-test.pc)
|
||||||
|
if (HWY_ENABLE_CONTRIB)
|
||||||
|
list(APPEND HWY_PC_FILES libhwy-contrib.pc)
|
||||||
|
endif() # HWY_ENABLE_CONTRIB
|
||||||
|
foreach (pc ${HWY_PC_FILES})
|
||||||
configure_file("${CMAKE_CURRENT_SOURCE_DIR}/${pc}.in" "${pc}" @ONLY)
|
configure_file("${CMAKE_CURRENT_SOURCE_DIR}/${pc}.in" "${pc}" @ONLY)
|
||||||
install(FILES "${CMAKE_CURRENT_BINARY_DIR}/${pc}"
|
install(FILES "${CMAKE_CURRENT_BINARY_DIR}/${pc}"
|
||||||
DESTINATION "${CMAKE_INSTALL_LIBDIR}/pkgconfig")
|
DESTINATION "${CMAKE_INSTALL_LIBDIR}/pkgconfig")
|
||||||
|
@ -360,8 +411,8 @@ set(gtest_force_shared_crt ON CACHE BOOL "" FORCE)
|
||||||
add_executable(hwy_benchmark hwy/examples/benchmark.cc)
|
add_executable(hwy_benchmark hwy/examples/benchmark.cc)
|
||||||
target_sources(hwy_benchmark PRIVATE
|
target_sources(hwy_benchmark PRIVATE
|
||||||
hwy/nanobenchmark.h)
|
hwy/nanobenchmark.h)
|
||||||
# Try adding either -DHWY_COMPILE_ONLY_SCALAR or -DHWY_COMPILE_ONLY_STATIC to
|
# Try adding one of -DHWY_COMPILE_ONLY_SCALAR, -DHWY_COMPILE_ONLY_EMU128 or
|
||||||
# observe the difference in targets printed.
|
# -DHWY_COMPILE_ONLY_STATIC to observe the difference in targets printed.
|
||||||
target_compile_options(hwy_benchmark PRIVATE ${HWY_FLAGS})
|
target_compile_options(hwy_benchmark PRIVATE ${HWY_FLAGS})
|
||||||
target_link_libraries(hwy_benchmark hwy)
|
target_link_libraries(hwy_benchmark hwy)
|
||||||
set_target_properties(hwy_benchmark
|
set_target_properties(hwy_benchmark
|
||||||
|
@ -414,31 +465,59 @@ endif()
|
||||||
endif() # HWY_SYSTEM_GTEST
|
endif() # HWY_SYSTEM_GTEST
|
||||||
|
|
||||||
set(HWY_TEST_FILES
|
set(HWY_TEST_FILES
|
||||||
|
hwy/contrib/algo/copy_test.cc
|
||||||
|
hwy/contrib/algo/find_test.cc
|
||||||
|
hwy/contrib/algo/transform_test.cc
|
||||||
|
hwy/aligned_allocator_test.cc
|
||||||
|
hwy/base_test.cc
|
||||||
|
hwy/highway_test.cc
|
||||||
|
hwy/nanobenchmark_test.cc
|
||||||
|
hwy/targets_test.cc
|
||||||
|
hwy/examples/skeleton_test.cc
|
||||||
|
hwy/tests/arithmetic_test.cc
|
||||||
|
hwy/tests/blockwise_test.cc
|
||||||
|
hwy/tests/blockwise_shift_test.cc
|
||||||
|
hwy/tests/combine_test.cc
|
||||||
|
hwy/tests/compare_test.cc
|
||||||
|
hwy/tests/compress_test.cc
|
||||||
|
hwy/tests/convert_test.cc
|
||||||
|
hwy/tests/crypto_test.cc
|
||||||
|
hwy/tests/demote_test.cc
|
||||||
|
hwy/tests/float_test.cc
|
||||||
|
hwy/tests/if_test.cc
|
||||||
|
hwy/tests/interleaved_test.cc
|
||||||
|
hwy/tests/logical_test.cc
|
||||||
|
hwy/tests/mask_test.cc
|
||||||
|
hwy/tests/mask_mem_test.cc
|
||||||
|
hwy/tests/memory_test.cc
|
||||||
|
hwy/tests/mul_test.cc
|
||||||
|
hwy/tests/reduction_test.cc
|
||||||
|
hwy/tests/reverse_test.cc
|
||||||
|
hwy/tests/shift_test.cc
|
||||||
|
hwy/tests/swizzle_test.cc
|
||||||
|
hwy/tests/test_util_test.cc
|
||||||
|
)
|
||||||
|
|
||||||
|
set(HWY_TEST_LIBS hwy hwy_test)
|
||||||
|
|
||||||
|
if (HWY_ENABLE_CONTRIB)
|
||||||
|
list(APPEND HWY_TEST_LIBS hwy_contrib)
|
||||||
|
|
||||||
|
list(APPEND HWY_TEST_FILES
|
||||||
hwy/contrib/dot/dot_test.cc
|
hwy/contrib/dot/dot_test.cc
|
||||||
hwy/contrib/image/image_test.cc
|
hwy/contrib/image/image_test.cc
|
||||||
# Disabled due to SIGILL in clang7 debug build during gtest discovery phase,
|
# Disabled due to SIGILL in clang7 debug build during gtest discovery phase,
|
||||||
# not reproducible locally. Still tested via bazel build.
|
# not reproducible locally. Still tested via bazel build.
|
||||||
# hwy/contrib/math/math_test.cc
|
# hwy/contrib/math/math_test.cc
|
||||||
hwy/contrib/sort/sort_test.cc
|
hwy/contrib/sort/sort_test.cc
|
||||||
hwy/aligned_allocator_test.cc
|
|
||||||
hwy/base_test.cc
|
|
||||||
hwy/highway_test.cc
|
|
||||||
hwy/targets_test.cc
|
|
||||||
hwy/examples/skeleton_test.cc
|
|
||||||
hwy/tests/arithmetic_test.cc
|
|
||||||
hwy/tests/blockwise_test.cc
|
|
||||||
hwy/tests/combine_test.cc
|
|
||||||
hwy/tests/compare_test.cc
|
|
||||||
hwy/tests/convert_test.cc
|
|
||||||
hwy/tests/crypto_test.cc
|
|
||||||
hwy/tests/demote_test.cc
|
|
||||||
hwy/tests/logical_test.cc
|
|
||||||
hwy/tests/mask_test.cc
|
|
||||||
hwy/tests/memory_test.cc
|
|
||||||
hwy/tests/shift_test.cc
|
|
||||||
hwy/tests/swizzle_test.cc
|
|
||||||
hwy/tests/test_util_test.cc
|
|
||||||
)
|
)
|
||||||
|
endif() # HWY_ENABLE_CONTRIB
|
||||||
|
|
||||||
|
if(HWY_SYSTEM_GTEST)
|
||||||
|
set(HWY_GTEST_LIBS GTest::GTest GTest::Main)
|
||||||
|
else()
|
||||||
|
set(HWY_GTEST_LIBS gtest gtest_main)
|
||||||
|
endif()
|
||||||
|
|
||||||
file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/tests)
|
file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/tests)
|
||||||
foreach (TESTFILE IN LISTS HWY_TEST_FILES)
|
foreach (TESTFILE IN LISTS HWY_TEST_FILES)
|
||||||
|
@ -452,11 +531,7 @@ foreach (TESTFILE IN LISTS HWY_TEST_FILES)
|
||||||
# that include us may set them.
|
# that include us may set them.
|
||||||
target_compile_options(${TESTNAME} PRIVATE -DHWY_IS_TEST=1)
|
target_compile_options(${TESTNAME} PRIVATE -DHWY_IS_TEST=1)
|
||||||
|
|
||||||
if(HWY_SYSTEM_GTEST)
|
target_link_libraries(${TESTNAME} ${HWY_TEST_LIBS} ${HWY_GTEST_LIBS})
|
||||||
target_link_libraries(${TESTNAME} hwy hwy_contrib hwy_test GTest::GTest GTest::Main)
|
|
||||||
else()
|
|
||||||
target_link_libraries(${TESTNAME} hwy hwy_contrib hwy_test gtest gtest_main)
|
|
||||||
endif()
|
|
||||||
# Output test targets in the test directory.
|
# Output test targets in the test directory.
|
||||||
set_target_properties(${TESTNAME} PROPERTIES RUNTIME_OUTPUT_DIRECTORY "tests")
|
set_target_properties(${TESTNAME} PROPERTIES RUNTIME_OUTPUT_DIRECTORY "tests")
|
||||||
|
|
||||||
|
|
|
@ -64,7 +64,7 @@ via the below email)
|
||||||
* [iresearch database index](https://github.com/iresearch-toolkit/iresearch/blob/e7638e7a4b99136ca41f82be6edccf01351a7223/core/utils/simd_utils.hpp)
|
* [iresearch database index](https://github.com/iresearch-toolkit/iresearch/blob/e7638e7a4b99136ca41f82be6edccf01351a7223/core/utils/simd_utils.hpp)
|
||||||
* [JPEG XL image codec](https://github.com/libjxl/libjxl)
|
* [JPEG XL image codec](https://github.com/libjxl/libjxl)
|
||||||
* [Grok JPEG 2000 image codec](https://github.com/GrokImageCompression/grok)
|
* [Grok JPEG 2000 image codec](https://github.com/GrokImageCompression/grok)
|
||||||
* [vectorized Quicksort](https://github.com/google/highway/tree/master/hwy/contrib/sort)
|
* [vectorized Quicksort](https://github.com/google/highway/tree/master/hwy/contrib/sort) ([paper](https://arxiv.org/abs/2205.05982))
|
||||||
|
|
||||||
## Current status
|
## Current status
|
||||||
|
|
||||||
|
@ -72,11 +72,9 @@ via the below email)
|
||||||
|
|
||||||
Supported targets: scalar, S-SSE3, SSE4, AVX2, AVX-512, AVX3_DL (~Icelake,
|
Supported targets: scalar, S-SSE3, SSE4, AVX2, AVX-512, AVX3_DL (~Icelake,
|
||||||
requires opt-in by defining `HWY_WANT_AVX3_DL`), NEON (ARMv7 and v8), SVE, SVE2,
|
requires opt-in by defining `HWY_WANT_AVX3_DL`), NEON (ARMv7 and v8), SVE, SVE2,
|
||||||
WASM SIMD.
|
WASM SIMD, RISC-V V.
|
||||||
|
|
||||||
SVE was initially tested using farm_sve (see acknowledgments). A subset of RVV
|
SVE was initially tested using farm_sve (see acknowledgments).
|
||||||
is implemented and tested with LLVM and QEMU. Work is underway to add RVV ops
|
|
||||||
which were not yet supported by GCC.
|
|
||||||
|
|
||||||
### Versioning
|
### Versioning
|
||||||
|
|
||||||
|
@ -85,18 +83,19 @@ incrementing MINOR after backward-compatible additions and PATCH after
|
||||||
backward-compatible fixes. We recommend using releases (rather than the Git tip)
|
backward-compatible fixes. We recommend using releases (rather than the Git tip)
|
||||||
because they are tested more extensively, see below.
|
because they are tested more extensively, see below.
|
||||||
|
|
||||||
Version 0.11 is considered stable enough to use in other projects.
|
The current version 1.0 signals an increased focus on backwards compatibility.
|
||||||
Version 1.0 will signal an increased focus on backwards compatibility and will
|
Applications using documented functionality will remain compatible with future
|
||||||
be reached after the RVV target is finished (planned for 2022H1).
|
updates that have the same major version number.
|
||||||
|
|
||||||
### Testing
|
### Testing
|
||||||
|
|
||||||
Continuous integration tests build with a recent version of Clang (running on
|
Continuous integration tests build with a recent version of Clang (running on
|
||||||
x86 and QEMU for ARM) and MSVC from VS2015 (running on x86).
|
native x86, or QEMU for RVV and ARM) and MSVC 2019 (v19.28, running on native
|
||||||
|
x86).
|
||||||
|
|
||||||
Before releases, we also test on x86 with Clang and GCC, and ARMv7/8 via
|
Before releases, we also test on x86 with Clang and GCC, and ARMv7/8 via GCC
|
||||||
GCC cross-compile and QEMU. See the
|
cross-compile. See the [testing process](g3doc/release_testing_process.md) for
|
||||||
[testing process](g3doc/release_testing_process.md) for details.
|
details.
|
||||||
|
|
||||||
### Related modules
|
### Related modules
|
||||||
|
|
||||||
|
@ -148,21 +147,29 @@ portability. To obtain them, pass a `ScalableTag<float>` (or equivalently
|
||||||
`HWY_FULL(float)`) tag to functions such as `Zero/Set/Load`. There are two
|
`HWY_FULL(float)`) tag to functions such as `Zero/Set/Load`. There are two
|
||||||
alternatives for use-cases requiring an upper bound on the lanes:
|
alternatives for use-cases requiring an upper bound on the lanes:
|
||||||
|
|
||||||
- For up to a power of two `N`, specify `CappedTag<T, N>` (or
|
- For up to `N` lanes, specify `CappedTag<T, N>` or the equivalent
|
||||||
equivalently `HWY_CAPPED(T, N)`). This is useful for data structures such as
|
`HWY_CAPPED(T, N)`. The actual number of lanes will be `N` rounded down to
|
||||||
a narrow matrix. A loop is still required because vectors may actually have
|
the nearest power of two, such as 4 if `N` is 5, or 8 if `N` is 8. This is
|
||||||
fewer than `N` lanes.
|
useful for data structures such as a narrow matrix. A loop is still required
|
||||||
|
because vectors may actually have fewer than `N` lanes.
|
||||||
|
|
||||||
- For exactly a power of two `N` lanes, specify `FixedTag<T, N>`. The largest
|
- For exactly a power of two `N` lanes, specify `FixedTag<T, N>`. The largest
|
||||||
supported `N` depends on the target, but is guaranteed to be at least
|
supported `N` depends on the target, but is guaranteed to be at least
|
||||||
`16/sizeof(T)`.
|
`16/sizeof(T)`.
|
||||||
|
|
||||||
Functions using Highway must either be inside `namespace HWY_NAMESPACE {`
|
Due to ADL restrictions, user code calling Highway ops must either:
|
||||||
(possibly nested in one or more other namespaces defined by the project), OR
|
* Reside inside `namespace hwy { namespace HWY_NAMESPACE {`; or
|
||||||
each op must be prefixed with `hn::`, e.g. `namespace hn = hwy::HWY_NAMESPACE;
|
* prefix each op with an alias such as `namespace hn = hwy::HWY_NAMESPACE;
|
||||||
hn::LoadDup128()`. Additionally, each function using Highway must either be
|
hn::Add()`; or
|
||||||
prefixed with `HWY_ATTR`, OR reside between `HWY_BEFORE_NAMESPACE()` and
|
* add using-declarations for each op used: `using hwy::HWY_NAMESPACE::Add;`.
|
||||||
`HWY_AFTER_NAMESPACE()`.
|
|
||||||
|
Additionally, each function that calls Highway ops must either be prefixed with
|
||||||
|
`HWY_ATTR`, OR reside between `HWY_BEFORE_NAMESPACE()` and
|
||||||
|
`HWY_AFTER_NAMESPACE()`. Lambda functions currently require `HWY_ATTR` before
|
||||||
|
their opening brace.
|
||||||
|
|
||||||
|
The entry points into code using Highway differ slightly depending on whether
|
||||||
|
they use static or dynamic dispatch.
|
||||||
|
|
||||||
* For static dispatch, `HWY_TARGET` will be the best available target among
|
* For static dispatch, `HWY_TARGET` will be the best available target among
|
||||||
`HWY_BASELINE_TARGETS`, i.e. those allowed for use by the compiler (see
|
`HWY_BASELINE_TARGETS`, i.e. those allowed for use by the compiler (see
|
||||||
|
@ -222,12 +229,26 @@ Highway offers several ways to express loops where `N` need not divide `count`:
|
||||||
this avoids the (potentially large) cost of predication or partial
|
this avoids the (potentially large) cost of predication or partial
|
||||||
loads/stores on older targets, and does not duplicate code.
|
loads/stores on older targets, and does not duplicate code.
|
||||||
|
|
||||||
|
* Use the `Transform*` functions in hwy/contrib/algo/transform-inl.h. This
|
||||||
|
takes care of the loop and remainder handling and you simply define a
|
||||||
|
generic lambda function (C++14) or functor which receives the current vector
|
||||||
|
from the input/output array, plus optionally vectors from up to two extra
|
||||||
|
input arrays, and returns the value to write to the input/output array.
|
||||||
|
|
||||||
|
Here is an example implementing the BLAS function SAXPY (`alpha * x + y`):
|
||||||
|
|
||||||
|
```
|
||||||
|
Transform1(d, x, n, y, [](auto d, const auto v, const auto v1) HWY_ATTR {
|
||||||
|
return MulAdd(Set(d, alpha), v, v1);
|
||||||
|
});
|
||||||
|
```
|
||||||
|
|
||||||
* Process whole vectors as above, followed by a scalar loop:
|
* Process whole vectors as above, followed by a scalar loop:
|
||||||
|
|
||||||
```
|
```
|
||||||
size_t i = 0;
|
size_t i = 0;
|
||||||
for (; i + N <= count; i += N) LoopBody<false>(d, i, 0);
|
for (; i + N <= count; i += N) LoopBody<false>(d, i, 0);
|
||||||
for (; i < count; ++i) LoopBody<false>(HWY_CAPPED(T, 1)(), i, 0);
|
for (; i < count; ++i) LoopBody<false>(CappedTag<T, 1>(), i, 0);
|
||||||
```
|
```
|
||||||
The template parameter and second function arguments are again not needed.
|
The template parameter and second function arguments are again not needed.
|
||||||
|
|
||||||
|
@ -247,10 +268,14 @@ Highway offers several ways to express loops where `N` need not divide `count`:
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
Now the template parameter and third function argument can be used inside
|
Now the template parameter and third function argument can be used inside
|
||||||
`LoopBody` to 'blend' the new partial vector with previous memory contents:
|
`LoopBody` to non-atomically 'blend' the first `num_remaining` lanes of `v`
|
||||||
`Store(IfThenElse(FirstN(d, N), partial, prev_full), d, aligned_pointer);`.
|
with the previous contents of memory at subsequent locations:
|
||||||
|
`BlendedStore(v, FirstN(d, num_remaining), d, pointer);`. Similarly,
|
||||||
|
`MaskedLoad(FirstN(d, num_remaining), d, pointer)` loads the first
|
||||||
|
`num_remaining` elements and returns zero in other lanes.
|
||||||
|
|
||||||
This is a good default when it is infeasible to ensure vectors are padded.
|
This is a good default when it is infeasible to ensure vectors are padded,
|
||||||
|
but is only safe `#if !HWY_MEM_OPS_MIGHT_FAULT`!
|
||||||
In contrast to the scalar loop, only a single final iteration is needed.
|
In contrast to the scalar loop, only a single final iteration is needed.
|
||||||
The increased code size from two loop bodies is expected to be worthwhile
|
The increased code size from two loop bodies is expected to be worthwhile
|
||||||
because it avoids the cost of masking in all but the final iteration.
|
because it avoids the cost of masking in all but the final iteration.
|
||||||
|
@ -260,6 +285,7 @@ Highway offers several ways to express loops where `N` need not divide `count`:
|
||||||
* [Highway introduction (slides)](g3doc/highway_intro.pdf)
|
* [Highway introduction (slides)](g3doc/highway_intro.pdf)
|
||||||
* [Overview of instructions per operation on different architectures](g3doc/instruction_matrix.pdf)
|
* [Overview of instructions per operation on different architectures](g3doc/instruction_matrix.pdf)
|
||||||
* [Design philosophy and comparison](g3doc/design_philosophy.md)
|
* [Design philosophy and comparison](g3doc/design_philosophy.md)
|
||||||
|
* [Implementation details](g3doc/impl_details.md)
|
||||||
|
|
||||||
## Acknowledgments
|
## Acknowledgments
|
||||||
|
|
||||||
|
|
|
@ -1,3 +1,37 @@
|
||||||
|
highway (1.0.0-1) UNRELEASED; urgency=medium
|
||||||
|
|
||||||
|
* ABI change: 64-bit target values, more room for expansion
|
||||||
|
* Add CompressBlocksNot, CompressNot, Lt128Upper, Min/Max128Upper, TruncateTo
|
||||||
|
* Add HWY_SVE2_128 target
|
||||||
|
* Sort speedups especially for 128-bit
|
||||||
|
* Documentation clarifications
|
||||||
|
* Faster NEON CountTrue/FindFirstTrue/AllFalse/AllTrue
|
||||||
|
* Improved SVE codegen
|
||||||
|
* Fix u16x8 ConcatEven/Odd, SSSE3 i64 Lt
|
||||||
|
* MSVC 2017 workarounds
|
||||||
|
* Support for runtime dispatch on Arm/GCC/Linux
|
||||||
|
|
||||||
|
-- Jan Wassenberg <janwas@google.com> Wed, 27 Jul 2022 10:00:00 +0200
|
||||||
|
|
||||||
|
highway (0.17.0-1) UNRELEASED; urgency=medium
|
||||||
|
|
||||||
|
* Add ExtractLane, InsertLane, IsInf, IsFinite, IsNaN
|
||||||
|
* Add StoreInterleaved2, LoadInterleaved2/3/4, BlendedStore, SafeFillN
|
||||||
|
* Add MulFixedPoint15, Or3
|
||||||
|
* Add Copy[If], Find[If], Generate, Replace[If] algos
|
||||||
|
* Add HWY_EMU128 target (replaces HWY_SCALAR)
|
||||||
|
* HWY_RVV is feature-complete
|
||||||
|
* Add HWY_ENABLE_CONTRIB build flag, HWY_NATIVE_FMA, HWY_WANT_SSSE3/SSE4 macros
|
||||||
|
* Extend ConcatOdd/Even and StoreInterleaved* to all types
|
||||||
|
* Allow CappedTag<T, nonPowerOfTwo>
|
||||||
|
* Sort speedups: 2x for AVX2, 1.09x for AVX3; avoid x86 malloc
|
||||||
|
* Expand documentation
|
||||||
|
* Fix RDTSCP crash in nanobenchmark
|
||||||
|
* Fix XCR0 check (was ignoring AVX3 on ICL)
|
||||||
|
* Support Arm/RISC-V timers
|
||||||
|
|
||||||
|
-- Jan Wassenberg <janwas@google.com> Fri, 20 May 2022 10:00:00 +0200
|
||||||
|
|
||||||
highway (0.16.0-1) UNRELEASED; urgency=medium
|
highway (0.16.0-1) UNRELEASED; urgency=medium
|
||||||
|
|
||||||
* Add contrib/sort (vectorized quicksort)
|
* Add contrib/sort (vectorized quicksort)
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -62,8 +63,8 @@ size_t NextAlignedOffset() {
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
void* AllocateAlignedBytes(const size_t payload_size, AllocPtr alloc_ptr,
|
HWY_DLLEXPORT void* AllocateAlignedBytes(const size_t payload_size,
|
||||||
void* opaque_ptr) {
|
AllocPtr alloc_ptr, void* opaque_ptr) {
|
||||||
HWY_ASSERT(payload_size != 0); // likely a bug in caller
|
HWY_ASSERT(payload_size != 0); // likely a bug in caller
|
||||||
if (payload_size >= std::numeric_limits<size_t>::max() / 2) {
|
if (payload_size >= std::numeric_limits<size_t>::max() / 2) {
|
||||||
HWY_DASSERT(false && "payload_size too large");
|
HWY_DASSERT(false && "payload_size too large");
|
||||||
|
@ -109,8 +110,8 @@ void* AllocateAlignedBytes(const size_t payload_size, AllocPtr alloc_ptr,
|
||||||
return HWY_ASSUME_ALIGNED(reinterpret_cast<void*>(payload), kAlignment);
|
return HWY_ASSUME_ALIGNED(reinterpret_cast<void*>(payload), kAlignment);
|
||||||
}
|
}
|
||||||
|
|
||||||
void FreeAlignedBytes(const void* aligned_pointer, FreePtr free_ptr,
|
HWY_DLLEXPORT void FreeAlignedBytes(const void* aligned_pointer,
|
||||||
void* opaque_ptr) {
|
FreePtr free_ptr, void* opaque_ptr) {
|
||||||
if (aligned_pointer == nullptr) return;
|
if (aligned_pointer == nullptr) return;
|
||||||
|
|
||||||
const uintptr_t payload = reinterpret_cast<uintptr_t>(aligned_pointer);
|
const uintptr_t payload = reinterpret_cast<uintptr_t>(aligned_pointer);
|
||||||
|
@ -126,7 +127,8 @@ void FreeAlignedBytes(const void* aligned_pointer, FreePtr free_ptr,
|
||||||
}
|
}
|
||||||
|
|
||||||
// static
|
// static
|
||||||
void AlignedDeleter::DeleteAlignedArray(void* aligned_pointer, FreePtr free_ptr,
|
HWY_DLLEXPORT void AlignedDeleter::DeleteAlignedArray(void* aligned_pointer,
|
||||||
|
FreePtr free_ptr,
|
||||||
void* opaque_ptr,
|
void* opaque_ptr,
|
||||||
ArrayDeleter deleter) {
|
ArrayDeleter deleter) {
|
||||||
if (aligned_pointer == nullptr) return;
|
if (aligned_pointer == nullptr) return;
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -22,7 +23,6 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "gtest/gtest.h"
|
#include "gtest/gtest.h"
|
||||||
#include "hwy/base.h"
|
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
|
@ -69,8 +69,8 @@ class FakeAllocator {
|
||||||
void Free(void* memory) {
|
void Free(void* memory) {
|
||||||
if (!memory) return;
|
if (!memory) return;
|
||||||
EXPECT_NE(allocs_.end(), allocs_.find(memory));
|
EXPECT_NE(allocs_.end(), allocs_.find(memory));
|
||||||
free(memory);
|
|
||||||
allocs_.erase(memory);
|
allocs_.erase(memory);
|
||||||
|
free(memory);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::set<void*> allocs_;
|
std::set<void*> allocs_;
|
||||||
|
@ -276,9 +276,3 @@ TEST(AlignedAllocatorTest, DefaultInit) {
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -20,12 +21,13 @@
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
|
||||||
#include <atomic>
|
|
||||||
#include <cfloat>
|
|
||||||
|
|
||||||
#include "hwy/detect_compiler_arch.h"
|
#include "hwy/detect_compiler_arch.h"
|
||||||
#include "hwy/highway_export.h"
|
#include "hwy/highway_export.h"
|
||||||
|
|
||||||
|
#if HWY_ARCH_X86
|
||||||
|
#include <atomic>
|
||||||
|
#endif
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Compiler-specific definitions
|
// Compiler-specific definitions
|
||||||
|
|
||||||
|
@ -57,7 +59,13 @@
|
||||||
#else
|
#else
|
||||||
|
|
||||||
#define HWY_RESTRICT __restrict__
|
#define HWY_RESTRICT __restrict__
|
||||||
|
// force inlining without optimization enabled creates very inefficient code
|
||||||
|
// that can cause compiler timeout
|
||||||
|
#ifdef __OPTIMIZE__
|
||||||
#define HWY_INLINE inline __attribute__((always_inline))
|
#define HWY_INLINE inline __attribute__((always_inline))
|
||||||
|
#else
|
||||||
|
#define HWY_INLINE inline
|
||||||
|
#endif
|
||||||
#define HWY_NOINLINE __attribute__((noinline))
|
#define HWY_NOINLINE __attribute__((noinline))
|
||||||
#define HWY_FLATTEN __attribute__((flatten))
|
#define HWY_FLATTEN __attribute__((flatten))
|
||||||
#define HWY_NORETURN __attribute__((noreturn))
|
#define HWY_NORETURN __attribute__((noreturn))
|
||||||
|
@ -165,6 +173,14 @@
|
||||||
#define HWY_IS_TSAN 0
|
#define HWY_IS_TSAN 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// MSAN may cause lengthy build times or false positives e.g. in AVX3 DemoteTo.
|
||||||
|
// You can disable MSAN by adding this attribute to the function that fails.
|
||||||
|
#if HWY_IS_MSAN
|
||||||
|
#define HWY_ATTR_NO_MSAN __attribute__((no_sanitize_memory))
|
||||||
|
#else
|
||||||
|
#define HWY_ATTR_NO_MSAN
|
||||||
|
#endif
|
||||||
|
|
||||||
// For enabling HWY_DASSERT and shortening tests in slower debug builds
|
// For enabling HWY_DASSERT and shortening tests in slower debug builds
|
||||||
#if !defined(HWY_IS_DEBUG_BUILD)
|
#if !defined(HWY_IS_DEBUG_BUILD)
|
||||||
// Clang does not define NDEBUG, but it and GCC define __OPTIMIZE__, and recent
|
// Clang does not define NDEBUG, but it and GCC define __OPTIMIZE__, and recent
|
||||||
|
@ -219,19 +235,17 @@ static constexpr HWY_MAYBE_UNUSED size_t kMaxVectorSize = 16;
|
||||||
// Match [u]int##_t naming scheme so rvv-inl.h macros can obtain the type name
|
// Match [u]int##_t naming scheme so rvv-inl.h macros can obtain the type name
|
||||||
// by concatenating base type and bits.
|
// by concatenating base type and bits.
|
||||||
|
|
||||||
#if HWY_ARCH_ARM && (__ARM_FP & 2)
|
|
||||||
#define HWY_NATIVE_FLOAT16 1
|
|
||||||
#else
|
|
||||||
#define HWY_NATIVE_FLOAT16 0
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#pragma pack(push, 1)
|
#pragma pack(push, 1)
|
||||||
|
|
||||||
#if HWY_NATIVE_FLOAT16
|
// ACLE (https://gcc.gnu.org/onlinedocs/gcc/Half-Precision.html):
|
||||||
|
// always supported on aarch64, for v7 only if -mfp16-format is given.
|
||||||
|
#if ((HWY_ARCH_ARM_A64 || (__ARM_FP & 2)) && HWY_COMPILER_GCC)
|
||||||
using float16_t = __fp16;
|
using float16_t = __fp16;
|
||||||
// Clang does not allow __fp16 arguments, but scalar.h requires LaneType
|
// C11 extension ISO/IEC TS 18661-3:2015 but not supported on all targets.
|
||||||
// arguments, so use a wrapper.
|
// Required for Clang RVV if the float16 extension is used.
|
||||||
// TODO(janwas): replace with _Float16 when that is supported?
|
#elif HWY_ARCH_RVV && HWY_COMPILER_CLANG && defined(__riscv_zvfh)
|
||||||
|
using float16_t = _Float16;
|
||||||
|
// Otherwise emulate
|
||||||
#else
|
#else
|
||||||
struct float16_t {
|
struct float16_t {
|
||||||
uint16_t bits;
|
uint16_t bits;
|
||||||
|
@ -247,6 +261,44 @@ struct bfloat16_t {
|
||||||
using float32_t = float;
|
using float32_t = float;
|
||||||
using float64_t = double;
|
using float64_t = double;
|
||||||
|
|
||||||
|
#pragma pack(push, 1)
|
||||||
|
|
||||||
|
// Aligned 128-bit type. Cannot use __int128 because clang doesn't yet align it:
|
||||||
|
// https://reviews.llvm.org/D86310
|
||||||
|
struct alignas(16) uint128_t {
|
||||||
|
uint64_t lo; // little-endian layout
|
||||||
|
uint64_t hi;
|
||||||
|
};
|
||||||
|
|
||||||
|
// 64 bit key plus 64 bit value. Faster than using uint128_t when only the key
|
||||||
|
// field is to be compared (Lt128Upper instead of Lt128).
|
||||||
|
struct alignas(16) K64V64 {
|
||||||
|
uint64_t value; // little-endian layout
|
||||||
|
uint64_t key;
|
||||||
|
};
|
||||||
|
|
||||||
|
#pragma pack(pop)
|
||||||
|
|
||||||
|
static inline HWY_MAYBE_UNUSED bool operator<(const uint128_t& a,
|
||||||
|
const uint128_t& b) {
|
||||||
|
return (a.hi == b.hi) ? a.lo < b.lo : a.hi < b.hi;
|
||||||
|
}
|
||||||
|
// Required for std::greater.
|
||||||
|
static inline HWY_MAYBE_UNUSED bool operator>(const uint128_t& a,
|
||||||
|
const uint128_t& b) {
|
||||||
|
return b < a;
|
||||||
|
}
|
||||||
|
|
||||||
|
static inline HWY_MAYBE_UNUSED bool operator<(const K64V64& a,
|
||||||
|
const K64V64& b) {
|
||||||
|
return a.key < b.key;
|
||||||
|
}
|
||||||
|
// Required for std::greater.
|
||||||
|
static inline HWY_MAYBE_UNUSED bool operator>(const K64V64& a,
|
||||||
|
const K64V64& b) {
|
||||||
|
return b < a;
|
||||||
|
}
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Controlling overload resolution (SFINAE)
|
// Controlling overload resolution (SFINAE)
|
||||||
|
|
||||||
|
@ -299,6 +351,11 @@ HWY_API constexpr bool IsSame() {
|
||||||
hwy::EnableIf<sizeof(T) == (bytes)>* = nullptr
|
hwy::EnableIf<sizeof(T) == (bytes)>* = nullptr
|
||||||
#define HWY_IF_NOT_LANE_SIZE(T, bytes) \
|
#define HWY_IF_NOT_LANE_SIZE(T, bytes) \
|
||||||
hwy::EnableIf<sizeof(T) != (bytes)>* = nullptr
|
hwy::EnableIf<sizeof(T) != (bytes)>* = nullptr
|
||||||
|
#define HWY_IF_LANE_SIZE_LT(T, bytes) \
|
||||||
|
hwy::EnableIf<sizeof(T) < (bytes)>* = nullptr
|
||||||
|
|
||||||
|
#define HWY_IF_LANES_PER_BLOCK(T, N, LANES) \
|
||||||
|
hwy::EnableIf<HWY_MIN(sizeof(T) * N, 16) / sizeof(T) == (LANES)>* = nullptr
|
||||||
|
|
||||||
// Empty struct used as a size tag type.
|
// Empty struct used as a size tag type.
|
||||||
template <size_t N>
|
template <size_t N>
|
||||||
|
@ -328,12 +385,16 @@ struct Relations<uint8_t> {
|
||||||
using Unsigned = uint8_t;
|
using Unsigned = uint8_t;
|
||||||
using Signed = int8_t;
|
using Signed = int8_t;
|
||||||
using Wide = uint16_t;
|
using Wide = uint16_t;
|
||||||
|
enum { is_signed = 0 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<int8_t> {
|
struct Relations<int8_t> {
|
||||||
using Unsigned = uint8_t;
|
using Unsigned = uint8_t;
|
||||||
using Signed = int8_t;
|
using Signed = int8_t;
|
||||||
using Wide = int16_t;
|
using Wide = int16_t;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<uint16_t> {
|
struct Relations<uint16_t> {
|
||||||
|
@ -341,6 +402,8 @@ struct Relations<uint16_t> {
|
||||||
using Signed = int16_t;
|
using Signed = int16_t;
|
||||||
using Wide = uint32_t;
|
using Wide = uint32_t;
|
||||||
using Narrow = uint8_t;
|
using Narrow = uint8_t;
|
||||||
|
enum { is_signed = 0 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<int16_t> {
|
struct Relations<int16_t> {
|
||||||
|
@ -348,6 +411,8 @@ struct Relations<int16_t> {
|
||||||
using Signed = int16_t;
|
using Signed = int16_t;
|
||||||
using Wide = int32_t;
|
using Wide = int32_t;
|
||||||
using Narrow = int8_t;
|
using Narrow = int8_t;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<uint32_t> {
|
struct Relations<uint32_t> {
|
||||||
|
@ -356,6 +421,8 @@ struct Relations<uint32_t> {
|
||||||
using Float = float;
|
using Float = float;
|
||||||
using Wide = uint64_t;
|
using Wide = uint64_t;
|
||||||
using Narrow = uint16_t;
|
using Narrow = uint16_t;
|
||||||
|
enum { is_signed = 0 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<int32_t> {
|
struct Relations<int32_t> {
|
||||||
|
@ -364,13 +431,18 @@ struct Relations<int32_t> {
|
||||||
using Float = float;
|
using Float = float;
|
||||||
using Wide = int64_t;
|
using Wide = int64_t;
|
||||||
using Narrow = int16_t;
|
using Narrow = int16_t;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<uint64_t> {
|
struct Relations<uint64_t> {
|
||||||
using Unsigned = uint64_t;
|
using Unsigned = uint64_t;
|
||||||
using Signed = int64_t;
|
using Signed = int64_t;
|
||||||
using Float = double;
|
using Float = double;
|
||||||
|
using Wide = uint128_t;
|
||||||
using Narrow = uint32_t;
|
using Narrow = uint32_t;
|
||||||
|
enum { is_signed = 0 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<int64_t> {
|
struct Relations<int64_t> {
|
||||||
|
@ -378,6 +450,15 @@ struct Relations<int64_t> {
|
||||||
using Signed = int64_t;
|
using Signed = int64_t;
|
||||||
using Float = double;
|
using Float = double;
|
||||||
using Narrow = int32_t;
|
using Narrow = int32_t;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 0 };
|
||||||
|
};
|
||||||
|
template <>
|
||||||
|
struct Relations<uint128_t> {
|
||||||
|
using Unsigned = uint128_t;
|
||||||
|
using Narrow = uint64_t;
|
||||||
|
enum { is_signed = 0 };
|
||||||
|
enum { is_float = 0 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<float16_t> {
|
struct Relations<float16_t> {
|
||||||
|
@ -385,12 +466,16 @@ struct Relations<float16_t> {
|
||||||
using Signed = int16_t;
|
using Signed = int16_t;
|
||||||
using Float = float16_t;
|
using Float = float16_t;
|
||||||
using Wide = float;
|
using Wide = float;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 1 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<bfloat16_t> {
|
struct Relations<bfloat16_t> {
|
||||||
using Unsigned = uint16_t;
|
using Unsigned = uint16_t;
|
||||||
using Signed = int16_t;
|
using Signed = int16_t;
|
||||||
using Wide = float;
|
using Wide = float;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 1 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<float> {
|
struct Relations<float> {
|
||||||
|
@ -399,6 +484,8 @@ struct Relations<float> {
|
||||||
using Float = float;
|
using Float = float;
|
||||||
using Wide = double;
|
using Wide = double;
|
||||||
using Narrow = float16_t;
|
using Narrow = float16_t;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 1 };
|
||||||
};
|
};
|
||||||
template <>
|
template <>
|
||||||
struct Relations<double> {
|
struct Relations<double> {
|
||||||
|
@ -406,6 +493,8 @@ struct Relations<double> {
|
||||||
using Signed = int64_t;
|
using Signed = int64_t;
|
||||||
using Float = double;
|
using Float = double;
|
||||||
using Narrow = float;
|
using Narrow = float;
|
||||||
|
enum { is_signed = 1 };
|
||||||
|
enum { is_float = 1 };
|
||||||
};
|
};
|
||||||
|
|
||||||
template <size_t N>
|
template <size_t N>
|
||||||
|
@ -432,6 +521,10 @@ struct TypeFromSize<8> {
|
||||||
using Signed = int64_t;
|
using Signed = int64_t;
|
||||||
using Float = double;
|
using Float = double;
|
||||||
};
|
};
|
||||||
|
template <>
|
||||||
|
struct TypeFromSize<16> {
|
||||||
|
using Unsigned = uint128_t;
|
||||||
|
};
|
||||||
|
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
|
|
||||||
|
@ -457,6 +550,24 @@ using SignedFromSize = typename detail::TypeFromSize<N>::Signed;
|
||||||
template <size_t N>
|
template <size_t N>
|
||||||
using FloatFromSize = typename detail::TypeFromSize<N>::Float;
|
using FloatFromSize = typename detail::TypeFromSize<N>::Float;
|
||||||
|
|
||||||
|
// Avoid confusion with SizeTag where the parameter is a lane size.
|
||||||
|
using UnsignedTag = SizeTag<0>;
|
||||||
|
using SignedTag = SizeTag<0x100>; // integer
|
||||||
|
using FloatTag = SizeTag<0x200>;
|
||||||
|
|
||||||
|
template <typename T, class R = detail::Relations<T>>
|
||||||
|
constexpr auto TypeTag() -> hwy::SizeTag<((R::is_signed + R::is_float) << 8)> {
|
||||||
|
return hwy::SizeTag<((R::is_signed + R::is_float) << 8)>();
|
||||||
|
}
|
||||||
|
|
||||||
|
// For when we only want to distinguish FloatTag from everything else.
|
||||||
|
using NonFloatTag = SizeTag<0x400>;
|
||||||
|
|
||||||
|
template <typename T, class R = detail::Relations<T>>
|
||||||
|
constexpr auto IsFloatTag() -> hwy::SizeTag<(R::is_float ? 0x200 : 0x400)> {
|
||||||
|
return hwy::SizeTag<(R::is_float ? 0x200 : 0x400)>();
|
||||||
|
}
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Type traits
|
// Type traits
|
||||||
|
|
||||||
|
@ -502,11 +613,11 @@ HWY_API constexpr T LowestValue() {
|
||||||
}
|
}
|
||||||
template <>
|
template <>
|
||||||
constexpr float LowestValue<float>() {
|
constexpr float LowestValue<float>() {
|
||||||
return -FLT_MAX;
|
return -3.402823466e+38F;
|
||||||
}
|
}
|
||||||
template <>
|
template <>
|
||||||
constexpr double LowestValue<double>() {
|
constexpr double LowestValue<double>() {
|
||||||
return -DBL_MAX;
|
return -1.7976931348623158e+308;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -515,41 +626,51 @@ HWY_API constexpr T HighestValue() {
|
||||||
}
|
}
|
||||||
template <>
|
template <>
|
||||||
constexpr float HighestValue<float>() {
|
constexpr float HighestValue<float>() {
|
||||||
return FLT_MAX;
|
return 3.402823466e+38F;
|
||||||
}
|
}
|
||||||
template <>
|
template <>
|
||||||
constexpr double HighestValue<double>() {
|
constexpr double HighestValue<double>() {
|
||||||
return DBL_MAX;
|
return 1.7976931348623158e+308;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns width in bits of the mantissa field in IEEE binary32/64.
|
||||||
|
template <typename T>
|
||||||
|
constexpr int MantissaBits() {
|
||||||
|
static_assert(sizeof(T) == 0, "Only instantiate the specializations");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
template <>
|
||||||
|
constexpr int MantissaBits<float>() {
|
||||||
|
return 23;
|
||||||
|
}
|
||||||
|
template <>
|
||||||
|
constexpr int MantissaBits<double>() {
|
||||||
|
return 52;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns the (left-shifted by one bit) IEEE binary32/64 representation with
|
||||||
|
// the largest possible (biased) exponent field. Used by IsInf.
|
||||||
|
template <typename T>
|
||||||
|
constexpr MakeSigned<T> MaxExponentTimes2() {
|
||||||
|
return -(MakeSigned<T>{1} << (MantissaBits<T>() + 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns bitmask of the sign bit in IEEE binary32/64.
|
||||||
|
template <typename T>
|
||||||
|
constexpr MakeUnsigned<T> SignMask() {
|
||||||
|
return MakeUnsigned<T>{1} << (sizeof(T) * 8 - 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns bitmask of the exponent field in IEEE binary32/64.
|
// Returns bitmask of the exponent field in IEEE binary32/64.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
constexpr T ExponentMask() {
|
constexpr MakeUnsigned<T> ExponentMask() {
|
||||||
static_assert(sizeof(T) == 0, "Only instantiate the specializations");
|
return (~(MakeUnsigned<T>{1} << MantissaBits<T>()) + 1) & ~SignMask<T>();
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
template <>
|
|
||||||
constexpr uint32_t ExponentMask<uint32_t>() {
|
|
||||||
return 0x7F800000;
|
|
||||||
}
|
|
||||||
template <>
|
|
||||||
constexpr uint64_t ExponentMask<uint64_t>() {
|
|
||||||
return 0x7FF0000000000000ULL;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns bitmask of the mantissa field in IEEE binary32/64.
|
// Returns bitmask of the mantissa field in IEEE binary32/64.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
constexpr T MantissaMask() {
|
constexpr MakeUnsigned<T> MantissaMask() {
|
||||||
static_assert(sizeof(T) == 0, "Only instantiate the specializations");
|
return (MakeUnsigned<T>{1} << MantissaBits<T>()) - 1;
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
template <>
|
|
||||||
constexpr uint32_t MantissaMask<uint32_t>() {
|
|
||||||
return 0x007FFFFF;
|
|
||||||
}
|
|
||||||
template <>
|
|
||||||
constexpr uint64_t MantissaMask<uint64_t>() {
|
|
||||||
return 0x000FFFFFFFFFFFFFULL;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns 1 << mantissa_bits as a floating-point number. All integers whose
|
// Returns 1 << mantissa_bits as a floating-point number. All integers whose
|
||||||
|
@ -569,6 +690,21 @@ constexpr double MantissaEnd<double>() {
|
||||||
return 4503599627370496.0; // 1 << 52
|
return 4503599627370496.0; // 1 << 52
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns width in bits of the exponent field in IEEE binary32/64.
|
||||||
|
template <typename T>
|
||||||
|
constexpr int ExponentBits() {
|
||||||
|
// Exponent := remaining bits after deducting sign and mantissa.
|
||||||
|
return 8 * sizeof(T) - 1 - MantissaBits<T>();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns largest value of the biased exponent field in IEEE binary32/64,
|
||||||
|
// right-shifted so that the LSB is bit zero. Example: 0xFF for float.
|
||||||
|
// This is expressed as a signed integer for more efficient comparison.
|
||||||
|
template <typename T>
|
||||||
|
constexpr MakeSigned<T> MaxExponentField() {
|
||||||
|
return (MakeSigned<T>{1} << ExponentBits<T>()) - 1;
|
||||||
|
}
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Helper functions
|
// Helper functions
|
||||||
|
|
||||||
|
@ -602,7 +738,7 @@ HWY_API size_t Num0BitsBelowLS1Bit_Nonzero64(const uint64_t x) {
|
||||||
#else // HWY_ARCH_X86_64
|
#else // HWY_ARCH_X86_64
|
||||||
// _BitScanForward64 not available
|
// _BitScanForward64 not available
|
||||||
uint32_t lsb = static_cast<uint32_t>(x & 0xFFFFFFFF);
|
uint32_t lsb = static_cast<uint32_t>(x & 0xFFFFFFFF);
|
||||||
unsigned long index;
|
unsigned long index; // NOLINT
|
||||||
if (lsb == 0) {
|
if (lsb == 0) {
|
||||||
uint32_t msb = static_cast<uint32_t>(x >> 32u);
|
uint32_t msb = static_cast<uint32_t>(x >> 32u);
|
||||||
_BitScanForward(&index, msb);
|
_BitScanForward(&index, msb);
|
||||||
|
@ -637,7 +773,7 @@ HWY_API size_t Num0BitsAboveMS1Bit_Nonzero64(const uint64_t x) {
|
||||||
#else // HWY_ARCH_X86_64
|
#else // HWY_ARCH_X86_64
|
||||||
// _BitScanReverse64 not available
|
// _BitScanReverse64 not available
|
||||||
const uint32_t msb = static_cast<uint32_t>(x >> 32u);
|
const uint32_t msb = static_cast<uint32_t>(x >> 32u);
|
||||||
unsigned long index;
|
unsigned long index; // NOLINT
|
||||||
if (msb == 0) {
|
if (msb == 0) {
|
||||||
const uint32_t lsb = static_cast<uint32_t>(x & 0xFFFFFFFF);
|
const uint32_t lsb = static_cast<uint32_t>(x & 0xFFFFFFFF);
|
||||||
_BitScanReverse(&index, lsb);
|
_BitScanReverse(&index, lsb);
|
||||||
|
@ -653,7 +789,7 @@ HWY_API size_t Num0BitsAboveMS1Bit_Nonzero64(const uint64_t x) {
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_API size_t PopCount(uint64_t x) {
|
HWY_API size_t PopCount(uint64_t x) {
|
||||||
#if HWY_COMPILER_CLANG || HWY_COMPILER_GCC
|
#if HWY_COMPILER_GCC // includes clang
|
||||||
return static_cast<size_t>(__builtin_popcountll(x));
|
return static_cast<size_t>(__builtin_popcountll(x));
|
||||||
// This instruction has a separate feature flag, but is often called from
|
// This instruction has a separate feature flag, but is often called from
|
||||||
// non-SIMD code, so we don't want to require dynamic dispatch. It was first
|
// non-SIMD code, so we don't want to require dynamic dispatch. It was first
|
||||||
|
@ -662,7 +798,8 @@ HWY_API size_t PopCount(uint64_t x) {
|
||||||
#elif HWY_COMPILER_MSVC && HWY_ARCH_X86_64 && defined(__AVX__)
|
#elif HWY_COMPILER_MSVC && HWY_ARCH_X86_64 && defined(__AVX__)
|
||||||
return _mm_popcnt_u64(x);
|
return _mm_popcnt_u64(x);
|
||||||
#elif HWY_COMPILER_MSVC && HWY_ARCH_X86_32 && defined(__AVX__)
|
#elif HWY_COMPILER_MSVC && HWY_ARCH_X86_32 && defined(__AVX__)
|
||||||
return _mm_popcnt_u32(uint32_t(x)) + _mm_popcnt_u32(uint32_t(x >> 32));
|
return _mm_popcnt_u32(static_cast<uint32_t>(x & 0xFFFFFFFFu)) +
|
||||||
|
_mm_popcnt_u32(static_cast<uint32_t>(x >> 32));
|
||||||
#else
|
#else
|
||||||
x -= ((x >> 1) & 0x5555555555555555ULL);
|
x -= ((x >> 1) & 0x5555555555555555ULL);
|
||||||
x = (((x >> 2) & 0x3333333333333333ULL) + (x & 0x3333333333333333ULL));
|
x = (((x >> 2) & 0x3333333333333333ULL) + (x & 0x3333333333333333ULL));
|
||||||
|
@ -715,22 +852,30 @@ HWY_API uint64_t Mul128(uint64_t a, uint64_t b, uint64_t* HWY_RESTRICT upper) {
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#if HWY_COMPILER_MSVC
|
||||||
|
#pragma intrinsic(memcpy)
|
||||||
|
#pragma intrinsic(memset)
|
||||||
|
#endif
|
||||||
|
|
||||||
// The source/destination must not overlap/alias.
|
// The source/destination must not overlap/alias.
|
||||||
template <size_t kBytes, typename From, typename To>
|
template <size_t kBytes, typename From, typename To>
|
||||||
HWY_API void CopyBytes(const From* from, To* to) {
|
HWY_API void CopyBytes(const From* from, To* to) {
|
||||||
#if HWY_COMPILER_MSVC
|
#if HWY_COMPILER_MSVC
|
||||||
const uint8_t* HWY_RESTRICT from_bytes =
|
memcpy(to, from, kBytes);
|
||||||
reinterpret_cast<const uint8_t*>(from);
|
|
||||||
uint8_t* HWY_RESTRICT to_bytes = reinterpret_cast<uint8_t*>(to);
|
|
||||||
for (size_t i = 0; i < kBytes; ++i) {
|
|
||||||
to_bytes[i] = from_bytes[i];
|
|
||||||
}
|
|
||||||
#else
|
#else
|
||||||
// Avoids horrible codegen on Clang (series of PINSRB)
|
|
||||||
__builtin_memcpy(to, from, kBytes);
|
__builtin_memcpy(to, from, kBytes);
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <size_t kBytes, typename To>
|
||||||
|
HWY_API void ZeroBytes(To* to) {
|
||||||
|
#if HWY_COMPILER_MSVC
|
||||||
|
memset(to, 0, kBytes);
|
||||||
|
#else
|
||||||
|
__builtin_memset(to, 0, kBytes);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
HWY_API float F32FromBF16(bfloat16_t bf) {
|
HWY_API float F32FromBF16(bfloat16_t bf) {
|
||||||
uint32_t bits = bf.bits;
|
uint32_t bits = bf.bits;
|
||||||
bits <<= 16;
|
bits <<= 16;
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -21,7 +22,7 @@
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "base_test.cc"
|
#define HWY_TARGET_INCLUDE "base_test.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
#include "hwy/tests/test_util-inl.h"
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
|
||||||
|
@ -30,25 +31,26 @@ namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllLimits() {
|
HWY_NOINLINE void TestAllLimits() {
|
||||||
HWY_ASSERT_EQ(uint8_t(0), LimitsMin<uint8_t>());
|
HWY_ASSERT_EQ(uint8_t{0}, LimitsMin<uint8_t>());
|
||||||
HWY_ASSERT_EQ(uint16_t(0), LimitsMin<uint16_t>());
|
HWY_ASSERT_EQ(uint16_t{0}, LimitsMin<uint16_t>());
|
||||||
HWY_ASSERT_EQ(uint32_t(0), LimitsMin<uint32_t>());
|
HWY_ASSERT_EQ(uint32_t{0}, LimitsMin<uint32_t>());
|
||||||
HWY_ASSERT_EQ(uint64_t(0), LimitsMin<uint64_t>());
|
HWY_ASSERT_EQ(uint64_t{0}, LimitsMin<uint64_t>());
|
||||||
|
|
||||||
HWY_ASSERT_EQ(int8_t(-128), LimitsMin<int8_t>());
|
HWY_ASSERT_EQ(int8_t{-128}, LimitsMin<int8_t>());
|
||||||
HWY_ASSERT_EQ(int16_t(-32768), LimitsMin<int16_t>());
|
HWY_ASSERT_EQ(int16_t{-32768}, LimitsMin<int16_t>());
|
||||||
HWY_ASSERT_EQ(int32_t(0x80000000u), LimitsMin<int32_t>());
|
HWY_ASSERT_EQ(static_cast<int32_t>(0x80000000u), LimitsMin<int32_t>());
|
||||||
HWY_ASSERT_EQ(int64_t(0x8000000000000000ull), LimitsMin<int64_t>());
|
HWY_ASSERT_EQ(static_cast<int64_t>(0x8000000000000000ull),
|
||||||
|
LimitsMin<int64_t>());
|
||||||
|
|
||||||
HWY_ASSERT_EQ(uint8_t(0xFF), LimitsMax<uint8_t>());
|
HWY_ASSERT_EQ(uint8_t{0xFF}, LimitsMax<uint8_t>());
|
||||||
HWY_ASSERT_EQ(uint16_t(0xFFFF), LimitsMax<uint16_t>());
|
HWY_ASSERT_EQ(uint16_t{0xFFFF}, LimitsMax<uint16_t>());
|
||||||
HWY_ASSERT_EQ(uint32_t(0xFFFFFFFFu), LimitsMax<uint32_t>());
|
HWY_ASSERT_EQ(uint32_t{0xFFFFFFFFu}, LimitsMax<uint32_t>());
|
||||||
HWY_ASSERT_EQ(uint64_t(0xFFFFFFFFFFFFFFFFull), LimitsMax<uint64_t>());
|
HWY_ASSERT_EQ(uint64_t{0xFFFFFFFFFFFFFFFFull}, LimitsMax<uint64_t>());
|
||||||
|
|
||||||
HWY_ASSERT_EQ(int8_t(0x7F), LimitsMax<int8_t>());
|
HWY_ASSERT_EQ(int8_t{0x7F}, LimitsMax<int8_t>());
|
||||||
HWY_ASSERT_EQ(int16_t(0x7FFF), LimitsMax<int16_t>());
|
HWY_ASSERT_EQ(int16_t{0x7FFF}, LimitsMax<int16_t>());
|
||||||
HWY_ASSERT_EQ(int32_t(0x7FFFFFFFu), LimitsMax<int32_t>());
|
HWY_ASSERT_EQ(int32_t{0x7FFFFFFFu}, LimitsMax<int32_t>());
|
||||||
HWY_ASSERT_EQ(int64_t(0x7FFFFFFFFFFFFFFFull), LimitsMax<int64_t>());
|
HWY_ASSERT_EQ(int64_t{0x7FFFFFFFFFFFFFFFull}, LimitsMax<int64_t>());
|
||||||
}
|
}
|
||||||
|
|
||||||
struct TestLowestHighest {
|
struct TestLowestHighest {
|
||||||
|
@ -88,6 +90,10 @@ HWY_NOINLINE void TestAllType() {
|
||||||
ForUnsignedTypes(TestIsUnsigned());
|
ForUnsignedTypes(TestIsUnsigned());
|
||||||
ForSignedTypes(TestIsSigned());
|
ForSignedTypes(TestIsSigned());
|
||||||
ForFloatTypes(TestIsFloat());
|
ForFloatTypes(TestIsFloat());
|
||||||
|
|
||||||
|
static_assert(sizeof(MakeUnsigned<hwy::uint128_t>) == 16, "");
|
||||||
|
static_assert(sizeof(MakeWide<uint64_t>) == 16, "Expected uint128_t");
|
||||||
|
static_assert(sizeof(MakeNarrow<hwy::uint128_t>) == 8, "Expected uint64_t");
|
||||||
}
|
}
|
||||||
|
|
||||||
struct TestIsSame {
|
struct TestIsSame {
|
||||||
|
@ -102,54 +108,54 @@ struct TestIsSame {
|
||||||
HWY_NOINLINE void TestAllIsSame() { ForAllTypes(TestIsSame()); }
|
HWY_NOINLINE void TestAllIsSame() { ForAllTypes(TestIsSame()); }
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllBitScan() {
|
HWY_NOINLINE void TestAllBitScan() {
|
||||||
HWY_ASSERT_EQ(size_t(0), Num0BitsAboveMS1Bit_Nonzero32(0x80000000u));
|
HWY_ASSERT_EQ(size_t{0}, Num0BitsAboveMS1Bit_Nonzero32(0x80000000u));
|
||||||
HWY_ASSERT_EQ(size_t(0), Num0BitsAboveMS1Bit_Nonzero32(0xFFFFFFFFu));
|
HWY_ASSERT_EQ(size_t{0}, Num0BitsAboveMS1Bit_Nonzero32(0xFFFFFFFFu));
|
||||||
HWY_ASSERT_EQ(size_t(1), Num0BitsAboveMS1Bit_Nonzero32(0x40000000u));
|
HWY_ASSERT_EQ(size_t{1}, Num0BitsAboveMS1Bit_Nonzero32(0x40000000u));
|
||||||
HWY_ASSERT_EQ(size_t(1), Num0BitsAboveMS1Bit_Nonzero32(0x40108210u));
|
HWY_ASSERT_EQ(size_t{1}, Num0BitsAboveMS1Bit_Nonzero32(0x40108210u));
|
||||||
HWY_ASSERT_EQ(size_t(30), Num0BitsAboveMS1Bit_Nonzero32(2u));
|
HWY_ASSERT_EQ(size_t{30}, Num0BitsAboveMS1Bit_Nonzero32(2u));
|
||||||
HWY_ASSERT_EQ(size_t(30), Num0BitsAboveMS1Bit_Nonzero32(3u));
|
HWY_ASSERT_EQ(size_t{30}, Num0BitsAboveMS1Bit_Nonzero32(3u));
|
||||||
HWY_ASSERT_EQ(size_t(31), Num0BitsAboveMS1Bit_Nonzero32(1u));
|
HWY_ASSERT_EQ(size_t{31}, Num0BitsAboveMS1Bit_Nonzero32(1u));
|
||||||
|
|
||||||
HWY_ASSERT_EQ(size_t(0),
|
HWY_ASSERT_EQ(size_t{0},
|
||||||
Num0BitsAboveMS1Bit_Nonzero64(0x8000000000000000ull));
|
Num0BitsAboveMS1Bit_Nonzero64(0x8000000000000000ull));
|
||||||
HWY_ASSERT_EQ(size_t(0),
|
HWY_ASSERT_EQ(size_t{0},
|
||||||
Num0BitsAboveMS1Bit_Nonzero64(0xFFFFFFFFFFFFFFFFull));
|
Num0BitsAboveMS1Bit_Nonzero64(0xFFFFFFFFFFFFFFFFull));
|
||||||
HWY_ASSERT_EQ(size_t(1),
|
HWY_ASSERT_EQ(size_t{1},
|
||||||
Num0BitsAboveMS1Bit_Nonzero64(0x4000000000000000ull));
|
Num0BitsAboveMS1Bit_Nonzero64(0x4000000000000000ull));
|
||||||
HWY_ASSERT_EQ(size_t(1),
|
HWY_ASSERT_EQ(size_t{1},
|
||||||
Num0BitsAboveMS1Bit_Nonzero64(0x4010821004200011ull));
|
Num0BitsAboveMS1Bit_Nonzero64(0x4010821004200011ull));
|
||||||
HWY_ASSERT_EQ(size_t(62), Num0BitsAboveMS1Bit_Nonzero64(2ull));
|
HWY_ASSERT_EQ(size_t{62}, Num0BitsAboveMS1Bit_Nonzero64(2ull));
|
||||||
HWY_ASSERT_EQ(size_t(62), Num0BitsAboveMS1Bit_Nonzero64(3ull));
|
HWY_ASSERT_EQ(size_t{62}, Num0BitsAboveMS1Bit_Nonzero64(3ull));
|
||||||
HWY_ASSERT_EQ(size_t(63), Num0BitsAboveMS1Bit_Nonzero64(1ull));
|
HWY_ASSERT_EQ(size_t{63}, Num0BitsAboveMS1Bit_Nonzero64(1ull));
|
||||||
|
|
||||||
HWY_ASSERT_EQ(size_t(0), Num0BitsBelowLS1Bit_Nonzero32(1u));
|
HWY_ASSERT_EQ(size_t{0}, Num0BitsBelowLS1Bit_Nonzero32(1u));
|
||||||
HWY_ASSERT_EQ(size_t(1), Num0BitsBelowLS1Bit_Nonzero32(2u));
|
HWY_ASSERT_EQ(size_t{1}, Num0BitsBelowLS1Bit_Nonzero32(2u));
|
||||||
HWY_ASSERT_EQ(size_t(30), Num0BitsBelowLS1Bit_Nonzero32(0xC0000000u));
|
HWY_ASSERT_EQ(size_t{30}, Num0BitsBelowLS1Bit_Nonzero32(0xC0000000u));
|
||||||
HWY_ASSERT_EQ(size_t(31), Num0BitsBelowLS1Bit_Nonzero32(0x80000000u));
|
HWY_ASSERT_EQ(size_t{31}, Num0BitsBelowLS1Bit_Nonzero32(0x80000000u));
|
||||||
|
|
||||||
HWY_ASSERT_EQ(size_t(0), Num0BitsBelowLS1Bit_Nonzero64(1ull));
|
HWY_ASSERT_EQ(size_t{0}, Num0BitsBelowLS1Bit_Nonzero64(1ull));
|
||||||
HWY_ASSERT_EQ(size_t(1), Num0BitsBelowLS1Bit_Nonzero64(2ull));
|
HWY_ASSERT_EQ(size_t{1}, Num0BitsBelowLS1Bit_Nonzero64(2ull));
|
||||||
HWY_ASSERT_EQ(size_t(62),
|
HWY_ASSERT_EQ(size_t{62},
|
||||||
Num0BitsBelowLS1Bit_Nonzero64(0xC000000000000000ull));
|
Num0BitsBelowLS1Bit_Nonzero64(0xC000000000000000ull));
|
||||||
HWY_ASSERT_EQ(size_t(63),
|
HWY_ASSERT_EQ(size_t{63},
|
||||||
Num0BitsBelowLS1Bit_Nonzero64(0x8000000000000000ull));
|
Num0BitsBelowLS1Bit_Nonzero64(0x8000000000000000ull));
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllPopCount() {
|
HWY_NOINLINE void TestAllPopCount() {
|
||||||
HWY_ASSERT_EQ(size_t(0), PopCount(0u));
|
HWY_ASSERT_EQ(size_t{0}, PopCount(0u));
|
||||||
HWY_ASSERT_EQ(size_t(1), PopCount(1u));
|
HWY_ASSERT_EQ(size_t{1}, PopCount(1u));
|
||||||
HWY_ASSERT_EQ(size_t(1), PopCount(2u));
|
HWY_ASSERT_EQ(size_t{1}, PopCount(2u));
|
||||||
HWY_ASSERT_EQ(size_t(2), PopCount(3u));
|
HWY_ASSERT_EQ(size_t{2}, PopCount(3u));
|
||||||
HWY_ASSERT_EQ(size_t(1), PopCount(0x80000000u));
|
HWY_ASSERT_EQ(size_t{1}, PopCount(0x80000000u));
|
||||||
HWY_ASSERT_EQ(size_t(31), PopCount(0x7FFFFFFFu));
|
HWY_ASSERT_EQ(size_t{31}, PopCount(0x7FFFFFFFu));
|
||||||
HWY_ASSERT_EQ(size_t(32), PopCount(0xFFFFFFFFu));
|
HWY_ASSERT_EQ(size_t{32}, PopCount(0xFFFFFFFFu));
|
||||||
|
|
||||||
HWY_ASSERT_EQ(size_t(1), PopCount(0x80000000ull));
|
HWY_ASSERT_EQ(size_t{1}, PopCount(0x80000000ull));
|
||||||
HWY_ASSERT_EQ(size_t(31), PopCount(0x7FFFFFFFull));
|
HWY_ASSERT_EQ(size_t{31}, PopCount(0x7FFFFFFFull));
|
||||||
HWY_ASSERT_EQ(size_t(32), PopCount(0xFFFFFFFFull));
|
HWY_ASSERT_EQ(size_t{32}, PopCount(0xFFFFFFFFull));
|
||||||
HWY_ASSERT_EQ(size_t(33), PopCount(0x10FFFFFFFFull));
|
HWY_ASSERT_EQ(size_t{33}, PopCount(0x10FFFFFFFFull));
|
||||||
HWY_ASSERT_EQ(size_t(63), PopCount(0xFFFEFFFFFFFFFFFFull));
|
HWY_ASSERT_EQ(size_t{63}, PopCount(0xFFFEFFFFFFFFFFFFull));
|
||||||
HWY_ASSERT_EQ(size_t(64), PopCount(0xFFFFFFFFFFFFFFFFull));
|
HWY_ASSERT_EQ(size_t{64}, PopCount(0xFFFFFFFFFFFFFFFFull));
|
||||||
}
|
}
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
@ -169,10 +175,4 @@ HWY_EXPORT_AND_TEST_P(BaseTest, TestAllBitScan);
|
||||||
HWY_EXPORT_AND_TEST_P(BaseTest, TestAllPopCount);
|
HWY_EXPORT_AND_TEST_P(BaseTest, TestAllPopCount);
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char **argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -50,10 +51,10 @@ namespace hwy {
|
||||||
#define HWY_ATTR_CACHE
|
#define HWY_ATTR_CACHE
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Delays subsequent loads until prior loads are visible. On Intel CPUs, also
|
// Delays subsequent loads until prior loads are visible. Beware of potentially
|
||||||
// serves as a full fence (waits for all prior instructions to complete).
|
// differing behavior across architectures and vendors: on Intel but not
|
||||||
// No effect on non-x86.
|
// AMD CPUs, also serves as a full fence (waits for all prior instructions to
|
||||||
// DEPRECATED due to differing behavior across architectures AND vendors.
|
// complete).
|
||||||
HWY_INLINE HWY_ATTR_CACHE void LoadFence() {
|
HWY_INLINE HWY_ATTR_CACHE void LoadFence() {
|
||||||
#if HWY_ARCH_X86 && !defined(HWY_DISABLE_CACHE_CONTROL)
|
#if HWY_ARCH_X86 && !defined(HWY_DISABLE_CACHE_CONTROL)
|
||||||
_mm_lfence();
|
_mm_lfence();
|
||||||
|
@ -76,7 +77,7 @@ template <typename T>
|
||||||
HWY_INLINE HWY_ATTR_CACHE void Prefetch(const T* p) {
|
HWY_INLINE HWY_ATTR_CACHE void Prefetch(const T* p) {
|
||||||
#if HWY_ARCH_X86 && !defined(HWY_DISABLE_CACHE_CONTROL)
|
#if HWY_ARCH_X86 && !defined(HWY_DISABLE_CACHE_CONTROL)
|
||||||
_mm_prefetch(reinterpret_cast<const char*>(p), _MM_HINT_T0);
|
_mm_prefetch(reinterpret_cast<const char*>(p), _MM_HINT_T0);
|
||||||
#elif HWY_COMPILER_GCC || HWY_COMPILER_CLANG
|
#elif HWY_COMPILER_GCC // includes clang
|
||||||
// Hint=0 (NTA) behavior differs, but skipping outer caches is probably not
|
// Hint=0 (NTA) behavior differs, but skipping outer caches is probably not
|
||||||
// desirable, so use the default 3 (keep in caches).
|
// desirable, so use the default 3 (keep in caches).
|
||||||
__builtin_prefetch(p, /*write=*/0, /*hint=*/3);
|
__builtin_prefetch(p, /*write=*/0, /*hint=*/3);
|
||||||
|
|
|
@ -0,0 +1,138 @@
|
||||||
|
// Copyright 2022 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.
|
||||||
|
|
||||||
|
// Per-target include guard
|
||||||
|
#if defined(HIGHWAY_HWY_CONTRIB_ALGO_COPY_INL_H_) == \
|
||||||
|
defined(HWY_TARGET_TOGGLE)
|
||||||
|
#ifdef HIGHWAY_HWY_CONTRIB_ALGO_COPY_INL_H_
|
||||||
|
#undef HIGHWAY_HWY_CONTRIB_ALGO_COPY_INL_H_
|
||||||
|
#else
|
||||||
|
#define HIGHWAY_HWY_CONTRIB_ALGO_COPY_INL_H_
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <string.h> // memcpy
|
||||||
|
|
||||||
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
// These functions avoid having to write a loop plus remainder handling in the
|
||||||
|
// (unfortunately still common) case where arrays are not aligned/padded. If the
|
||||||
|
// inputs are known to be aligned/padded, it is more efficient to write a single
|
||||||
|
// loop using Load(). We do not provide a CopyAlignedPadded because it
|
||||||
|
// would be more verbose than such a loop.
|
||||||
|
|
||||||
|
// Fills `to`[0, `count`) with `value`.
|
||||||
|
template <class D, typename T = TFromD<D>>
|
||||||
|
void Fill(D d, T value, size_t count, T* HWY_RESTRICT to) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const Vec<D> v = Set(d, value);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
StoreU(v, d, to + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
SafeFillN(remaining, value, d, to + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Copies `from`[0, `count`) to `to`, which must not overlap `from`.
|
||||||
|
template <class D, typename T = TFromD<D>>
|
||||||
|
void Copy(D d, const T* HWY_RESTRICT from, size_t count, T* HWY_RESTRICT to) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
const Vec<D> v = LoadU(d, from + idx);
|
||||||
|
StoreU(v, d, to + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
SafeCopyN(remaining, d, from + idx, to + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// For idx in [0, count) in ascending order, appends `from[idx]` to `to` if the
|
||||||
|
// corresponding mask element of `func(d, v)` is true. Returns the STL-style end
|
||||||
|
// of the newly written elements in `to`.
|
||||||
|
//
|
||||||
|
// `func` is either a functor with a templated operator()(d, v) returning a
|
||||||
|
// mask, or a generic lambda if using C++14. Due to apparent limitations of
|
||||||
|
// Clang on Windows, it is currently necessary to add HWY_ATTR before the
|
||||||
|
// opening { of the lambda to avoid errors about "function .. requires target".
|
||||||
|
//
|
||||||
|
// NOTE: this is only supported for 16-, 32- or 64-bit types.
|
||||||
|
// NOTE: Func may be called a second time for elements it has already seen, but
|
||||||
|
// these elements will not be written to `to` again.
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
T* CopyIf(D d, const T* HWY_RESTRICT from, size_t count, T* HWY_RESTRICT to,
|
||||||
|
const Func& func) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
const Vec<D> v = LoadU(d, from + idx);
|
||||||
|
to += CompressBlendedStore(v, func(d, v), d, to);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return to;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
// Workaround for -Waggressive-loop-optimizations on GCC 8
|
||||||
|
// (iteration 2305843009213693951 invokes undefined behavior for T=i64)
|
||||||
|
const uintptr_t addr = reinterpret_cast<uintptr_t>(from);
|
||||||
|
const T* HWY_RESTRICT from_idx =
|
||||||
|
reinterpret_cast<const T * HWY_RESTRICT>(addr + (idx * sizeof(T)));
|
||||||
|
const V1 v = LoadU(d1, from_idx);
|
||||||
|
// Avoid storing to `to` unless we know it should be kept - otherwise, we
|
||||||
|
// might overrun the end if it was allocated for the exact count.
|
||||||
|
if (CountTrue(d1, func(d1, v)) == 0) continue;
|
||||||
|
StoreU(v, d1, to);
|
||||||
|
to += 1;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
// Start index of the last unaligned whole vector, ending at the array end.
|
||||||
|
const size_t last = count - N;
|
||||||
|
// Number of elements before `from` or already written.
|
||||||
|
const size_t invalid = idx - last;
|
||||||
|
HWY_DASSERT(0 != invalid && invalid < N);
|
||||||
|
const Mask<D> mask = Not(FirstN(d, invalid));
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, from + last);
|
||||||
|
to += CompressBlendedStore(v, And(mask, func(d, v)), d, to);
|
||||||
|
#endif
|
||||||
|
return to;
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#endif // HIGHWAY_HWY_CONTRIB_ALGO_COPY_INL_H_
|
|
@ -0,0 +1,199 @@
|
||||||
|
// Copyright 2022 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/aligned_allocator.h"
|
||||||
|
|
||||||
|
// clang-format off
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/algo/copy_test.cc"
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
#include "hwy/contrib/algo/copy-inl.h"
|
||||||
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
// clang-format on
|
||||||
|
|
||||||
|
// If your project requires C++14 or later, you can ignore this and pass lambdas
|
||||||
|
// directly to Transform, without requiring an lvalue as we do here for C++11.
|
||||||
|
#if __cplusplus < 201402L
|
||||||
|
#define HWY_GENERIC_LAMBDA 0
|
||||||
|
#else
|
||||||
|
#define HWY_GENERIC_LAMBDA 1
|
||||||
|
#endif
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
// Returns random integer in [0, 128), which fits in any lane type.
|
||||||
|
template <typename T>
|
||||||
|
T Random7Bit(RandomState& rng) {
|
||||||
|
return static_cast<T>(Random32(&rng) & 127);
|
||||||
|
}
|
||||||
|
|
||||||
|
// In C++14, we can instead define these as generic lambdas next to where they
|
||||||
|
// are invoked.
|
||||||
|
#if !HWY_GENERIC_LAMBDA
|
||||||
|
|
||||||
|
struct IsOdd {
|
||||||
|
template <class D, class V>
|
||||||
|
Mask<D> operator()(D d, V v) const {
|
||||||
|
return TestBit(v, Set(d, TFromD<D>{1}));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // !HWY_GENERIC_LAMBDA
|
||||||
|
|
||||||
|
// Invokes Test (e.g. TestCopyIf) with all arg combinations. T comes from
|
||||||
|
// ForFloatTypes.
|
||||||
|
template <class Test>
|
||||||
|
struct ForeachCountAndMisalign {
|
||||||
|
template <typename T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) const {
|
||||||
|
RandomState rng;
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const size_t misalignments[3] = {0, N / 4, 3 * N / 5};
|
||||||
|
|
||||||
|
for (size_t count = 0; count < 2 * N; ++count) {
|
||||||
|
for (size_t ma : misalignments) {
|
||||||
|
for (size_t mb : misalignments) {
|
||||||
|
Test()(d, count, ma, mb, rng);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct TestFill {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// HWY_MAX prevents error when misalign == count == 0.
|
||||||
|
AlignedFreeUniquePtr<T[]> pa =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
T* expected = pa.get() + misalign_a;
|
||||||
|
const T value = Random7Bit<T>(rng);
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
expected[i] = value;
|
||||||
|
}
|
||||||
|
AlignedFreeUniquePtr<T[]> pb = AllocateAligned<T>(misalign_b + count + 1);
|
||||||
|
T* actual = pb.get() + misalign_b;
|
||||||
|
|
||||||
|
actual[count] = T{0}; // sentinel
|
||||||
|
Fill(d, value, count, actual);
|
||||||
|
HWY_ASSERT_EQ(T{0}, actual[count]); // did not write past end
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, expected, actual, count, target_name,
|
||||||
|
__FILE__, __LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TestAllFill() {
|
||||||
|
ForAllTypes(ForPartialVectors<ForeachCountAndMisalign<TestFill>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TestCopy {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// Prevents error if size to allocate is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> pa =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
T* a = pa.get() + misalign_a;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
a[i] = Random7Bit<T>(rng);
|
||||||
|
}
|
||||||
|
AlignedFreeUniquePtr<T[]> pb =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_b + count));
|
||||||
|
T* b = pb.get() + misalign_b;
|
||||||
|
|
||||||
|
Copy(d, a, count, b);
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, a, b, count, target_name, __FILE__,
|
||||||
|
__LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TestAllCopy() {
|
||||||
|
ForAllTypes(ForPartialVectors<ForeachCountAndMisalign<TestCopy>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TestCopyIf {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// Prevents error if size to allocate is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> pa =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
T* a = pa.get() + misalign_a;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
a[i] = Random7Bit<T>(rng);
|
||||||
|
}
|
||||||
|
const size_t padding = Lanes(ScalableTag<T>());
|
||||||
|
AlignedFreeUniquePtr<T[]> pb =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_b + count + padding));
|
||||||
|
T* b = pb.get() + misalign_b;
|
||||||
|
|
||||||
|
AlignedFreeUniquePtr<T[]> expected = AllocateAligned<T>(HWY_MAX(1, count));
|
||||||
|
size_t num_odd = 0;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
if (a[i] & 1) {
|
||||||
|
expected[num_odd++] = a[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#if HWY_GENERIC_LAMBDA
|
||||||
|
const auto is_odd = [](const auto d, const auto v) HWY_ATTR {
|
||||||
|
return TestBit(v, Set(d, TFromD<decltype(d)>{1}));
|
||||||
|
};
|
||||||
|
#else
|
||||||
|
const IsOdd is_odd;
|
||||||
|
#endif
|
||||||
|
T* end = CopyIf(d, a, count, b, is_odd);
|
||||||
|
const size_t num_written = static_cast<size_t>(end - b);
|
||||||
|
HWY_ASSERT_EQ(num_odd, num_written);
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, expected.get(), b, num_odd, target_name,
|
||||||
|
__FILE__, __LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TestAllCopyIf() {
|
||||||
|
ForUI163264(ForPartialVectors<ForeachCountAndMisalign<TestCopyIf>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#if HWY_ONCE
|
||||||
|
|
||||||
|
namespace hwy {
|
||||||
|
HWY_BEFORE_TEST(CopyTest);
|
||||||
|
HWY_EXPORT_AND_TEST_P(CopyTest, TestAllFill);
|
||||||
|
HWY_EXPORT_AND_TEST_P(CopyTest, TestAllCopy);
|
||||||
|
HWY_EXPORT_AND_TEST_P(CopyTest, TestAllCopyIf);
|
||||||
|
} // namespace hwy
|
||||||
|
|
||||||
|
#endif
|
|
@ -0,0 +1,109 @@
|
||||||
|
// Copyright 2022 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.
|
||||||
|
|
||||||
|
// Per-target include guard
|
||||||
|
#if defined(HIGHWAY_HWY_CONTRIB_ALGO_FIND_INL_H_) == \
|
||||||
|
defined(HWY_TARGET_TOGGLE)
|
||||||
|
#ifdef HIGHWAY_HWY_CONTRIB_ALGO_FIND_INL_H_
|
||||||
|
#undef HIGHWAY_HWY_CONTRIB_ALGO_FIND_INL_H_
|
||||||
|
#else
|
||||||
|
#define HIGHWAY_HWY_CONTRIB_ALGO_FIND_INL_H_
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
// Returns index of the first element equal to `value` in `in[0, count)`, or
|
||||||
|
// `count` if not found.
|
||||||
|
template <class D, typename T = TFromD<D>>
|
||||||
|
size_t Find(D d, T value, const T* HWY_RESTRICT in, size_t count) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const Vec<D> broadcasted = Set(d, value);
|
||||||
|
|
||||||
|
size_t i = 0;
|
||||||
|
for (; i + N <= count; i += N) {
|
||||||
|
const intptr_t pos = FindFirstTrue(d, Eq(broadcasted, LoadU(d, in + i)));
|
||||||
|
if (pos >= 0) return i + static_cast<size_t>(pos);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (i != count) {
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Scan single elements.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
const V1 broadcasted1 = Set(d1, GetLane(broadcasted));
|
||||||
|
for (; i < count; ++i) {
|
||||||
|
if (AllTrue(d1, Eq(broadcasted1, LoadU(d1, in + i)))) {
|
||||||
|
return i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - i;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, in + i);
|
||||||
|
// Apply mask so that we don't 'find' the zero-padding from MaskedLoad.
|
||||||
|
const intptr_t pos = FindFirstTrue(d, And(Eq(broadcasted, v), mask));
|
||||||
|
if (pos >= 0) return i + static_cast<size_t>(pos);
|
||||||
|
#endif // HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
}
|
||||||
|
|
||||||
|
return count; // not found
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns index of the first element in `in[0, count)` for which `func(d, vec)`
|
||||||
|
// returns true, otherwise `count`.
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
size_t FindIf(D d, const T* HWY_RESTRICT in, size_t count, const Func& func) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t i = 0;
|
||||||
|
for (; i + N <= count; i += N) {
|
||||||
|
const intptr_t pos = FindFirstTrue(d, func(d, LoadU(d, in + i)));
|
||||||
|
if (pos >= 0) return i + static_cast<size_t>(pos);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (i != count) {
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Scan single elements.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
for (; i < count; ++i) {
|
||||||
|
if (AllTrue(d1, func(d1, LoadU(d1, in + i)))) {
|
||||||
|
return i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - i;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, in + i);
|
||||||
|
// Apply mask so that we don't 'find' the zero-padding from MaskedLoad.
|
||||||
|
const intptr_t pos = FindFirstTrue(d, And(func(d, v), mask));
|
||||||
|
if (pos >= 0) return i + static_cast<size_t>(pos);
|
||||||
|
#endif // HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
}
|
||||||
|
|
||||||
|
return count; // not found
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#endif // HIGHWAY_HWY_CONTRIB_ALGO_FIND_INL_H_
|
|
@ -0,0 +1,219 @@
|
||||||
|
// Copyright 2022 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 <algorithm>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "hwy/aligned_allocator.h"
|
||||||
|
#include "hwy/base.h"
|
||||||
|
#include "hwy/print.h"
|
||||||
|
|
||||||
|
// clang-format off
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/algo/find_test.cc"
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
#include "hwy/contrib/algo/find-inl.h"
|
||||||
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
// clang-format on
|
||||||
|
|
||||||
|
// If your project requires C++14 or later, you can ignore this and pass lambdas
|
||||||
|
// directly to FindIf, without requiring an lvalue as we do here for C++11.
|
||||||
|
#if __cplusplus < 201402L
|
||||||
|
#define HWY_GENERIC_LAMBDA 0
|
||||||
|
#else
|
||||||
|
#define HWY_GENERIC_LAMBDA 1
|
||||||
|
#endif
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
// Returns random number in [-8, 8) - we use knowledge of the range to Find()
|
||||||
|
// values we know are not present.
|
||||||
|
template <typename T>
|
||||||
|
T Random(RandomState& rng) {
|
||||||
|
const int32_t bits = static_cast<int32_t>(Random32(&rng)) & 1023;
|
||||||
|
const double val = (bits - 512) / 64.0;
|
||||||
|
// Clamp negative to zero for unsigned types.
|
||||||
|
return static_cast<T>(HWY_MAX(hwy::LowestValue<T>(), val));
|
||||||
|
}
|
||||||
|
|
||||||
|
// In C++14, we can instead define these as generic lambdas next to where they
|
||||||
|
// are invoked.
|
||||||
|
#if !HWY_GENERIC_LAMBDA
|
||||||
|
|
||||||
|
class GreaterThan {
|
||||||
|
public:
|
||||||
|
GreaterThan(int val) : val_(val) {}
|
||||||
|
template <class D, class V>
|
||||||
|
Mask<D> operator()(D d, V v) const {
|
||||||
|
return Gt(v, Set(d, static_cast<TFromD<D>>(val_)));
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
int val_;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // !HWY_GENERIC_LAMBDA
|
||||||
|
|
||||||
|
// Invokes Test (e.g. TestFind) with all arg combinations.
|
||||||
|
template <class Test>
|
||||||
|
struct ForeachCountAndMisalign {
|
||||||
|
template <typename T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) const {
|
||||||
|
RandomState rng;
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const size_t misalignments[3] = {0, N / 4, 3 * N / 5};
|
||||||
|
|
||||||
|
// Find() checks 8 vectors at a time, so we want to cover a fairly large
|
||||||
|
// range without oversampling (checking every possible count).
|
||||||
|
std::vector<size_t> counts(AdjustedReps(512));
|
||||||
|
for (size_t& count : counts) {
|
||||||
|
count = static_cast<size_t>(rng()) % (16 * N + 1);
|
||||||
|
}
|
||||||
|
counts[0] = 0; // ensure we test count=0.
|
||||||
|
|
||||||
|
for (size_t count : counts) {
|
||||||
|
for (size_t m : misalignments) {
|
||||||
|
Test()(d, count, m, rng);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct TestFind {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign, RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// Must allocate at least one even if count is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> storage =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign + count));
|
||||||
|
T* in = storage.get() + misalign;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
in[i] = Random<T>(rng);
|
||||||
|
}
|
||||||
|
|
||||||
|
// For each position, search for that element (which we know is there)
|
||||||
|
for (size_t pos = 0; pos < count; ++pos) {
|
||||||
|
const size_t actual = Find(d, in[pos], in, count);
|
||||||
|
|
||||||
|
// We may have found an earlier occurrence of the same value; ensure the
|
||||||
|
// value is the same, and that it is the first.
|
||||||
|
if (!IsEqual(in[pos], in[actual])) {
|
||||||
|
fprintf(stderr, "%s count %d, found %.15f at %d but wanted %.15f\n",
|
||||||
|
hwy::TypeName(T(), Lanes(d)).c_str(), static_cast<int>(count),
|
||||||
|
static_cast<double>(in[actual]), static_cast<int>(actual),
|
||||||
|
static_cast<double>(in[pos]));
|
||||||
|
HWY_ASSERT(false);
|
||||||
|
}
|
||||||
|
for (size_t i = 0; i < actual; ++i) {
|
||||||
|
if (IsEqual(in[i], in[pos])) {
|
||||||
|
fprintf(stderr, "%s count %d, found %f at %d but Find returned %d\n",
|
||||||
|
hwy::TypeName(T(), Lanes(d)).c_str(), static_cast<int>(count),
|
||||||
|
static_cast<double>(in[i]), static_cast<int>(i),
|
||||||
|
static_cast<int>(actual));
|
||||||
|
HWY_ASSERT(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also search for values we know not to be present (out of range)
|
||||||
|
HWY_ASSERT_EQ(count, Find(d, T{9}, in, count));
|
||||||
|
HWY_ASSERT_EQ(count, Find(d, static_cast<T>(-9), in, count));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TestAllFind() {
|
||||||
|
ForAllTypes(ForPartialVectors<ForeachCountAndMisalign<TestFind>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TestFindIf {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign, RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
using TI = MakeSigned<T>;
|
||||||
|
// Must allocate at least one even if count is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> storage =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign + count));
|
||||||
|
T* in = storage.get() + misalign;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
in[i] = Random<T>(rng);
|
||||||
|
HWY_ASSERT(in[i] < 8);
|
||||||
|
HWY_ASSERT(!hwy::IsSigned<T>() || static_cast<TI>(in[i]) >= -8);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool found_any = false;
|
||||||
|
bool not_found_any = false;
|
||||||
|
|
||||||
|
// unsigned T would be promoted to signed and compare greater than any
|
||||||
|
// negative val, whereas Set() would just cast to an unsigned value and the
|
||||||
|
// comparison remains unsigned, so avoid negative numbers there.
|
||||||
|
const int min_val = IsSigned<T>() ? -9 : 0;
|
||||||
|
// Includes out-of-range value 9 to test the not-found path.
|
||||||
|
for (int val = min_val; val <= 9; ++val) {
|
||||||
|
#if HWY_GENERIC_LAMBDA
|
||||||
|
const auto greater = [val](const auto d, const auto v) HWY_ATTR {
|
||||||
|
return Gt(v, Set(d, static_cast<T>(val)));
|
||||||
|
};
|
||||||
|
#else
|
||||||
|
const GreaterThan greater(val);
|
||||||
|
#endif
|
||||||
|
const size_t actual = FindIf(d, in, count, greater);
|
||||||
|
found_any |= actual < count;
|
||||||
|
not_found_any |= actual == count;
|
||||||
|
|
||||||
|
const auto pos = std::find_if(
|
||||||
|
in, in + count, [val](T x) { return x > static_cast<T>(val); });
|
||||||
|
// Convert returned iterator to index.
|
||||||
|
const size_t expected = static_cast<size_t>(pos - in);
|
||||||
|
if (expected != actual) {
|
||||||
|
fprintf(stderr, "%s count %d val %d, expected %d actual %d\n",
|
||||||
|
hwy::TypeName(T(), Lanes(d)).c_str(), static_cast<int>(count),
|
||||||
|
val, static_cast<int>(expected), static_cast<int>(actual));
|
||||||
|
hwy::detail::PrintArray(hwy::detail::MakeTypeInfo<T>(), "in", in, count,
|
||||||
|
0, count);
|
||||||
|
HWY_ASSERT(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// We will always not-find something due to val=9.
|
||||||
|
HWY_ASSERT(not_found_any);
|
||||||
|
// We'll find something unless the input is empty or {0} - because 0 > i
|
||||||
|
// is false for all i=[0,9].
|
||||||
|
if (count != 0 && in[0] != 0) {
|
||||||
|
HWY_ASSERT(found_any);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TestAllFindIf() {
|
||||||
|
ForAllTypes(ForPartialVectors<ForeachCountAndMisalign<TestFindIf>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#if HWY_ONCE
|
||||||
|
|
||||||
|
namespace hwy {
|
||||||
|
HWY_BEFORE_TEST(FindTest);
|
||||||
|
HWY_EXPORT_AND_TEST_P(FindTest, TestAllFind);
|
||||||
|
HWY_EXPORT_AND_TEST_P(FindTest, TestAllFindIf);
|
||||||
|
} // namespace hwy
|
||||||
|
|
||||||
|
#endif
|
|
@ -0,0 +1,262 @@
|
||||||
|
// Copyright 2022 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.
|
||||||
|
|
||||||
|
// Per-target include guard
|
||||||
|
#if defined(HIGHWAY_HWY_CONTRIB_ALGO_TRANSFORM_INL_H_) == \
|
||||||
|
defined(HWY_TARGET_TOGGLE)
|
||||||
|
#ifdef HIGHWAY_HWY_CONTRIB_ALGO_TRANSFORM_INL_H_
|
||||||
|
#undef HIGHWAY_HWY_CONTRIB_ALGO_TRANSFORM_INL_H_
|
||||||
|
#else
|
||||||
|
#define HIGHWAY_HWY_CONTRIB_ALGO_TRANSFORM_INL_H_
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
// These functions avoid having to write a loop plus remainder handling in the
|
||||||
|
// (unfortunately still common) case where arrays are not aligned/padded. If the
|
||||||
|
// inputs are known to be aligned/padded, it is more efficient to write a single
|
||||||
|
// loop using Load(). We do not provide a TransformAlignedPadded because it
|
||||||
|
// would be more verbose than such a loop.
|
||||||
|
//
|
||||||
|
// Func is either a functor with a templated operator()(d, v[, v1[, v2]]), or a
|
||||||
|
// generic lambda if using C++14. Due to apparent limitations of Clang on
|
||||||
|
// Windows, it is currently necessary to add HWY_ATTR before the opening { of
|
||||||
|
// the lambda to avoid errors about "always_inline function .. requires target".
|
||||||
|
//
|
||||||
|
// If HWY_MEM_OPS_MIGHT_FAULT, we use scalar code instead of masking. Otherwise,
|
||||||
|
// we used `MaskedLoad` and `BlendedStore` to read/write the final partial
|
||||||
|
// vector.
|
||||||
|
|
||||||
|
// Fills `out[0, count)` with the vectors returned by `func(d, index_vec)`,
|
||||||
|
// where `index_vec` is `Vec<RebindToUnsigned<D>>`. On the first call to `func`,
|
||||||
|
// the value of its lane i is i, and increases by `Lanes(d)` after every call.
|
||||||
|
// Note that some of these indices may be `>= count`, but the elements that
|
||||||
|
// `func` returns in those lanes will not be written to `out`.
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
void Generate(D d, T* HWY_RESTRICT out, size_t count, const Func& func) {
|
||||||
|
const RebindToUnsigned<D> du;
|
||||||
|
using TU = TFromD<decltype(du)>;
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
Vec<decltype(du)> vidx = Iota(du, 0);
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
StoreU(func(d, vidx), d, out + idx);
|
||||||
|
vidx = Add(vidx, Set(du, static_cast<TU>(N)));
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
const RebindToUnsigned<decltype(d1)> du1;
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
StoreU(func(d1, Set(du1, static_cast<TU>(idx))), d1, out + idx);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
BlendedStore(func(d, vidx), mask, d, out + idx);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// Replaces `inout[idx]` with `func(d, inout[idx])`. Example usage: multiplying
|
||||||
|
// array elements by a constant.
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
void Transform(D d, T* HWY_RESTRICT inout, size_t count, const Func& func) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
const Vec<D> v = LoadU(d, inout + idx);
|
||||||
|
StoreU(func(d, v), d, inout + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
const V1 v = LoadU(d1, inout + idx);
|
||||||
|
StoreU(func(d1, v), d1, inout + idx);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, inout + idx);
|
||||||
|
BlendedStore(func(d, v), mask, d, inout + idx);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// Replaces `inout[idx]` with `func(d, inout[idx], in1[idx])`. Example usage:
|
||||||
|
// multiplying array elements by those of another array.
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
void Transform1(D d, T* HWY_RESTRICT inout, size_t count,
|
||||||
|
const T* HWY_RESTRICT in1, const Func& func) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
const Vec<D> v = LoadU(d, inout + idx);
|
||||||
|
const Vec<D> v1 = LoadU(d, in1 + idx);
|
||||||
|
StoreU(func(d, v, v1), d, inout + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
const V1 v = LoadU(d1, inout + idx);
|
||||||
|
const V1 v1 = LoadU(d1, in1 + idx);
|
||||||
|
StoreU(func(d1, v, v1), d1, inout + idx);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, inout + idx);
|
||||||
|
const Vec<D> v1 = MaskedLoad(mask, d, in1 + idx);
|
||||||
|
BlendedStore(func(d, v, v1), mask, d, inout + idx);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// Replaces `inout[idx]` with `func(d, inout[idx], in1[idx], in2[idx])`. Example
|
||||||
|
// usage: FMA of elements from three arrays, stored into the first array.
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
void Transform2(D d, T* HWY_RESTRICT inout, size_t count,
|
||||||
|
const T* HWY_RESTRICT in1, const T* HWY_RESTRICT in2,
|
||||||
|
const Func& func) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
const Vec<D> v = LoadU(d, inout + idx);
|
||||||
|
const Vec<D> v1 = LoadU(d, in1 + idx);
|
||||||
|
const Vec<D> v2 = LoadU(d, in2 + idx);
|
||||||
|
StoreU(func(d, v, v1, v2), d, inout + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
const V1 v = LoadU(d1, inout + idx);
|
||||||
|
const V1 v1 = LoadU(d1, in1 + idx);
|
||||||
|
const V1 v2 = LoadU(d1, in2 + idx);
|
||||||
|
StoreU(func(d1, v, v1, v2), d1, inout + idx);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, inout + idx);
|
||||||
|
const Vec<D> v1 = MaskedLoad(mask, d, in1 + idx);
|
||||||
|
const Vec<D> v2 = MaskedLoad(mask, d, in2 + idx);
|
||||||
|
BlendedStore(func(d, v, v1, v2), mask, d, inout + idx);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D, typename T = TFromD<D>>
|
||||||
|
void Replace(D d, T* HWY_RESTRICT inout, size_t count, T new_t, T old_t) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const Vec<D> old_v = Set(d, old_t);
|
||||||
|
const Vec<D> new_v = Set(d, new_t);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
Vec<D> v = LoadU(d, inout + idx);
|
||||||
|
StoreU(IfThenElse(Eq(v, old_v), new_v, v), d, inout + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
const Vec<decltype(d1)> old_v1 = Set(d1, old_t);
|
||||||
|
const Vec<decltype(d1)> new_v1 = Set(d1, new_t);
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
const V1 v1 = LoadU(d1, inout + idx);
|
||||||
|
StoreU(IfThenElse(Eq(v1, old_v1), new_v1, v1), d1, inout + idx);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, inout + idx);
|
||||||
|
BlendedStore(IfThenElse(Eq(v, old_v), new_v, v), mask, d, inout + idx);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D, class Func, typename T = TFromD<D>>
|
||||||
|
void ReplaceIf(D d, T* HWY_RESTRICT inout, size_t count, T new_t,
|
||||||
|
const Func& func) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const Vec<D> new_v = Set(d, new_t);
|
||||||
|
|
||||||
|
size_t idx = 0;
|
||||||
|
for (; idx + N <= count; idx += N) {
|
||||||
|
Vec<D> v = LoadU(d, inout + idx);
|
||||||
|
StoreU(IfThenElse(func(d, v), new_v, v), d, inout + idx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// `count` was a multiple of the vector length `N`: already done.
|
||||||
|
if (HWY_UNLIKELY(idx == count)) return;
|
||||||
|
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
// Proceed one by one.
|
||||||
|
const CappedTag<T, 1> d1;
|
||||||
|
const Vec<decltype(d1)> new_v1 = Set(d1, new_t);
|
||||||
|
for (; idx < count; ++idx) {
|
||||||
|
using V1 = Vec<decltype(d1)>;
|
||||||
|
const V1 v = LoadU(d1, inout + idx);
|
||||||
|
StoreU(IfThenElse(func(d1, v), new_v1, v), d1, inout + idx);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const size_t remaining = count - idx;
|
||||||
|
HWY_DASSERT(0 != remaining && remaining < N);
|
||||||
|
const Mask<D> mask = FirstN(d, remaining);
|
||||||
|
const Vec<D> v = MaskedLoad(mask, d, inout + idx);
|
||||||
|
BlendedStore(IfThenElse(func(d, v), new_v, v), mask, d, inout + idx);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#endif // HIGHWAY_HWY_CONTRIB_ALGO_TRANSFORM_INL_H_
|
|
@ -0,0 +1,372 @@
|
||||||
|
// Copyright 2022 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 <string.h>
|
||||||
|
|
||||||
|
#include "hwy/aligned_allocator.h"
|
||||||
|
|
||||||
|
// clang-format off
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/algo/transform_test.cc" //NOLINT
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
#include "hwy/contrib/algo/transform-inl.h"
|
||||||
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
// clang-format on
|
||||||
|
|
||||||
|
// If your project requires C++14 or later, you can ignore this and pass lambdas
|
||||||
|
// directly to Transform, without requiring an lvalue as we do here for C++11.
|
||||||
|
#if __cplusplus < 201402L
|
||||||
|
#define HWY_GENERIC_LAMBDA 0
|
||||||
|
#else
|
||||||
|
#define HWY_GENERIC_LAMBDA 1
|
||||||
|
#endif
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
T Alpha() {
|
||||||
|
return static_cast<T>(1.5); // arbitrary scalar
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns random floating-point number in [-8, 8) to ensure computations do
|
||||||
|
// not exceed float32 precision.
|
||||||
|
template <typename T>
|
||||||
|
T Random(RandomState& rng) {
|
||||||
|
const int32_t bits = static_cast<int32_t>(Random32(&rng)) & 1023;
|
||||||
|
const double val = (bits - 512) / 64.0;
|
||||||
|
// Clamp negative to zero for unsigned types.
|
||||||
|
return static_cast<T>(HWY_MAX(hwy::LowestValue<T>(), val));
|
||||||
|
}
|
||||||
|
|
||||||
|
// SCAL, AXPY names are from BLAS.
|
||||||
|
template <typename T>
|
||||||
|
HWY_NOINLINE void SimpleSCAL(const T* x, T* out, size_t count) {
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
out[i] = Alpha<T>() * x[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_NOINLINE void SimpleAXPY(const T* x, const T* y, T* out, size_t count) {
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
out[i] = Alpha<T>() * x[i] + y[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_NOINLINE void SimpleFMA4(const T* x, const T* y, const T* z, T* out,
|
||||||
|
size_t count) {
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
out[i] = x[i] * y[i] + z[i];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// In C++14, we can instead define these as generic lambdas next to where they
|
||||||
|
// are invoked.
|
||||||
|
#if !HWY_GENERIC_LAMBDA
|
||||||
|
|
||||||
|
// Generator that returns even numbers by doubling the output indices.
|
||||||
|
struct Gen2 {
|
||||||
|
template <class D, class VU>
|
||||||
|
Vec<D> operator()(D d, VU vidx) const {
|
||||||
|
return BitCast(d, Add(vidx, vidx));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct SCAL {
|
||||||
|
template <class D, class V>
|
||||||
|
Vec<D> operator()(D d, V v) const {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
return Mul(Set(d, Alpha<T>()), v);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct AXPY {
|
||||||
|
template <class D, class V>
|
||||||
|
Vec<D> operator()(D d, V v, V v1) const {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
return MulAdd(Set(d, Alpha<T>()), v, v1);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct FMA4 {
|
||||||
|
template <class D, class V>
|
||||||
|
Vec<D> operator()(D /*d*/, V v, V v1, V v2) const {
|
||||||
|
return MulAdd(v, v1, v2);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // !HWY_GENERIC_LAMBDA
|
||||||
|
|
||||||
|
// Invokes Test (e.g. TestTransform1) with all arg combinations. T comes from
|
||||||
|
// ForFloatTypes.
|
||||||
|
template <class Test>
|
||||||
|
struct ForeachCountAndMisalign {
|
||||||
|
template <typename T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) const {
|
||||||
|
RandomState rng;
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
const size_t misalignments[3] = {0, N / 4, 3 * N / 5};
|
||||||
|
|
||||||
|
for (size_t count = 0; count < 2 * N; ++count) {
|
||||||
|
for (size_t ma : misalignments) {
|
||||||
|
for (size_t mb : misalignments) {
|
||||||
|
Test()(d, count, ma, mb, rng);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Output-only, no loads
|
||||||
|
struct TestGenerate {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t /*misalign_b*/,
|
||||||
|
RandomState& /*rng*/) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
AlignedFreeUniquePtr<T[]> pa = AllocateAligned<T>(misalign_a + count + 1);
|
||||||
|
T* actual = pa.get() + misalign_a;
|
||||||
|
|
||||||
|
AlignedFreeUniquePtr<T[]> expected = AllocateAligned<T>(HWY_MAX(1, count));
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
expected[i] = static_cast<T>(2 * i);
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO(janwas): can we update the apply_to in HWY_PUSH_ATTRIBUTES so that
|
||||||
|
// the attribute also applies to lambdas? If so, remove HWY_ATTR.
|
||||||
|
#if HWY_GENERIC_LAMBDA
|
||||||
|
const auto gen2 = [](const auto d, const auto vidx)
|
||||||
|
HWY_ATTR { return BitCast(d, Add(vidx, vidx)); };
|
||||||
|
#else
|
||||||
|
const Gen2 gen2;
|
||||||
|
#endif
|
||||||
|
actual[count] = T{0}; // sentinel
|
||||||
|
Generate(d, actual, count, gen2);
|
||||||
|
HWY_ASSERT_EQ(T{0}, actual[count]); // did not write past end
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, expected.get(), actual, count,
|
||||||
|
target_name, __FILE__, __LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Zero extra input arrays
|
||||||
|
struct TestTransform {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
if (misalign_b != 0) return;
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// Prevents error if size to allocate is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> pa =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
T* a = pa.get() + misalign_a;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
a[i] = Random<T>(rng);
|
||||||
|
}
|
||||||
|
|
||||||
|
AlignedFreeUniquePtr<T[]> expected = AllocateAligned<T>(HWY_MAX(1, count));
|
||||||
|
SimpleSCAL(a, expected.get(), count);
|
||||||
|
|
||||||
|
// TODO(janwas): can we update the apply_to in HWY_PUSH_ATTRIBUTES so that
|
||||||
|
// the attribute also applies to lambdas? If so, remove HWY_ATTR.
|
||||||
|
#if HWY_GENERIC_LAMBDA
|
||||||
|
const auto scal = [](const auto d, const auto v)
|
||||||
|
HWY_ATTR { return Mul(Set(d, Alpha<T>()), v); };
|
||||||
|
#else
|
||||||
|
const SCAL scal;
|
||||||
|
#endif
|
||||||
|
Transform(d, a, count, scal);
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, expected.get(), a, count, target_name,
|
||||||
|
__FILE__, __LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// One extra input array
|
||||||
|
struct TestTransform1 {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// Prevents error if size to allocate is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> pa =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
AlignedFreeUniquePtr<T[]> pb =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_b + count));
|
||||||
|
T* a = pa.get() + misalign_a;
|
||||||
|
T* b = pb.get() + misalign_b;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
a[i] = Random<T>(rng);
|
||||||
|
b[i] = Random<T>(rng);
|
||||||
|
}
|
||||||
|
|
||||||
|
AlignedFreeUniquePtr<T[]> expected = AllocateAligned<T>(HWY_MAX(1, count));
|
||||||
|
SimpleAXPY(a, b, expected.get(), count);
|
||||||
|
|
||||||
|
#if HWY_GENERIC_LAMBDA
|
||||||
|
const auto axpy = [](const auto d, const auto v, const auto v1) HWY_ATTR {
|
||||||
|
return MulAdd(Set(d, Alpha<T>()), v, v1);
|
||||||
|
};
|
||||||
|
#else
|
||||||
|
const AXPY axpy;
|
||||||
|
#endif
|
||||||
|
Transform1(d, a, count, b, axpy);
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, expected.get(), a, count, target_name,
|
||||||
|
__FILE__, __LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Two extra input arrays
|
||||||
|
struct TestTransform2 {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
using T = TFromD<D>;
|
||||||
|
// Prevents error if size to allocate is zero.
|
||||||
|
AlignedFreeUniquePtr<T[]> pa =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
AlignedFreeUniquePtr<T[]> pb =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_b + count));
|
||||||
|
AlignedFreeUniquePtr<T[]> pc =
|
||||||
|
AllocateAligned<T>(HWY_MAX(1, misalign_a + count));
|
||||||
|
T* a = pa.get() + misalign_a;
|
||||||
|
T* b = pb.get() + misalign_b;
|
||||||
|
T* c = pc.get() + misalign_a;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
a[i] = Random<T>(rng);
|
||||||
|
b[i] = Random<T>(rng);
|
||||||
|
c[i] = Random<T>(rng);
|
||||||
|
}
|
||||||
|
|
||||||
|
AlignedFreeUniquePtr<T[]> expected = AllocateAligned<T>(HWY_MAX(1, count));
|
||||||
|
SimpleFMA4(a, b, c, expected.get(), count);
|
||||||
|
|
||||||
|
#if HWY_GENERIC_LAMBDA
|
||||||
|
const auto fma4 = [](auto /*d*/, auto v, auto v1, auto v2)
|
||||||
|
HWY_ATTR { return MulAdd(v, v1, v2); };
|
||||||
|
#else
|
||||||
|
const FMA4 fma4;
|
||||||
|
#endif
|
||||||
|
Transform2(d, a, count, b, c, fma4);
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
const char* target_name = hwy::TargetName(HWY_TARGET);
|
||||||
|
hwy::detail::AssertArrayEqual(info, expected.get(), a, count, target_name,
|
||||||
|
__FILE__, __LINE__);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
class IfEq {
|
||||||
|
public:
|
||||||
|
IfEq(T val) : val_(val) {}
|
||||||
|
|
||||||
|
template <class D, class V>
|
||||||
|
Mask<D> operator()(D d, V v) const {
|
||||||
|
return Eq(v, Set(d, val_));
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
T val_;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct TestReplace {
|
||||||
|
template <class D>
|
||||||
|
void operator()(D d, size_t count, size_t misalign_a, size_t misalign_b,
|
||||||
|
RandomState& rng) {
|
||||||
|
if (misalign_b != 0) return;
|
||||||
|
if (count == 0) return;
|
||||||
|
using T = TFromD<D>;
|
||||||
|
AlignedFreeUniquePtr<T[]> pa = AllocateAligned<T>(misalign_a + count);
|
||||||
|
T* a = pa.get() + misalign_a;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
a[i] = Random<T>(rng);
|
||||||
|
}
|
||||||
|
AlignedFreeUniquePtr<T[]> pb = AllocateAligned<T>(count);
|
||||||
|
|
||||||
|
AlignedFreeUniquePtr<T[]> expected = AllocateAligned<T>(count);
|
||||||
|
|
||||||
|
std::vector<size_t> positions(AdjustedReps(count));
|
||||||
|
for (size_t& pos : positions) {
|
||||||
|
pos = static_cast<size_t>(rng()) % count;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (size_t pos = 0; pos < count; ++pos) {
|
||||||
|
const T old_t = a[pos];
|
||||||
|
const T new_t = Random<T>(rng);
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
expected[i] = IsEqual(a[i], old_t) ? new_t : a[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
// Copy so ReplaceIf gets the same input (and thus also outputs expected)
|
||||||
|
memcpy(pb.get(), a, count * sizeof(T));
|
||||||
|
|
||||||
|
Replace(d, a, count, new_t, old_t);
|
||||||
|
HWY_ASSERT_ARRAY_EQ(expected.get(), a, count);
|
||||||
|
|
||||||
|
ReplaceIf(d, pb.get(), count, new_t, IfEq<T>(old_t));
|
||||||
|
HWY_ASSERT_ARRAY_EQ(expected.get(), pb.get(), count);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TestAllGenerate() {
|
||||||
|
// The test BitCast-s the indices, which does not work for floats.
|
||||||
|
ForIntegerTypes(ForPartialVectors<ForeachCountAndMisalign<TestGenerate>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
void TestAllTransform() {
|
||||||
|
ForFloatTypes(ForPartialVectors<ForeachCountAndMisalign<TestTransform>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
void TestAllTransform1() {
|
||||||
|
ForFloatTypes(ForPartialVectors<ForeachCountAndMisalign<TestTransform1>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
void TestAllTransform2() {
|
||||||
|
ForFloatTypes(ForPartialVectors<ForeachCountAndMisalign<TestTransform2>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
void TestAllReplace() {
|
||||||
|
ForFloatTypes(ForPartialVectors<ForeachCountAndMisalign<TestReplace>>());
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#if HWY_ONCE
|
||||||
|
|
||||||
|
namespace hwy {
|
||||||
|
HWY_BEFORE_TEST(TransformTest);
|
||||||
|
HWY_EXPORT_AND_TEST_P(TransformTest, TestAllGenerate);
|
||||||
|
HWY_EXPORT_AND_TEST_P(TransformTest, TestAllTransform);
|
||||||
|
HWY_EXPORT_AND_TEST_P(TransformTest, TestAllTransform1);
|
||||||
|
HWY_EXPORT_AND_TEST_P(TransformTest, TestAllTransform2);
|
||||||
|
HWY_EXPORT_AND_TEST_P(TransformTest, TestAllReplace);
|
||||||
|
} // namespace hwy
|
||||||
|
|
||||||
|
#endif
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -34,23 +35,19 @@ struct Dot {
|
||||||
// argument to Compute. Each one may improve performance or reduce code size,
|
// argument to Compute. Each one may improve performance or reduce code size,
|
||||||
// at the cost of additional requirements on the arguments.
|
// at the cost of additional requirements on the arguments.
|
||||||
enum Assumptions {
|
enum Assumptions {
|
||||||
// num_elements is at least N, which may be up to HWY_MAX_LANES(T).
|
// num_elements is at least N, which may be up to HWY_MAX_BYTES / sizeof(T).
|
||||||
kAtLeastOneVector = 1,
|
kAtLeastOneVector = 1,
|
||||||
// num_elements is divisible by N (a power of two, so this can be used if
|
// num_elements is divisible by N (a power of two, so this can be used if
|
||||||
// the problem size is known to be a power of two >= HWY_MAX_LANES(T)).
|
// the problem size is known to be a power of two >= HWY_MAX_BYTES /
|
||||||
|
// sizeof(T)).
|
||||||
kMultipleOfVector = 2,
|
kMultipleOfVector = 2,
|
||||||
// RoundUpTo(num_elements, N) elements are accessible; their value does not
|
// RoundUpTo(num_elements, N) elements are accessible; their value does not
|
||||||
// matter (will be treated as if they were zero).
|
// matter (will be treated as if they were zero).
|
||||||
kPaddedToVector = 4,
|
kPaddedToVector = 4,
|
||||||
// Pointers pa and pb, respectively, are multiples of N * sizeof(T).
|
|
||||||
// For example, aligned_allocator.h ensures this. Note that it is still
|
|
||||||
// beneficial to ensure such alignment even if these flags are not set.
|
|
||||||
// If not set, the pointers need only be aligned to alignof(T).
|
|
||||||
kVectorAlignedA = 8,
|
|
||||||
kVectorAlignedB = 16,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// Returns sum{pa[i] * pb[i]} for float or double inputs.
|
// Returns sum{pa[i] * pb[i]} for float or double inputs. Aligning the
|
||||||
|
// pointers to a multiple of N elements is helpful but not required.
|
||||||
template <int kAssumptions, class D, typename T = TFromD<D>,
|
template <int kAssumptions, class D, typename T = TFromD<D>,
|
||||||
HWY_IF_NOT_LANE_SIZE_D(D, 2)>
|
HWY_IF_NOT_LANE_SIZE_D(D, 2)>
|
||||||
static HWY_INLINE T Compute(const D d, const T* const HWY_RESTRICT pa,
|
static HWY_INLINE T Compute(const D d, const T* const HWY_RESTRICT pa,
|
||||||
|
@ -67,8 +64,6 @@ struct Dot {
|
||||||
constexpr bool kIsMultipleOfVector =
|
constexpr bool kIsMultipleOfVector =
|
||||||
(kAssumptions & kMultipleOfVector) != 0;
|
(kAssumptions & kMultipleOfVector) != 0;
|
||||||
constexpr bool kIsPaddedToVector = (kAssumptions & kPaddedToVector) != 0;
|
constexpr bool kIsPaddedToVector = (kAssumptions & kPaddedToVector) != 0;
|
||||||
constexpr bool kIsAlignedA = (kAssumptions & kVectorAlignedA) != 0;
|
|
||||||
constexpr bool kIsAlignedB = (kAssumptions & kVectorAlignedB) != 0;
|
|
||||||
|
|
||||||
// Won't be able to do a full vector load without padding => scalar loop.
|
// Won't be able to do a full vector load without padding => scalar loop.
|
||||||
if (!kIsAtLeastOneVector && !kIsMultipleOfVector && !kIsPaddedToVector &&
|
if (!kIsAtLeastOneVector && !kIsMultipleOfVector && !kIsPaddedToVector &&
|
||||||
|
@ -98,28 +93,28 @@ struct Dot {
|
||||||
|
|
||||||
// Main loop: unrolled
|
// Main loop: unrolled
|
||||||
for (; i + 4 * N <= num_elements; /* i += 4 * N */) { // incr in loop
|
for (; i + 4 * N <= num_elements; /* i += 4 * N */) { // incr in loop
|
||||||
const auto a0 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a0 = LoadU(d, pa + i);
|
||||||
const auto b0 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b0 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum0 = MulAdd(a0, b0, sum0);
|
sum0 = MulAdd(a0, b0, sum0);
|
||||||
const auto a1 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a1 = LoadU(d, pa + i);
|
||||||
const auto b1 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b1 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum1 = MulAdd(a1, b1, sum1);
|
sum1 = MulAdd(a1, b1, sum1);
|
||||||
const auto a2 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a2 = LoadU(d, pa + i);
|
||||||
const auto b2 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b2 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum2 = MulAdd(a2, b2, sum2);
|
sum2 = MulAdd(a2, b2, sum2);
|
||||||
const auto a3 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a3 = LoadU(d, pa + i);
|
||||||
const auto b3 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b3 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum3 = MulAdd(a3, b3, sum3);
|
sum3 = MulAdd(a3, b3, sum3);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Up to 3 iterations of whole vectors
|
// Up to 3 iterations of whole vectors
|
||||||
for (; i + N <= num_elements; i += N) {
|
for (; i + N <= num_elements; i += N) {
|
||||||
const auto a = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a = LoadU(d, pa + i);
|
||||||
const auto b = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b = LoadU(d, pb + i);
|
||||||
sum0 = MulAdd(a, b, sum0);
|
sum0 = MulAdd(a, b, sum0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -128,8 +123,8 @@ struct Dot {
|
||||||
if (remaining != 0) {
|
if (remaining != 0) {
|
||||||
if (kIsPaddedToVector) {
|
if (kIsPaddedToVector) {
|
||||||
const auto mask = FirstN(d, remaining);
|
const auto mask = FirstN(d, remaining);
|
||||||
const auto a = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a = LoadU(d, pa + i);
|
||||||
const auto b = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b = LoadU(d, pb + i);
|
||||||
sum1 = MulAdd(IfThenElseZero(mask, a), IfThenElseZero(mask, b), sum1);
|
sum1 = MulAdd(IfThenElseZero(mask, a), IfThenElseZero(mask, b), sum1);
|
||||||
} else {
|
} else {
|
||||||
// Unaligned load such that the last element is in the highest lane -
|
// Unaligned load such that the last element is in the highest lane -
|
||||||
|
@ -152,7 +147,8 @@ struct Dot {
|
||||||
return GetLane(SumOfLanes(d, sum0));
|
return GetLane(SumOfLanes(d, sum0));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns sum{pa[i] * pb[i]} for bfloat16 inputs.
|
// Returns sum{pa[i] * pb[i]} for bfloat16 inputs. Aligning the pointers to a
|
||||||
|
// multiple of N elements is helpful but not required.
|
||||||
template <int kAssumptions, class D>
|
template <int kAssumptions, class D>
|
||||||
static HWY_INLINE float Compute(const D d,
|
static HWY_INLINE float Compute(const D d,
|
||||||
const bfloat16_t* const HWY_RESTRICT pa,
|
const bfloat16_t* const HWY_RESTRICT pa,
|
||||||
|
@ -170,8 +166,6 @@ struct Dot {
|
||||||
constexpr bool kIsMultipleOfVector =
|
constexpr bool kIsMultipleOfVector =
|
||||||
(kAssumptions & kMultipleOfVector) != 0;
|
(kAssumptions & kMultipleOfVector) != 0;
|
||||||
constexpr bool kIsPaddedToVector = (kAssumptions & kPaddedToVector) != 0;
|
constexpr bool kIsPaddedToVector = (kAssumptions & kPaddedToVector) != 0;
|
||||||
constexpr bool kIsAlignedA = (kAssumptions & kVectorAlignedA) != 0;
|
|
||||||
constexpr bool kIsAlignedB = (kAssumptions & kVectorAlignedB) != 0;
|
|
||||||
|
|
||||||
// Won't be able to do a full vector load without padding => scalar loop.
|
// Won't be able to do a full vector load without padding => scalar loop.
|
||||||
if (!kIsAtLeastOneVector && !kIsMultipleOfVector && !kIsPaddedToVector &&
|
if (!kIsAtLeastOneVector && !kIsMultipleOfVector && !kIsPaddedToVector &&
|
||||||
|
@ -197,20 +191,20 @@ struct Dot {
|
||||||
|
|
||||||
// Main loop: unrolled
|
// Main loop: unrolled
|
||||||
for (; i + 2 * N <= num_elements; /* i += 2 * N */) { // incr in loop
|
for (; i + 2 * N <= num_elements; /* i += 2 * N */) { // incr in loop
|
||||||
const auto a0 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a0 = LoadU(d, pa + i);
|
||||||
const auto b0 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b0 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum0 = ReorderWidenMulAccumulate(df32, a0, b0, sum0, sum1);
|
sum0 = ReorderWidenMulAccumulate(df32, a0, b0, sum0, sum1);
|
||||||
const auto a1 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a1 = LoadU(d, pa + i);
|
||||||
const auto b1 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b1 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum2 = ReorderWidenMulAccumulate(df32, a1, b1, sum2, sum3);
|
sum2 = ReorderWidenMulAccumulate(df32, a1, b1, sum2, sum3);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Possibly one more iteration of whole vectors
|
// Possibly one more iteration of whole vectors
|
||||||
if (i + N <= num_elements) {
|
if (i + N <= num_elements) {
|
||||||
const auto a0 = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto a0 = LoadU(d, pa + i);
|
||||||
const auto b0 = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto b0 = LoadU(d, pb + i);
|
||||||
i += N;
|
i += N;
|
||||||
sum0 = ReorderWidenMulAccumulate(df32, a0, b0, sum0, sum1);
|
sum0 = ReorderWidenMulAccumulate(df32, a0, b0, sum0, sum1);
|
||||||
}
|
}
|
||||||
|
@ -220,8 +214,8 @@ struct Dot {
|
||||||
if (remaining != 0) {
|
if (remaining != 0) {
|
||||||
if (kIsPaddedToVector) {
|
if (kIsPaddedToVector) {
|
||||||
const auto mask = FirstN(du16, remaining);
|
const auto mask = FirstN(du16, remaining);
|
||||||
const auto va = kIsAlignedA ? Load(d, pa + i) : LoadU(d, pa + i);
|
const auto va = LoadU(d, pa + i);
|
||||||
const auto vb = kIsAlignedB ? Load(d, pb + i) : LoadU(d, pb + i);
|
const auto vb = LoadU(d, pb + i);
|
||||||
const auto a16 = BitCast(d, IfThenElseZero(mask, BitCast(du16, va)));
|
const auto a16 = BitCast(d, IfThenElseZero(mask, BitCast(du16, va)));
|
||||||
const auto b16 = BitCast(d, IfThenElseZero(mask, BitCast(du16, vb)));
|
const auto b16 = BitCast(d, IfThenElseZero(mask, BitCast(du16, vb)));
|
||||||
sum2 = ReorderWidenMulAccumulate(df32, a16, b16, sum2, sum3);
|
sum2 = ReorderWidenMulAccumulate(df32, a16, b16, sum2, sum3);
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -21,7 +22,7 @@
|
||||||
// clang-format off
|
// clang-format off
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/dot/dot_test.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/dot/dot_test.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
#include "hwy/contrib/dot/dot-inl.h"
|
#include "hwy/contrib/dot/dot-inl.h"
|
||||||
#include "hwy/tests/test_util-inl.h"
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
@ -69,11 +70,6 @@ class TestDot {
|
||||||
return static_cast<float>(bits - 512) * (1.0f / 64);
|
return static_cast<float>(bits - 512) * (1.0f / 64);
|
||||||
};
|
};
|
||||||
|
|
||||||
const bool kIsAlignedA = (kAssumptions & Dot::kVectorAlignedA) != 0;
|
|
||||||
const bool kIsAlignedB = (kAssumptions & Dot::kVectorAlignedB) != 0;
|
|
||||||
|
|
||||||
HWY_ASSERT(!kIsAlignedA || misalign_a == 0);
|
|
||||||
HWY_ASSERT(!kIsAlignedB || misalign_b == 0);
|
|
||||||
const size_t padded =
|
const size_t padded =
|
||||||
(kAssumptions & Dot::kPaddedToVector) ? RoundUpTo(num, N) : num;
|
(kAssumptions & Dot::kPaddedToVector) ? RoundUpTo(num, N) : num;
|
||||||
AlignedFreeUniquePtr<T[]> pa = AllocateAligned<T>(misalign_a + padded);
|
AlignedFreeUniquePtr<T[]> pa = AllocateAligned<T>(misalign_a + padded);
|
||||||
|
@ -99,27 +95,11 @@ class TestDot {
|
||||||
HWY_ASSERT(expected - 1E-4 <= actual && actual <= expected + 1E-4);
|
HWY_ASSERT(expected - 1E-4 <= actual && actual <= expected + 1E-4);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Runs tests with various alignments compatible with the given assumptions.
|
// Runs tests with various alignments.
|
||||||
template <int kAssumptions, class D>
|
template <int kAssumptions, class D>
|
||||||
void ForeachMisalign(D d, size_t num, RandomState& rng) {
|
void ForeachMisalign(D d, size_t num, RandomState& rng) {
|
||||||
static_assert(
|
|
||||||
(kAssumptions & (Dot::kVectorAlignedA | Dot::kVectorAlignedB)) == 0,
|
|
||||||
"Alignment must not be specified by caller");
|
|
||||||
|
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
const size_t misalignments[3] = {0, N / 4, 3 * N / 5};
|
const size_t misalignments[3] = {0, N / 4, 3 * N / 5};
|
||||||
|
|
||||||
// Both flags, both aligned
|
|
||||||
Test<kAssumptions | Dot::kVectorAlignedA | Dot::kVectorAlignedB>(d, num, 0,
|
|
||||||
0, rng);
|
|
||||||
|
|
||||||
// One flag and aligned, other aligned/misaligned
|
|
||||||
for (size_t m : misalignments) {
|
|
||||||
Test<kAssumptions | Dot::kVectorAlignedA>(d, num, 0, m, rng);
|
|
||||||
Test<kAssumptions | Dot::kVectorAlignedB>(d, num, m, 0, rng);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Neither flag, all combinations of aligned/misaligned
|
|
||||||
for (size_t ma : misalignments) {
|
for (size_t ma : misalignments) {
|
||||||
for (size_t mb : misalignments) {
|
for (size_t mb : misalignments) {
|
||||||
Test<kAssumptions>(d, num, ma, mb, rng);
|
Test<kAssumptions>(d, num, ma, mb, rng);
|
||||||
|
@ -184,10 +164,4 @@ HWY_EXPORT_AND_TEST_P(DotTest, TestAllDot);
|
||||||
HWY_EXPORT_AND_TEST_P(DotTest, TestAllDotBF16);
|
HWY_EXPORT_AND_TEST_P(DotTest, TestAllDotBF16);
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -19,7 +20,7 @@
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/image/image.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/image/image.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
@ -104,7 +105,7 @@ ImageBase::ImageBase(const size_t xsize, const size_t ysize,
|
||||||
}
|
}
|
||||||
|
|
||||||
void ImageBase::InitializePadding(const size_t sizeof_t, Padding padding) {
|
void ImageBase::InitializePadding(const size_t sizeof_t, Padding padding) {
|
||||||
#if defined(MEMORY_SANITIZER) || HWY_IDE
|
#if HWY_IS_MSAN || HWY_IDE
|
||||||
if (xsize_ == 0 || ysize_ == 0) return;
|
if (xsize_ == 0 || ysize_ == 0) return;
|
||||||
|
|
||||||
const size_t vec_size = VectorSize(); // Bytes, independent of sizeof_t!
|
const size_t vec_size = VectorSize(); // Bytes, independent of sizeof_t!
|
||||||
|
@ -130,7 +131,7 @@ void ImageBase::InitializePadding(const size_t sizeof_t, Padding padding) {
|
||||||
#else
|
#else
|
||||||
(void)sizeof_t;
|
(void)sizeof_t;
|
||||||
(void)padding;
|
(void)padding;
|
||||||
#endif // MEMORY_SANITIZER
|
#endif // HWY_IS_MSAN
|
||||||
}
|
}
|
||||||
|
|
||||||
void ImageBase::Swap(ImageBase& other) {
|
void ImageBase::Swap(ImageBase& other) {
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright (c) the JPEG XL Project
|
// Copyright (c) the JPEG XL Project
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -14,14 +15,7 @@
|
||||||
|
|
||||||
#include "hwy/contrib/image/image.h"
|
#include "hwy/contrib/image/image.h"
|
||||||
|
|
||||||
#include <cstddef>
|
#include <stddef.h>
|
||||||
|
|
||||||
#include "hwy/base.h"
|
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/image/image_test.cc"
|
|
||||||
#include "hwy/foreach_target.h"
|
|
||||||
|
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
@ -29,6 +23,11 @@
|
||||||
#include <random>
|
#include <random>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/image/image_test.cc"
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
// After foreach_target:
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
#include "hwy/tests/test_util-inl.h"
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
|
||||||
|
@ -82,7 +81,7 @@ struct TestUnalignedT {
|
||||||
|
|
||||||
// This test reads padding, which only works if it was initialized,
|
// This test reads padding, which only works if it was initialized,
|
||||||
// which only happens in MSAN builds.
|
// which only happens in MSAN builds.
|
||||||
#if defined(MEMORY_SANITIZER) || HWY_IDE
|
#if HWY_IS_MSAN || HWY_IDE
|
||||||
// Initialize only the valid samples
|
// Initialize only the valid samples
|
||||||
for (size_t y = 0; y < ysize; ++y) {
|
for (size_t y = 0; y < ysize; ++y) {
|
||||||
T* HWY_RESTRICT row = img.MutableRow(y);
|
T* HWY_RESTRICT row = img.MutableRow(y);
|
||||||
|
@ -150,10 +149,4 @@ HWY_EXPORT_AND_TEST_P(ImageTest, TestAligned);
|
||||||
HWY_EXPORT_AND_TEST_P(ImageTest, TestUnaligned);
|
HWY_EXPORT_AND_TEST_P(ImageTest, TestUnaligned);
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -20,7 +21,7 @@
|
||||||
// clang-format off
|
// clang-format off
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/math/math_test.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/math/math_test.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
#include "hwy/contrib/math/math-inl.h"
|
#include "hwy/contrib/math/math-inl.h"
|
||||||
#include "hwy/tests/test_util-inl.h"
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
@ -219,10 +220,4 @@ HWY_EXPORT_AND_TEST_P(HwyMathTest, TestAllSinh);
|
||||||
HWY_EXPORT_AND_TEST_P(HwyMathTest, TestAllTanh);
|
HWY_EXPORT_AND_TEST_P(HwyMathTest, TestAllTanh);
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char **argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -8,35 +8,91 @@ COMPAT = [
|
||||||
"//buildenv/target:non_prod", # includes mobile/vendor.
|
"//buildenv/target:non_prod", # includes mobile/vendor.
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# cc_library(
|
||||||
|
# name = "vxsort",
|
||||||
|
# srcs = [
|
||||||
|
# "vxsort/isa_detection.cpp",
|
||||||
|
# "vxsort/isa_detection_msvc.cpp",
|
||||||
|
# "vxsort/isa_detection_sane.cpp",
|
||||||
|
# "vxsort/machine_traits.avx2.cpp",
|
||||||
|
# "vxsort/smallsort/avx2_load_mask_tables.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.double.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.float.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.int32_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.int64_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.uint32_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.uint64_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.double.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.float.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.int32_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.int64_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.uint32_t.generated.cpp",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.uint64_t.generated.cpp",
|
||||||
|
# "vxsort/vxsort_stats.cpp",
|
||||||
|
# ],
|
||||||
|
# hdrs = [
|
||||||
|
# "vxsort/alignment.h",
|
||||||
|
# "vxsort/defs.h",
|
||||||
|
# "vxsort/isa_detection.h",
|
||||||
|
# "vxsort/machine_traits.avx2.h",
|
||||||
|
# "vxsort/machine_traits.avx512.h",
|
||||||
|
# "vxsort/machine_traits.h",
|
||||||
|
# "vxsort/packer.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.double.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.float.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.int32_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.int64_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.uint32_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX2.uint64_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.double.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.float.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.int32_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.int64_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.uint32_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.AVX512.uint64_t.generated.h",
|
||||||
|
# "vxsort/smallsort/bitonic_sort.h",
|
||||||
|
# "vxsort/vxsort.h",
|
||||||
|
# "vxsort/vxsort_stats.h",
|
||||||
|
# ],
|
||||||
|
# compatible_with = [],
|
||||||
|
# textual_hdrs = [
|
||||||
|
# "vxsort/vxsort_targets_disable.h",
|
||||||
|
# "vxsort/vxsort_targets_enable_avx2.h",
|
||||||
|
# "vxsort/vxsort_targets_enable_avx512.h",
|
||||||
|
# ],
|
||||||
|
# )
|
||||||
|
|
||||||
cc_library(
|
cc_library(
|
||||||
name = "vqsort",
|
name = "vqsort",
|
||||||
srcs = [
|
srcs = [
|
||||||
# Split into separate files to reduce MSVC build time.
|
# Split into separate files to reduce MSVC build time.
|
||||||
"vqsort.cc",
|
"vqsort.cc",
|
||||||
"vqsort_i16a.cc",
|
|
||||||
"vqsort_i16d.cc",
|
|
||||||
"vqsort_u16a.cc",
|
|
||||||
"vqsort_u16d.cc",
|
|
||||||
"vqsort_f32a.cc",
|
|
||||||
"vqsort_f32d.cc",
|
|
||||||
"vqsort_i32a.cc",
|
|
||||||
"vqsort_i32d.cc",
|
|
||||||
"vqsort_u32a.cc",
|
|
||||||
"vqsort_u32d.cc",
|
|
||||||
"vqsort_f64a.cc",
|
|
||||||
"vqsort_f64d.cc",
|
|
||||||
"vqsort_i64a.cc",
|
|
||||||
"vqsort_i64d.cc",
|
|
||||||
"vqsort_u64a.cc",
|
|
||||||
"vqsort_u64d.cc",
|
|
||||||
"vqsort_128a.cc",
|
"vqsort_128a.cc",
|
||||||
"vqsort_128d.cc",
|
"vqsort_128d.cc",
|
||||||
|
"vqsort_f32a.cc",
|
||||||
|
"vqsort_f32d.cc",
|
||||||
|
"vqsort_f64a.cc",
|
||||||
|
"vqsort_f64d.cc",
|
||||||
|
"vqsort_i16a.cc",
|
||||||
|
"vqsort_i16d.cc",
|
||||||
|
"vqsort_i32a.cc",
|
||||||
|
"vqsort_i32d.cc",
|
||||||
|
"vqsort_i64a.cc",
|
||||||
|
"vqsort_i64d.cc",
|
||||||
|
"vqsort_kv128a.cc",
|
||||||
|
"vqsort_kv128d.cc",
|
||||||
|
"vqsort_u16a.cc",
|
||||||
|
"vqsort_u16d.cc",
|
||||||
|
"vqsort_u32a.cc",
|
||||||
|
"vqsort_u32d.cc",
|
||||||
|
"vqsort_u64a.cc",
|
||||||
|
"vqsort_u64d.cc",
|
||||||
],
|
],
|
||||||
hdrs = [
|
hdrs = [
|
||||||
"disabled_targets.h",
|
|
||||||
"vqsort.h", # public interface
|
"vqsort.h", # public interface
|
||||||
],
|
],
|
||||||
compatible_with = [],
|
compatible_with = [],
|
||||||
|
local_defines = ["hwy_contrib_EXPORTS"],
|
||||||
textual_hdrs = [
|
textual_hdrs = [
|
||||||
"shared-inl.h",
|
"shared-inl.h",
|
||||||
"sorting_networks-inl.h",
|
"sorting_networks-inl.h",
|
||||||
|
@ -48,6 +104,7 @@ cc_library(
|
||||||
# Only if VQSORT_SECURE_RNG is set.
|
# Only if VQSORT_SECURE_RNG is set.
|
||||||
# "//third_party/absl/random",
|
# "//third_party/absl/random",
|
||||||
"//:hwy",
|
"//:hwy",
|
||||||
|
# ":vxsort", # required if HAVE_VXSORT
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -86,8 +143,7 @@ cc_test(
|
||||||
name = "sort_test",
|
name = "sort_test",
|
||||||
size = "medium",
|
size = "medium",
|
||||||
srcs = ["sort_test.cc"],
|
srcs = ["sort_test.cc"],
|
||||||
features = ["fully_static_link"],
|
# Do not enable fully_static_link (pthread crash on bazel)
|
||||||
linkstatic = True,
|
|
||||||
local_defines = ["HWY_IS_TEST"],
|
local_defines = ["HWY_IS_TEST"],
|
||||||
# for test_suite.
|
# for test_suite.
|
||||||
tags = ["hwy_ops_test"],
|
tags = ["hwy_ops_test"],
|
||||||
|
@ -104,8 +160,7 @@ cc_binary(
|
||||||
name = "bench_sort",
|
name = "bench_sort",
|
||||||
testonly = 1,
|
testonly = 1,
|
||||||
srcs = ["bench_sort.cc"],
|
srcs = ["bench_sort.cc"],
|
||||||
features = ["fully_static_link"],
|
# Do not enable fully_static_link (pthread crash on bazel)
|
||||||
linkstatic = True,
|
|
||||||
local_defines = ["HWY_IS_TEST"],
|
local_defines = ["HWY_IS_TEST"],
|
||||||
deps = [
|
deps = [
|
||||||
":helpers",
|
":helpers",
|
||||||
|
@ -120,8 +175,7 @@ cc_binary(
|
||||||
name = "bench_parallel",
|
name = "bench_parallel",
|
||||||
testonly = 1,
|
testonly = 1,
|
||||||
srcs = ["bench_parallel.cc"],
|
srcs = ["bench_parallel.cc"],
|
||||||
features = ["fully_static_link"],
|
# Do not enable fully_static_link (pthread crash on bazel)
|
||||||
linkstatic = True,
|
|
||||||
local_defines = ["HWY_IS_TEST"],
|
local_defines = ["HWY_IS_TEST"],
|
||||||
deps = [
|
deps = [
|
||||||
":helpers",
|
":helpers",
|
||||||
|
|
|
@ -0,0 +1,81 @@
|
||||||
|
# Vectorized and performance-portable Quicksort
|
||||||
|
|
||||||
|
## Introduction
|
||||||
|
|
||||||
|
As of 2022-06-07 this sorts large arrays of built-in types about ten times as
|
||||||
|
fast as `std::sort`. See also our
|
||||||
|
[blog post](https://opensource.googleblog.com/2022/06/Vectorized%20and%20performance%20portable%20Quicksort.html)
|
||||||
|
and [paper](https://arxiv.org/abs/2205.05982).
|
||||||
|
|
||||||
|
## Instructions
|
||||||
|
|
||||||
|
Here are instructions for reproducing our results on x86 Linux (AVX2, AVX-512)
|
||||||
|
and Arm V1 (NEON, SVE).
|
||||||
|
|
||||||
|
### x86 (Linux)
|
||||||
|
|
||||||
|
Please first ensure golang, and Clang (tested with 13.0.1) are installed via
|
||||||
|
your system's package manager.
|
||||||
|
|
||||||
|
```
|
||||||
|
go install github.com/bazelbuild/bazelisk@latest
|
||||||
|
git clone https://github.com/google/highway
|
||||||
|
cd highway
|
||||||
|
CC=clang CXX=clang++ ~/go/bin/bazelisk build -c opt hwy/contrib/sort:all
|
||||||
|
bazel-bin/hwy/contrib/sort/sort_test
|
||||||
|
bazel-bin/hwy/contrib/sort/bench_sort
|
||||||
|
```
|
||||||
|
|
||||||
|
### AWS Graviton3
|
||||||
|
|
||||||
|
Instance config: amazon linux 5.10 arm64, c7g.8xlarge (largest allowed config is
|
||||||
|
32 vCPU). Initial launch will fail. Wait a few minutes for an email saying the
|
||||||
|
config is verified, then re-launch. See IPv4 hostname in list of instances.
|
||||||
|
|
||||||
|
`ssh -i /path/key.pem ec2-user@hostname`
|
||||||
|
|
||||||
|
Note that the AWS CMake package is too old for llvm, so we build it first:
|
||||||
|
```
|
||||||
|
wget https://cmake.org/files/v3.23/cmake-3.23.2.tar.gz
|
||||||
|
tar -xvzf cmake-3.23.2.tar.gz && cd cmake-3.23.2/
|
||||||
|
./bootstrap -- -DCMAKE_USE_OPENSSL=OFF
|
||||||
|
make -j8 && sudo make install
|
||||||
|
cd ..
|
||||||
|
```
|
||||||
|
|
||||||
|
AWS clang is at version 11.1, which generates unnecessary AND instructions which
|
||||||
|
slow down the sort by 1.15x. We tested with clang trunk as of June 13
|
||||||
|
(which reports Git hash 8f6512fea000c3a0d394864bb94e524bee375069). To build:
|
||||||
|
```
|
||||||
|
git clone --depth 1 https://github.com/llvm/llvm-project.git
|
||||||
|
cd llvm-project
|
||||||
|
mkdir -p build && cd build
|
||||||
|
/usr/local/bin/cmake ../llvm -DLLVM_ENABLE_PROJECTS="clang" -DLLVM_ENABLE_RUNTIMES="libcxx;libcxxabi" -DCMAKE_BUILD_TYPE=Release
|
||||||
|
make -j32 && sudo make install
|
||||||
|
```
|
||||||
|
|
||||||
|
```
|
||||||
|
sudo yum install go
|
||||||
|
go install github.com/bazelbuild/bazelisk@latest
|
||||||
|
git clone https://github.com/google/highway
|
||||||
|
cd highway
|
||||||
|
CC=/usr/local/bin/clang CXX=/usr/local/bin/clang++ ~/go/bin/bazelisk build -c opt --copt=-march=armv8.2-a+sve hwy/contrib/sort:all
|
||||||
|
bazel-bin/hwy/contrib/sort/sort_test
|
||||||
|
bazel-bin/hwy/contrib/sort/bench_sort
|
||||||
|
```
|
||||||
|
|
||||||
|
## Results
|
||||||
|
|
||||||
|
`bench_sort` outputs the instruction set (AVX3 refers to AVX-512), the sort
|
||||||
|
algorithm (std for `std::sort`, vq for our vqsort), the type of keys being
|
||||||
|
sorted (f32 is float), the distribution of keys (uniform32 for uniform random
|
||||||
|
with range 0-2^32), the number of keys, then the throughput of sorted keys (i.e.
|
||||||
|
number of key bytes output per second).
|
||||||
|
|
||||||
|
Example excerpt from Xeon 6154 (Skylake-X) CPU clocked at 3 GHz:
|
||||||
|
|
||||||
|
```
|
||||||
|
[ RUN ] BenchSortGroup/BenchSort.BenchAllSort/AVX3
|
||||||
|
AVX3: std: f32: uniform32: 1.00E+06 54 MB/s ( 1 threads)
|
||||||
|
AVX3: vq: f32: uniform32: 1.00E+06 1143 MB/s ( 1 threads)
|
||||||
|
```
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -29,16 +30,18 @@
|
||||||
// Third-party algorithms
|
// Third-party algorithms
|
||||||
#define HAVE_AVX2SORT 0
|
#define HAVE_AVX2SORT 0
|
||||||
#define HAVE_IPS4O 0
|
#define HAVE_IPS4O 0
|
||||||
|
// When enabling, consider changing max_threads (required for Table 1a)
|
||||||
#define HAVE_PARALLEL_IPS4O (HAVE_IPS4O && 1)
|
#define HAVE_PARALLEL_IPS4O (HAVE_IPS4O && 1)
|
||||||
#define HAVE_PDQSORT 0
|
#define HAVE_PDQSORT 0
|
||||||
#define HAVE_SORT512 0
|
#define HAVE_SORT512 0
|
||||||
|
#define HAVE_VXSORT 0
|
||||||
|
|
||||||
#if HAVE_AVX2SORT
|
#if HAVE_AVX2SORT
|
||||||
HWY_PUSH_ATTRIBUTES("avx2,avx")
|
HWY_PUSH_ATTRIBUTES("avx2,avx")
|
||||||
#include "avx2sort.h"
|
#include "avx2sort.h" //NOLINT
|
||||||
HWY_POP_ATTRIBUTES
|
HWY_POP_ATTRIBUTES
|
||||||
#endif
|
#endif
|
||||||
#if HAVE_IPS4O
|
#if HAVE_IPS4O || HAVE_PARALLEL_IPS4O
|
||||||
#include "third_party/ips4o/include/ips4o.hpp"
|
#include "third_party/ips4o/include/ips4o.hpp"
|
||||||
#include "third_party/ips4o/include/ips4o/thread_pool.hpp"
|
#include "third_party/ips4o/include/ips4o/thread_pool.hpp"
|
||||||
#endif
|
#endif
|
||||||
|
@ -46,18 +49,59 @@ HWY_POP_ATTRIBUTES
|
||||||
#include "third_party/boost/allowed/sort/sort.hpp"
|
#include "third_party/boost/allowed/sort/sort.hpp"
|
||||||
#endif
|
#endif
|
||||||
#if HAVE_SORT512
|
#if HAVE_SORT512
|
||||||
#include "sort512.h"
|
#include "sort512.h" //NOLINT
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// vxsort is difficult to compile for multiple targets because it also uses
|
||||||
|
// .cpp files, and we'd also have to #undef its include guards. Instead, compile
|
||||||
|
// only for AVX2 or AVX3 depending on this macro.
|
||||||
|
#define VXSORT_AVX3 1
|
||||||
|
#if HAVE_VXSORT
|
||||||
|
// inlined from vxsort_targets_enable_avx512 (must close before end of header)
|
||||||
|
#ifdef __GNUC__
|
||||||
|
#ifdef __clang__
|
||||||
|
#if VXSORT_AVX3
|
||||||
|
#pragma clang attribute push(__attribute__((target("avx512f,avx512dq"))), \
|
||||||
|
apply_to = any(function))
|
||||||
|
#else
|
||||||
|
#pragma clang attribute push(__attribute__((target("avx2"))), \
|
||||||
|
apply_to = any(function))
|
||||||
|
#endif // VXSORT_AVX3
|
||||||
|
|
||||||
|
#else
|
||||||
|
#pragma GCC push_options
|
||||||
|
#if VXSORT_AVX3
|
||||||
|
#pragma GCC target("avx512f,avx512dq")
|
||||||
|
#else
|
||||||
|
#pragma GCC target("avx2")
|
||||||
|
#endif // VXSORT_AVX3
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if VXSORT_AVX3
|
||||||
|
#include "vxsort/machine_traits.avx512.h"
|
||||||
|
#else
|
||||||
|
#include "vxsort/machine_traits.avx2.h"
|
||||||
|
#endif // VXSORT_AVX3
|
||||||
|
#include "vxsort/vxsort.h"
|
||||||
|
#ifdef __GNUC__
|
||||||
|
#ifdef __clang__
|
||||||
|
#pragma clang attribute pop
|
||||||
|
#else
|
||||||
|
#pragma GCC pop_options
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
#endif // HAVE_VXSORT
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
|
|
||||||
enum class Dist { kUniform8, kUniform16, kUniform32 };
|
enum class Dist { kUniform8, kUniform16, kUniform32 };
|
||||||
|
|
||||||
std::vector<Dist> AllDist() {
|
static inline std::vector<Dist> AllDist() {
|
||||||
return {/*Dist::kUniform8,*/ Dist::kUniform16, Dist::kUniform32};
|
return {/*Dist::kUniform8, Dist::kUniform16,*/ Dist::kUniform32};
|
||||||
}
|
}
|
||||||
|
|
||||||
const char* DistName(Dist dist) {
|
static inline const char* DistName(Dist dist) {
|
||||||
switch (dist) {
|
switch (dist) {
|
||||||
case Dist::kUniform8:
|
case Dist::kUniform8:
|
||||||
return "uniform8";
|
return "uniform8";
|
||||||
|
@ -75,7 +119,13 @@ class InputStats {
|
||||||
void Notify(T value) {
|
void Notify(T value) {
|
||||||
min_ = std::min(min_, value);
|
min_ = std::min(min_, value);
|
||||||
max_ = std::max(max_, value);
|
max_ = std::max(max_, value);
|
||||||
sumf_ += static_cast<double>(value);
|
// Converting to integer would truncate floats, multiplying to save digits
|
||||||
|
// risks overflow especially when casting, so instead take the sum of the
|
||||||
|
// bit representations as the checksum.
|
||||||
|
uint64_t bits = 0;
|
||||||
|
static_assert(sizeof(T) <= 8, "Expected a built-in type");
|
||||||
|
CopyBytes<sizeof(T)>(&value, &bits);
|
||||||
|
sum_ += bits;
|
||||||
count_ += 1;
|
count_ += 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -86,20 +136,16 @@ class InputStats {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (min_ != other.min_ || max_ != other.max_) {
|
if (min_ != other.min_ || max_ != other.max_) {
|
||||||
HWY_ABORT("minmax %f/%f vs %f/%f\n", double(min_), double(max_),
|
HWY_ABORT("minmax %f/%f vs %f/%f\n", static_cast<double>(min_),
|
||||||
double(other.min_), double(other.max_));
|
static_cast<double>(max_), static_cast<double>(other.min_),
|
||||||
|
static_cast<double>(other.max_));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Sum helps detect duplicated/lost values
|
// Sum helps detect duplicated/lost values
|
||||||
if (sumf_ != other.sumf_) {
|
if (sum_ != other.sum_) {
|
||||||
// Allow some tolerance because kUniform32 * num can exceed double
|
HWY_ABORT("Sum mismatch %g %g; min %g max %g\n",
|
||||||
// precision.
|
static_cast<double>(sum_), static_cast<double>(other.sum_),
|
||||||
const double mul = 1E-9; // prevent destructive cancellation
|
static_cast<double>(min_), static_cast<double>(max_));
|
||||||
const double err = std::abs(sumf_ * mul - other.sumf_ * mul);
|
|
||||||
if (err > 1E-3) {
|
|
||||||
HWY_ABORT("Sum mismatch %.15e %.15e (%f) min %g max %g\n", sumf_,
|
|
||||||
other.sumf_, err, double(min_), double(max_));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
@ -108,7 +154,7 @@ class InputStats {
|
||||||
private:
|
private:
|
||||||
T min_ = hwy::HighestValue<T>();
|
T min_ = hwy::HighestValue<T>();
|
||||||
T max_ = hwy::LowestValue<T>();
|
T max_ = hwy::LowestValue<T>();
|
||||||
double sumf_ = 0.0;
|
uint64_t sum_ = 0;
|
||||||
size_t count_ = 0;
|
size_t count_ = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -127,13 +173,16 @@ enum class Algo {
|
||||||
#endif
|
#endif
|
||||||
#if HAVE_SORT512
|
#if HAVE_SORT512
|
||||||
kSort512,
|
kSort512,
|
||||||
|
#endif
|
||||||
|
#if HAVE_VXSORT
|
||||||
|
kVXSort,
|
||||||
#endif
|
#endif
|
||||||
kStd,
|
kStd,
|
||||||
kVQSort,
|
kVQSort,
|
||||||
kHeap,
|
kHeap,
|
||||||
};
|
};
|
||||||
|
|
||||||
const char* AlgoName(Algo algo) {
|
static inline const char* AlgoName(Algo algo) {
|
||||||
switch (algo) {
|
switch (algo) {
|
||||||
#if HAVE_AVX2SORT
|
#if HAVE_AVX2SORT
|
||||||
case Algo::kSEA:
|
case Algo::kSEA:
|
||||||
|
@ -154,6 +203,10 @@ const char* AlgoName(Algo algo) {
|
||||||
#if HAVE_SORT512
|
#if HAVE_SORT512
|
||||||
case Algo::kSort512:
|
case Algo::kSort512:
|
||||||
return "sort512";
|
return "sort512";
|
||||||
|
#endif
|
||||||
|
#if HAVE_VXSORT
|
||||||
|
case Algo::kVXSort:
|
||||||
|
return "vxsort";
|
||||||
#endif
|
#endif
|
||||||
case Algo::kStd:
|
case Algo::kStd:
|
||||||
return "std";
|
return "std";
|
||||||
|
@ -205,12 +258,11 @@ class Xorshift128Plus {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Need to pass in the state because vector cannot be class members.
|
// Need to pass in the state because vector cannot be class members.
|
||||||
template <class DU64>
|
template <class VU64>
|
||||||
static Vec<DU64> RandomBits(DU64 /* tag */, Vec<DU64>& state0,
|
static VU64 RandomBits(VU64& state0, VU64& state1) {
|
||||||
Vec<DU64>& state1) {
|
VU64 s1 = state0;
|
||||||
Vec<DU64> s1 = state0;
|
VU64 s0 = state1;
|
||||||
Vec<DU64> s0 = state1;
|
const VU64 bits = Add(s1, s0);
|
||||||
const Vec<DU64> bits = Add(s1, s0);
|
|
||||||
state0 = s0;
|
state0 = s0;
|
||||||
s1 = Xor(s1, ShiftLeft<23>(s1));
|
s1 = Xor(s1, ShiftLeft<23>(s1));
|
||||||
state1 = Xor(s1, Xor(s0, Xor(ShiftRight<18>(s1), ShiftRight<5>(s0))));
|
state1 = Xor(s1, Xor(s0, Xor(ShiftRight<18>(s1), ShiftRight<5>(s0))));
|
||||||
|
@ -218,30 +270,34 @@ class Xorshift128Plus {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename T, class DU64, HWY_IF_NOT_FLOAT(T)>
|
template <class D, class VU64, HWY_IF_NOT_FLOAT_D(D)>
|
||||||
Vec<DU64> RandomValues(DU64 du64, Vec<DU64>& s0, Vec<DU64>& s1,
|
Vec<D> RandomValues(D d, VU64& s0, VU64& s1, const VU64 mask) {
|
||||||
const Vec<DU64> mask) {
|
const VU64 bits = Xorshift128Plus::RandomBits(s0, s1);
|
||||||
const Vec<DU64> bits = Xorshift128Plus::RandomBits(du64, s0, s1);
|
return BitCast(d, And(bits, mask));
|
||||||
return And(bits, mask);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Important to avoid denormals, which are flushed to zero by SIMD but not
|
// It is important to avoid denormals, which are flushed to zero by SIMD but not
|
||||||
// scalar sorts, and NaN, which may be ordered differently in scalar vs. SIMD.
|
// scalar sorts, and NaN, which may be ordered differently in scalar vs. SIMD.
|
||||||
template <typename T, class DU64, HWY_IF_FLOAT(T)>
|
template <class DF, class VU64, HWY_IF_FLOAT_D(DF)>
|
||||||
Vec<DU64> RandomValues(DU64 du64, Vec<DU64>& s0, Vec<DU64>& s1,
|
Vec<DF> RandomValues(DF df, VU64& s0, VU64& s1, const VU64 mask) {
|
||||||
const Vec<DU64> mask) {
|
using TF = TFromD<DF>;
|
||||||
const Vec<DU64> bits = Xorshift128Plus::RandomBits(du64, s0, s1);
|
const RebindToUnsigned<decltype(df)> du;
|
||||||
const Vec<DU64> values = And(bits, mask);
|
using VU = Vec<decltype(du)>;
|
||||||
#if HWY_TARGET == HWY_SCALAR // Cannot repartition u64 to i32
|
|
||||||
const RebindToSigned<DU64> di;
|
const VU64 bits64 = And(Xorshift128Plus::RandomBits(s0, s1), mask);
|
||||||
|
|
||||||
|
#if HWY_TARGET == HWY_SCALAR // Cannot repartition u64 to smaller types
|
||||||
|
using TU = MakeUnsigned<TF>;
|
||||||
|
const VU bits = Set(du, static_cast<TU>(GetLane(bits64) & LimitsMax<TU>()));
|
||||||
#else
|
#else
|
||||||
const Repartition<MakeSigned<T>, DU64> di;
|
const VU bits = BitCast(du, bits64);
|
||||||
#endif
|
#endif
|
||||||
const RebindToFloat<decltype(di)> df;
|
// Avoid NaN/denormal by only generating values in [1, 2), i.e. random
|
||||||
// Avoid NaN/denormal by converting from (range-limited) integer.
|
// mantissas with the exponent taken from the representation of 1.0.
|
||||||
const Vec<DU64> no_nan =
|
const VU k1 = BitCast(du, Set(df, TF{1.0}));
|
||||||
And(values, Set(du64, MantissaMask<MakeUnsigned<T>>()));
|
const VU mantissa_mask = Set(du, MantissaMask<TF>());
|
||||||
return BitCast(du64, ConvertTo(df, BitCast(di, no_nan)));
|
const VU representation = OrAnd(k1, bits, mantissa_mask);
|
||||||
|
return BitCast(df, representation);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class DU64>
|
template <class DU64>
|
||||||
|
@ -269,29 +325,29 @@ InputStats<T> GenerateInput(const Dist dist, T* v, size_t num) {
|
||||||
SortTag<uint64_t> du64;
|
SortTag<uint64_t> du64;
|
||||||
using VU64 = Vec<decltype(du64)>;
|
using VU64 = Vec<decltype(du64)>;
|
||||||
const size_t N64 = Lanes(du64);
|
const size_t N64 = Lanes(du64);
|
||||||
auto buf = hwy::AllocateAligned<uint64_t>(2 * N64);
|
auto seeds = hwy::AllocateAligned<uint64_t>(2 * N64);
|
||||||
Xorshift128Plus::GenerateSeeds(du64, buf.get());
|
Xorshift128Plus::GenerateSeeds(du64, seeds.get());
|
||||||
auto s0 = Load(du64, buf.get());
|
VU64 s0 = Load(du64, seeds.get());
|
||||||
auto s1 = Load(du64, buf.get() + N64);
|
VU64 s1 = Load(du64, seeds.get() + N64);
|
||||||
|
|
||||||
const VU64 mask = MaskForDist(du64, dist, sizeof(T));
|
|
||||||
|
|
||||||
|
#if HWY_TARGET == HWY_SCALAR
|
||||||
|
const Sisd<T> d;
|
||||||
|
#else
|
||||||
const Repartition<T, decltype(du64)> d;
|
const Repartition<T, decltype(du64)> d;
|
||||||
|
#endif
|
||||||
|
using V = Vec<decltype(d)>;
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
|
const VU64 mask = MaskForDist(du64, dist, sizeof(T));
|
||||||
|
auto buf = hwy::AllocateAligned<T>(N);
|
||||||
|
|
||||||
size_t i = 0;
|
size_t i = 0;
|
||||||
for (; i + N <= num; i += N) {
|
for (; i + N <= num; i += N) {
|
||||||
const VU64 bits = RandomValues<T>(du64, s0, s1, mask);
|
const V values = RandomValues(d, s0, s1, mask);
|
||||||
#if HWY_ARCH_RVV
|
StoreU(values, d, v + i);
|
||||||
// v may not be 64-bit aligned
|
|
||||||
StoreU(bits, du64, buf.get());
|
|
||||||
memcpy(v + i, buf.get(), N64 * sizeof(uint64_t));
|
|
||||||
#else
|
|
||||||
StoreU(bits, du64, reinterpret_cast<uint64_t*>(v + i));
|
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
if (i < num) {
|
if (i < num) {
|
||||||
const VU64 bits = RandomValues<T>(du64, s0, s1, mask);
|
const V values = RandomValues(d, s0, s1, mask);
|
||||||
StoreU(bits, du64, buf.get());
|
StoreU(values, d, buf.get());
|
||||||
memcpy(v + i, buf.get(), (num - i) * sizeof(T));
|
memcpy(v + i, buf.get(), (num - i) * sizeof(T));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -308,18 +364,65 @@ struct ThreadLocal {
|
||||||
|
|
||||||
struct SharedState {
|
struct SharedState {
|
||||||
#if HAVE_PARALLEL_IPS4O
|
#if HAVE_PARALLEL_IPS4O
|
||||||
ips4o::StdThreadPool pool{
|
const unsigned max_threads = hwy::LimitsMax<unsigned>(); // 16 for Table 1a
|
||||||
HWY_MIN(16, static_cast<int>(std::thread::hardware_concurrency() / 2))};
|
ips4o::StdThreadPool pool{static_cast<int>(
|
||||||
|
HWY_MIN(max_threads, std::thread::hardware_concurrency() / 2))};
|
||||||
#endif
|
#endif
|
||||||
std::vector<ThreadLocal> tls{1};
|
std::vector<ThreadLocal> tls{1};
|
||||||
};
|
};
|
||||||
|
|
||||||
template <class Order, typename T>
|
// Bridge from keys (passed to Run) to lanes as expected by HeapSort. For
|
||||||
void Run(Algo algo, T* HWY_RESTRICT inout, size_t num, SharedState& shared,
|
// non-128-bit keys they are the same:
|
||||||
size_t thread) {
|
template <class Order, typename KeyType, HWY_IF_NOT_LANE_SIZE(KeyType, 16)>
|
||||||
using detail::HeapSort;
|
void CallHeapSort(KeyType* HWY_RESTRICT keys, const size_t num_keys) {
|
||||||
using detail::LaneTraits;
|
using detail::TraitsLane;
|
||||||
using detail::SharedTraits;
|
using detail::SharedTraits;
|
||||||
|
if (Order().IsAscending()) {
|
||||||
|
const SharedTraits<TraitsLane<detail::OrderAscending<KeyType>>> st;
|
||||||
|
return detail::HeapSort(st, keys, num_keys);
|
||||||
|
} else {
|
||||||
|
const SharedTraits<TraitsLane<detail::OrderDescending<KeyType>>> st;
|
||||||
|
return detail::HeapSort(st, keys, num_keys);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#if VQSORT_ENABLED
|
||||||
|
template <class Order>
|
||||||
|
void CallHeapSort(hwy::uint128_t* HWY_RESTRICT keys, const size_t num_keys) {
|
||||||
|
using detail::SharedTraits;
|
||||||
|
using detail::Traits128;
|
||||||
|
uint64_t* lanes = reinterpret_cast<uint64_t*>(keys);
|
||||||
|
const size_t num_lanes = num_keys * 2;
|
||||||
|
if (Order().IsAscending()) {
|
||||||
|
const SharedTraits<Traits128<detail::OrderAscending128>> st;
|
||||||
|
return detail::HeapSort(st, lanes, num_lanes);
|
||||||
|
} else {
|
||||||
|
const SharedTraits<Traits128<detail::OrderDescending128>> st;
|
||||||
|
return detail::HeapSort(st, lanes, num_lanes);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class Order>
|
||||||
|
void CallHeapSort(K64V64* HWY_RESTRICT keys, const size_t num_keys) {
|
||||||
|
using detail::SharedTraits;
|
||||||
|
using detail::Traits128;
|
||||||
|
uint64_t* lanes = reinterpret_cast<uint64_t*>(keys);
|
||||||
|
const size_t num_lanes = num_keys * 2;
|
||||||
|
if (Order().IsAscending()) {
|
||||||
|
const SharedTraits<Traits128<detail::OrderAscendingKV128>> st;
|
||||||
|
return detail::HeapSort(st, lanes, num_lanes);
|
||||||
|
} else {
|
||||||
|
const SharedTraits<Traits128<detail::OrderDescendingKV128>> st;
|
||||||
|
return detail::HeapSort(st, lanes, num_lanes);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif // VQSORT_ENABLED
|
||||||
|
|
||||||
|
template <class Order, typename KeyType>
|
||||||
|
void Run(Algo algo, KeyType* HWY_RESTRICT inout, size_t num,
|
||||||
|
SharedState& shared, size_t thread) {
|
||||||
|
const std::less<KeyType> less;
|
||||||
|
const std::greater<KeyType> greater;
|
||||||
|
|
||||||
switch (algo) {
|
switch (algo) {
|
||||||
#if HAVE_AVX2SORT
|
#if HAVE_AVX2SORT
|
||||||
|
@ -330,18 +433,18 @@ void Run(Algo algo, T* HWY_RESTRICT inout, size_t num, SharedState& shared,
|
||||||
#if HAVE_IPS4O
|
#if HAVE_IPS4O
|
||||||
case Algo::kIPS4O:
|
case Algo::kIPS4O:
|
||||||
if (Order().IsAscending()) {
|
if (Order().IsAscending()) {
|
||||||
return ips4o::sort(inout, inout + num, std::less<T>());
|
return ips4o::sort(inout, inout + num, less);
|
||||||
} else {
|
} else {
|
||||||
return ips4o::sort(inout, inout + num, std::greater<T>());
|
return ips4o::sort(inout, inout + num, greater);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HAVE_PARALLEL_IPS4O
|
#if HAVE_PARALLEL_IPS4O
|
||||||
case Algo::kParallelIPS4O:
|
case Algo::kParallelIPS4O:
|
||||||
if (Order().IsAscending()) {
|
if (Order().IsAscending()) {
|
||||||
return ips4o::parallel::sort(inout, inout + num, std::less<T>());
|
return ips4o::parallel::sort(inout, inout + num, less, shared.pool);
|
||||||
} else {
|
} else {
|
||||||
return ips4o::parallel::sort(inout, inout + num, std::greater<T>());
|
return ips4o::parallel::sort(inout, inout + num, greater, shared.pool);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
@ -354,33 +457,47 @@ void Run(Algo algo, T* HWY_RESTRICT inout, size_t num, SharedState& shared,
|
||||||
#if HAVE_PDQSORT
|
#if HAVE_PDQSORT
|
||||||
case Algo::kPDQ:
|
case Algo::kPDQ:
|
||||||
if (Order().IsAscending()) {
|
if (Order().IsAscending()) {
|
||||||
return boost::sort::pdqsort_branchless(inout, inout + num,
|
return boost::sort::pdqsort_branchless(inout, inout + num, less);
|
||||||
std::less<T>());
|
|
||||||
} else {
|
} else {
|
||||||
return boost::sort::pdqsort_branchless(inout, inout + num,
|
return boost::sort::pdqsort_branchless(inout, inout + num, greater);
|
||||||
std::greater<T>());
|
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if HAVE_VXSORT
|
||||||
|
case Algo::kVXSort: {
|
||||||
|
#if (VXSORT_AVX3 && HWY_TARGET != HWY_AVX3) || \
|
||||||
|
(!VXSORT_AVX3 && HWY_TARGET != HWY_AVX2)
|
||||||
|
fprintf(stderr, "Do not call for target %s\n",
|
||||||
|
hwy::TargetName(HWY_TARGET));
|
||||||
|
return;
|
||||||
|
#else
|
||||||
|
#if VXSORT_AVX3
|
||||||
|
vxsort::vxsort<KeyType, vxsort::AVX512> vx;
|
||||||
|
#else
|
||||||
|
vxsort::vxsort<KeyType, vxsort::AVX2> vx;
|
||||||
|
#endif
|
||||||
|
if (Order().IsAscending()) {
|
||||||
|
return vx.sort(inout, inout + num - 1);
|
||||||
|
} else {
|
||||||
|
fprintf(stderr, "Skipping VX - does not support descending order\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
#endif // enabled for this target
|
||||||
|
}
|
||||||
|
#endif // HAVE_VXSORT
|
||||||
|
|
||||||
case Algo::kStd:
|
case Algo::kStd:
|
||||||
if (Order().IsAscending()) {
|
if (Order().IsAscending()) {
|
||||||
return std::sort(inout, inout + num, std::less<T>());
|
return std::sort(inout, inout + num, less);
|
||||||
} else {
|
} else {
|
||||||
return std::sort(inout, inout + num, std::greater<T>());
|
return std::sort(inout, inout + num, greater);
|
||||||
}
|
}
|
||||||
|
|
||||||
case Algo::kVQSort:
|
case Algo::kVQSort:
|
||||||
return shared.tls[thread].sorter(inout, num, Order());
|
return shared.tls[thread].sorter(inout, num, Order());
|
||||||
|
|
||||||
case Algo::kHeap:
|
case Algo::kHeap:
|
||||||
HWY_ASSERT(sizeof(T) < 16);
|
return CallHeapSort<Order>(inout, num);
|
||||||
if (Order().IsAscending()) {
|
|
||||||
const SharedTraits<LaneTraits<detail::OrderAscending>> st;
|
|
||||||
return HeapSort(st, inout, num);
|
|
||||||
} else {
|
|
||||||
const SharedTraits<LaneTraits<detail::OrderDescending>> st;
|
|
||||||
return HeapSort(st, inout, num);
|
|
||||||
}
|
|
||||||
|
|
||||||
default:
|
default:
|
||||||
HWY_ABORT("Not implemented");
|
HWY_ABORT("Not implemented");
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -15,20 +16,6 @@
|
||||||
// Concurrent, independent sorts for generating more memory traffic and testing
|
// Concurrent, independent sorts for generating more memory traffic and testing
|
||||||
// scalability.
|
// scalability.
|
||||||
|
|
||||||
// clang-format off
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/bench_parallel.cc"
|
|
||||||
#include "hwy/foreach_target.h"
|
|
||||||
|
|
||||||
// After foreach_target
|
|
||||||
#include "hwy/contrib/sort/algo-inl.h"
|
|
||||||
#include "hwy/contrib/sort/result-inl.h"
|
|
||||||
#include "hwy/aligned_allocator.h"
|
|
||||||
// Last
|
|
||||||
#include "hwy/tests/test_util-inl.h"
|
|
||||||
// clang-format on
|
|
||||||
|
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
|
@ -40,18 +27,29 @@
|
||||||
#include <utility>
|
#include <utility>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
// clang-format off
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/bench_parallel.cc" //NOLINT
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
// After foreach_target
|
||||||
|
#include "hwy/contrib/sort/algo-inl.h"
|
||||||
|
#include "hwy/contrib/sort/result-inl.h"
|
||||||
|
#include "hwy/aligned_allocator.h"
|
||||||
|
// Last
|
||||||
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
// clang-format on
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
#if HWY_TARGET != HWY_SCALAR
|
|
||||||
|
|
||||||
class ThreadPool {
|
class ThreadPool {
|
||||||
public:
|
public:
|
||||||
// Starts the given number of worker threads and blocks until they are ready.
|
// Starts the given number of worker threads and blocks until they are ready.
|
||||||
explicit ThreadPool(
|
explicit ThreadPool(
|
||||||
const size_t num_threads = std::thread::hardware_concurrency() / 2)
|
const size_t num_threads = std::thread::hardware_concurrency())
|
||||||
: num_threads_(num_threads) {
|
: num_threads_(num_threads) {
|
||||||
HWY_ASSERT(num_threads_ > 0);
|
HWY_ASSERT(num_threads_ > 0);
|
||||||
threads_.reserve(num_threads_);
|
threads_.reserve(num_threads_);
|
||||||
|
@ -168,36 +166,42 @@ class ThreadPool {
|
||||||
const void* data_; // points to caller's Func
|
const void* data_; // points to caller's Func
|
||||||
};
|
};
|
||||||
|
|
||||||
template <class Order, typename T>
|
template <class Traits>
|
||||||
void RunWithoutVerify(const Dist dist, const size_t num, const Algo algo,
|
void RunWithoutVerify(Traits st, const Dist dist, const size_t num_keys,
|
||||||
SharedState& shared, size_t thread) {
|
const Algo algo, SharedState& shared, size_t thread) {
|
||||||
auto aligned = hwy::AllocateAligned<T>(num);
|
using LaneType = typename Traits::LaneType;
|
||||||
|
using KeyType = typename Traits::KeyType;
|
||||||
|
using Order = typename Traits::Order;
|
||||||
|
const size_t num_lanes = num_keys * st.LanesPerKey();
|
||||||
|
auto aligned = hwy::AllocateAligned<LaneType>(num_lanes);
|
||||||
|
|
||||||
(void)GenerateInput(dist, aligned.get(), num);
|
(void)GenerateInput(dist, aligned.get(), num_lanes);
|
||||||
|
|
||||||
const Timestamp t0;
|
const Timestamp t0;
|
||||||
Run<Order>(algo, aligned.get(), num, shared, thread);
|
Run<Order>(algo, reinterpret_cast<KeyType*>(aligned.get()), num_keys, shared,
|
||||||
HWY_ASSERT(aligned[0] < aligned[num - 1]);
|
thread);
|
||||||
|
HWY_ASSERT(aligned[0] < aligned[num_lanes - 1]);
|
||||||
}
|
}
|
||||||
|
|
||||||
void BenchParallel() {
|
void BenchParallel() {
|
||||||
// Not interested in benchmark results for other targets
|
// Not interested in benchmark results for other targets on x86
|
||||||
if (HWY_TARGET != HWY_AVX3) return;
|
if (HWY_ARCH_X86 && (HWY_TARGET != HWY_AVX2 && HWY_TARGET != HWY_AVX3)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
ThreadPool pool;
|
ThreadPool pool;
|
||||||
const size_t NT = pool.NumThreads();
|
const size_t NT = pool.NumThreads();
|
||||||
|
|
||||||
using T = int64_t;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<int64_t>>> st;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
using KeyType = typename decltype(st)::KeyType;
|
||||||
|
const size_t num_keys = size_t{100} * 1000 * 1000;
|
||||||
size_t num = 100 * 1000 * 1000;
|
|
||||||
|
|
||||||
#if HAVE_IPS4O
|
#if HAVE_IPS4O
|
||||||
const Algo algo = Algo::kIPS4O;
|
const Algo algo = Algo::kIPS4O;
|
||||||
#else
|
#else
|
||||||
const Algo algo = Algo::kVQSort;
|
const Algo algo = Algo::kVQSort;
|
||||||
#endif
|
#endif
|
||||||
const Dist dist = Dist::kUniform16;
|
const Dist dist = Dist::kUniform32;
|
||||||
|
|
||||||
SharedState shared;
|
SharedState shared;
|
||||||
shared.tls.resize(NT);
|
shared.tls.resize(NT);
|
||||||
|
@ -207,18 +211,15 @@ void BenchParallel() {
|
||||||
Timestamp t0;
|
Timestamp t0;
|
||||||
// Default capture because MSVC wants algo/dist but clang does not.
|
// Default capture because MSVC wants algo/dist but clang does not.
|
||||||
pool.RunOnThreads(nt, [=, &shared](size_t thread) {
|
pool.RunOnThreads(nt, [=, &shared](size_t thread) {
|
||||||
RunWithoutVerify<SortAscending, T>(dist, num, algo, shared, thread);
|
RunWithoutVerify(st, dist, num_keys, algo, shared, thread);
|
||||||
});
|
});
|
||||||
const double sec = SecondsSince(t0);
|
const double sec = SecondsSince(t0);
|
||||||
results.push_back(MakeResult<T>(algo, dist, st, num, nt, sec));
|
results.emplace_back(algo, dist, num_keys, nt, sec, sizeof(KeyType),
|
||||||
|
st.KeyString());
|
||||||
results.back().Print();
|
results.back().Print();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#else
|
|
||||||
void BenchParallel() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
} // namespace HWY_NAMESPACE
|
} // namespace HWY_NAMESPACE
|
||||||
|
@ -234,10 +235,4 @@ HWY_EXPORT_AND_TEST_P(BenchParallel, BenchParallel);
|
||||||
} // namespace
|
} // namespace
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,71 +13,87 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
// clang-format off
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/bench_sort.cc"
|
|
||||||
#include "hwy/foreach_target.h"
|
|
||||||
|
|
||||||
// After foreach_target
|
|
||||||
#include "hwy/contrib/sort/algo-inl.h"
|
|
||||||
#include "hwy/contrib/sort/result-inl.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
|
||||||
#include "hwy/contrib/sort/sorting_networks-inl.h" // SharedTraits
|
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
|
||||||
#include "hwy/contrib/sort/traits128-inl.h"
|
|
||||||
#include "hwy/tests/test_util-inl.h"
|
|
||||||
// clang-format on
|
|
||||||
|
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <string.h> // memcpy
|
#include <string.h> // memcpy
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
// clang-format off
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/bench_sort.cc"
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
// After foreach_target
|
||||||
|
#include "hwy/contrib/sort/algo-inl.h"
|
||||||
|
#include "hwy/contrib/sort/result-inl.h"
|
||||||
|
#include "hwy/contrib/sort/sorting_networks-inl.h" // SharedTraits
|
||||||
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
#include "hwy/contrib/sort/traits128-inl.h"
|
||||||
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
// clang-format on
|
||||||
|
|
||||||
|
// Mode for larger sorts because M1 is able to access more than the per-core
|
||||||
|
// share of L2, so 1M elements might still be in cache.
|
||||||
|
#define SORT_100M 0
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
|
// Defined within HWY_ONCE, used by BenchAllSort.
|
||||||
|
extern int64_t first_sort_target;
|
||||||
|
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace {
|
namespace {
|
||||||
using detail::LaneTraits;
|
using detail::TraitsLane;
|
||||||
using detail::OrderAscending;
|
using detail::OrderAscending;
|
||||||
using detail::OrderDescending;
|
using detail::OrderDescending;
|
||||||
using detail::SharedTraits;
|
using detail::SharedTraits;
|
||||||
|
|
||||||
#if HWY_TARGET != HWY_SCALAR
|
#if VQSORT_ENABLED || HWY_IDE
|
||||||
using detail::OrderAscending128;
|
using detail::OrderAscending128;
|
||||||
using detail::OrderDescending128;
|
|
||||||
using detail::Traits128;
|
using detail::Traits128;
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
HWY_NOINLINE void BenchPartition() {
|
HWY_NOINLINE void BenchPartition() {
|
||||||
const SortTag<T> d;
|
using LaneType = typename Traits::LaneType;
|
||||||
|
using KeyType = typename Traits::KeyType;
|
||||||
|
const SortTag<LaneType> d;
|
||||||
detail::SharedTraits<Traits> st;
|
detail::SharedTraits<Traits> st;
|
||||||
const Dist dist = Dist::kUniform8;
|
const Dist dist = Dist::kUniform8;
|
||||||
double sum = 0.0;
|
double sum = 0.0;
|
||||||
|
|
||||||
|
detail::Generator rng(&sum, 123); // for ChoosePivot
|
||||||
|
|
||||||
const size_t max_log2 = AdjustedLog2Reps(20);
|
const size_t max_log2 = AdjustedLog2Reps(20);
|
||||||
for (size_t log2 = max_log2; log2 < max_log2 + 1; ++log2) {
|
for (size_t log2 = max_log2; log2 < max_log2 + 1; ++log2) {
|
||||||
const size_t num = 1ull << log2;
|
const size_t num_lanes = 1ull << log2;
|
||||||
auto aligned = hwy::AllocateAligned<T>(num);
|
const size_t num_keys = num_lanes / st.LanesPerKey();
|
||||||
auto buf =
|
auto aligned = hwy::AllocateAligned<LaneType>(num_lanes);
|
||||||
hwy::AllocateAligned<T>(hwy::SortConstants::PartitionBufNum(Lanes(d)));
|
auto buf = hwy::AllocateAligned<LaneType>(
|
||||||
|
HWY_MAX(hwy::SortConstants::PartitionBufNum(Lanes(d)),
|
||||||
|
hwy::SortConstants::PivotBufNum(sizeof(LaneType), Lanes(d))));
|
||||||
|
|
||||||
std::vector<double> seconds;
|
std::vector<double> seconds;
|
||||||
const size_t num_reps = (1ull << (14 - log2 / 2)) * kReps;
|
const size_t num_reps = (1ull << (14 - log2 / 2)) * 30;
|
||||||
for (size_t rep = 0; rep < num_reps; ++rep) {
|
for (size_t rep = 0; rep < num_reps; ++rep) {
|
||||||
(void)GenerateInput(dist, aligned.get(), num);
|
(void)GenerateInput(dist, aligned.get(), num_lanes);
|
||||||
|
|
||||||
|
// The pivot value can influence performance. Do exactly what vqsort will
|
||||||
|
// do so that the performance (influenced by prefetching and branch
|
||||||
|
// prediction) is likely to predict the actual performance inside vqsort.
|
||||||
|
const auto pivot = detail::ChoosePivot(d, st, aligned.get(), 0, num_lanes,
|
||||||
|
buf.get(), rng);
|
||||||
|
|
||||||
const Timestamp t0;
|
const Timestamp t0;
|
||||||
|
detail::Partition(d, st, aligned.get(), 0, num_lanes - 1, pivot,
|
||||||
detail::Partition(d, st, aligned.get(), 0, num - 1, Set(d, T(128)),
|
|
||||||
buf.get());
|
buf.get());
|
||||||
seconds.push_back(SecondsSince(t0));
|
seconds.push_back(SecondsSince(t0));
|
||||||
// 'Use' the result to prevent optimizing out the partition.
|
// 'Use' the result to prevent optimizing out the partition.
|
||||||
sum += static_cast<double>(aligned.get()[num / 2]);
|
sum += static_cast<double>(aligned.get()[num_lanes / 2]);
|
||||||
}
|
}
|
||||||
|
|
||||||
MakeResult<T>(Algo::kVQSort, dist, st, num, 1,
|
Result(Algo::kVQSort, dist, num_keys, 1, SummarizeMeasurements(seconds),
|
||||||
SummarizeMeasurements(seconds))
|
sizeof(KeyType), st.KeyString())
|
||||||
.Print();
|
.Print();
|
||||||
}
|
}
|
||||||
HWY_ASSERT(sum != 999999); // Prevent optimizing out
|
HWY_ASSERT(sum != 999999); // Prevent optimizing out
|
||||||
|
@ -84,52 +101,60 @@ HWY_NOINLINE void BenchPartition() {
|
||||||
|
|
||||||
HWY_NOINLINE void BenchAllPartition() {
|
HWY_NOINLINE void BenchAllPartition() {
|
||||||
// Not interested in benchmark results for these targets
|
// Not interested in benchmark results for these targets
|
||||||
if (HWY_TARGET == HWY_SSSE3 || HWY_TARGET == HWY_SSE4 ||
|
if (HWY_TARGET == HWY_SSSE3) {
|
||||||
HWY_TARGET == HWY_AVX2) {
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
BenchPartition<LaneTraits<OrderDescending>, float>();
|
BenchPartition<TraitsLane<OrderDescending<float>>>();
|
||||||
BenchPartition<LaneTraits<OrderAscending>, int64_t>();
|
BenchPartition<TraitsLane<OrderDescending<int32_t>>>();
|
||||||
BenchPartition<Traits128<OrderDescending128>, uint64_t>();
|
BenchPartition<TraitsLane<OrderDescending<int64_t>>>();
|
||||||
|
BenchPartition<Traits128<OrderAscending128>>();
|
||||||
|
// BenchPartition<Traits128<OrderDescending128>>();
|
||||||
|
// BenchPartition<Traits128<OrderAscendingKV128>>();
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
HWY_NOINLINE void BenchBase(std::vector<Result>& results) {
|
HWY_NOINLINE void BenchBase(std::vector<Result>& results) {
|
||||||
// Not interested in benchmark results for these targets
|
// Not interested in benchmark results for these targets
|
||||||
if (HWY_TARGET == HWY_SSSE3 || HWY_TARGET == HWY_SSE4) {
|
if (HWY_TARGET == HWY_SSSE3 || HWY_TARGET == HWY_SSE4) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const SortTag<T> d;
|
using LaneType = typename Traits::LaneType;
|
||||||
|
using KeyType = typename Traits::KeyType;
|
||||||
|
const SortTag<LaneType> d;
|
||||||
detail::SharedTraits<Traits> st;
|
detail::SharedTraits<Traits> st;
|
||||||
const Dist dist = Dist::kUniform32;
|
const Dist dist = Dist::kUniform32;
|
||||||
|
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
const size_t num = SortConstants::BaseCaseNum(N);
|
const size_t num_lanes = SortConstants::BaseCaseNum(N);
|
||||||
auto keys = hwy::AllocateAligned<T>(num);
|
const size_t num_keys = num_lanes / st.LanesPerKey();
|
||||||
auto buf = hwy::AllocateAligned<T>(num + N);
|
auto keys = hwy::AllocateAligned<LaneType>(num_lanes);
|
||||||
|
auto buf = hwy::AllocateAligned<LaneType>(num_lanes + N);
|
||||||
|
|
||||||
std::vector<double> seconds;
|
std::vector<double> seconds;
|
||||||
double sum = 0; // prevents elision
|
double sum = 0; // prevents elision
|
||||||
constexpr size_t kMul = AdjustedReps(600); // ensures long enough to measure
|
constexpr size_t kMul = AdjustedReps(600); // ensures long enough to measure
|
||||||
|
|
||||||
for (size_t rep = 0; rep < kReps; ++rep) {
|
for (size_t rep = 0; rep < 30; ++rep) {
|
||||||
InputStats<T> input_stats = GenerateInput(dist, keys.get(), num);
|
InputStats<LaneType> input_stats =
|
||||||
|
GenerateInput(dist, keys.get(), num_lanes);
|
||||||
|
|
||||||
const Timestamp t0;
|
const Timestamp t0;
|
||||||
for (size_t i = 0; i < kMul; ++i) {
|
for (size_t i = 0; i < kMul; ++i) {
|
||||||
detail::BaseCase(d, st, keys.get(), num, buf.get());
|
detail::BaseCase(d, st, keys.get(), keys.get() + num_lanes, num_lanes,
|
||||||
|
buf.get());
|
||||||
sum += static_cast<double>(keys[0]);
|
sum += static_cast<double>(keys[0]);
|
||||||
}
|
}
|
||||||
seconds.push_back(SecondsSince(t0));
|
seconds.push_back(SecondsSince(t0));
|
||||||
// printf("%f\n", seconds.back());
|
// printf("%f\n", seconds.back());
|
||||||
|
|
||||||
HWY_ASSERT(VerifySort(st, input_stats, keys.get(), num, "BenchBase"));
|
HWY_ASSERT(VerifySort(st, input_stats, keys.get(), num_lanes, "BenchBase"));
|
||||||
}
|
}
|
||||||
HWY_ASSERT(sum < 1E99);
|
HWY_ASSERT(sum < 1E99);
|
||||||
results.push_back(MakeResult<T>(Algo::kVQSort, dist, st, num * kMul, 1,
|
results.emplace_back(Algo::kVQSort, dist, num_keys * kMul, 1,
|
||||||
SummarizeMeasurements(seconds)));
|
SummarizeMeasurements(seconds), sizeof(KeyType),
|
||||||
|
st.KeyString());
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_NOINLINE void BenchAllBase() {
|
HWY_NOINLINE void BenchAllBase() {
|
||||||
|
@ -139,14 +164,19 @@ HWY_NOINLINE void BenchAllBase() {
|
||||||
}
|
}
|
||||||
|
|
||||||
std::vector<Result> results;
|
std::vector<Result> results;
|
||||||
BenchBase<LaneTraits<OrderAscending>, float>(results);
|
BenchBase<TraitsLane<OrderAscending<float>>>(results);
|
||||||
BenchBase<LaneTraits<OrderDescending>, int64_t>(results);
|
BenchBase<TraitsLane<OrderDescending<int64_t>>>(results);
|
||||||
BenchBase<Traits128<OrderAscending128>, uint64_t>(results);
|
BenchBase<Traits128<OrderAscending128>>(results);
|
||||||
for (const Result& r : results) {
|
for (const Result& r : results) {
|
||||||
r.Print();
|
r.Print();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
void BenchAllPartition() {}
|
||||||
|
void BenchAllBase() {}
|
||||||
|
#endif // VQSORT_ENABLED
|
||||||
|
|
||||||
std::vector<Algo> AlgoForBench() {
|
std::vector<Algo> AlgoForBench() {
|
||||||
return {
|
return {
|
||||||
#if HAVE_AVX2SORT
|
#if HAVE_AVX2SORT
|
||||||
|
@ -154,8 +184,7 @@ std::vector<Algo> AlgoForBench() {
|
||||||
#endif
|
#endif
|
||||||
#if HAVE_PARALLEL_IPS4O
|
#if HAVE_PARALLEL_IPS4O
|
||||||
Algo::kParallelIPS4O,
|
Algo::kParallelIPS4O,
|
||||||
#endif
|
#elif HAVE_IPS4O
|
||||||
#if HAVE_IPS4O
|
|
||||||
Algo::kIPS4O,
|
Algo::kIPS4O,
|
||||||
#endif
|
#endif
|
||||||
#if HAVE_PDQSORT
|
#if HAVE_PDQSORT
|
||||||
|
@ -164,33 +193,64 @@ std::vector<Algo> AlgoForBench() {
|
||||||
#if HAVE_SORT512
|
#if HAVE_SORT512
|
||||||
Algo::kSort512,
|
Algo::kSort512,
|
||||||
#endif
|
#endif
|
||||||
// Algo::kStd, // too slow to always benchmark
|
// Only include if we're compiling for the target it supports.
|
||||||
// Algo::kHeap, // too slow to always benchmark
|
#if HAVE_VXSORT && ((VXSORT_AVX3 && HWY_TARGET == HWY_AVX3) || \
|
||||||
Algo::kVQSort,
|
(!VXSORT_AVX3 && HWY_TARGET == HWY_AVX2))
|
||||||
|
Algo::kVXSort,
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if !HAVE_PARALLEL_IPS4O
|
||||||
|
#if !SORT_100M
|
||||||
|
// These are 10-20x slower, but that's OK for the default size when we
|
||||||
|
// are not testing the parallel nor 100M modes.
|
||||||
|
Algo::kStd, Algo::kHeap,
|
||||||
|
#endif
|
||||||
|
|
||||||
|
Algo::kVQSort, // only ~4x slower, but not required for Table 1a
|
||||||
|
#endif
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
HWY_NOINLINE void BenchSort(size_t num) {
|
HWY_NOINLINE void BenchSort(size_t num_keys) {
|
||||||
|
if (first_sort_target == 0) first_sort_target = HWY_TARGET;
|
||||||
|
|
||||||
SharedState shared;
|
SharedState shared;
|
||||||
detail::SharedTraits<Traits> st;
|
detail::SharedTraits<Traits> st;
|
||||||
auto aligned = hwy::AllocateAligned<T>(num);
|
using Order = typename Traits::Order;
|
||||||
|
using LaneType = typename Traits::LaneType;
|
||||||
|
using KeyType = typename Traits::KeyType;
|
||||||
|
const size_t num_lanes = num_keys * st.LanesPerKey();
|
||||||
|
auto aligned = hwy::AllocateAligned<LaneType>(num_lanes);
|
||||||
|
|
||||||
|
const size_t reps = num_keys > 1000 * 1000 ? 10 : 30;
|
||||||
|
|
||||||
for (Algo algo : AlgoForBench()) {
|
for (Algo algo : AlgoForBench()) {
|
||||||
|
// Other algorithms don't depend on the vector instructions, so only run
|
||||||
|
// them for the first target.
|
||||||
|
#if !HAVE_VXSORT
|
||||||
|
if (algo != Algo::kVQSort && HWY_TARGET != first_sort_target) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
for (Dist dist : AllDist()) {
|
for (Dist dist : AllDist()) {
|
||||||
std::vector<double> seconds;
|
std::vector<double> seconds;
|
||||||
for (size_t rep = 0; rep < kReps; ++rep) {
|
for (size_t rep = 0; rep < reps; ++rep) {
|
||||||
InputStats<T> input_stats = GenerateInput(dist, aligned.get(), num);
|
InputStats<LaneType> input_stats =
|
||||||
|
GenerateInput(dist, aligned.get(), num_lanes);
|
||||||
|
|
||||||
const Timestamp t0;
|
const Timestamp t0;
|
||||||
Run<typename Traits::Order>(algo, aligned.get(), num, shared,
|
Run<Order>(algo, reinterpret_cast<KeyType*>(aligned.get()), num_keys,
|
||||||
/*thread=*/0);
|
shared, /*thread=*/0);
|
||||||
seconds.push_back(SecondsSince(t0));
|
seconds.push_back(SecondsSince(t0));
|
||||||
// printf("%f\n", seconds.back());
|
// printf("%f\n", seconds.back());
|
||||||
|
|
||||||
HWY_ASSERT(
|
HWY_ASSERT(
|
||||||
VerifySort(st, input_stats, aligned.get(), num, "BenchSort"));
|
VerifySort(st, input_stats, aligned.get(), num_lanes, "BenchSort"));
|
||||||
}
|
}
|
||||||
MakeResult<T>(algo, dist, st, num, 1, SummarizeMeasurements(seconds))
|
Result(algo, dist, num_keys, 1, SummarizeMeasurements(seconds),
|
||||||
|
sizeof(KeyType), st.KeyString())
|
||||||
.Print();
|
.Print();
|
||||||
} // dist
|
} // dist
|
||||||
} // algo
|
} // algo
|
||||||
|
@ -198,41 +258,40 @@ HWY_NOINLINE void BenchSort(size_t num) {
|
||||||
|
|
||||||
HWY_NOINLINE void BenchAllSort() {
|
HWY_NOINLINE void BenchAllSort() {
|
||||||
// Not interested in benchmark results for these targets
|
// Not interested in benchmark results for these targets
|
||||||
if (HWY_TARGET == HWY_SSSE3 || HWY_TARGET == HWY_SSE4) {
|
if (HWY_TARGET == HWY_SSSE3 || HWY_TARGET == HWY_SSE4 ||
|
||||||
|
HWY_TARGET == HWY_EMU128) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
// Only enable EMU128 on x86 - it's slow on emulators.
|
||||||
|
if (!HWY_ARCH_X86 && (HWY_TARGET == HWY_EMU128)) return;
|
||||||
|
|
||||||
constexpr size_t K = 1000;
|
constexpr size_t K = 1000;
|
||||||
constexpr size_t M = K * K;
|
constexpr size_t M = K * K;
|
||||||
(void)K;
|
(void)K;
|
||||||
(void)M;
|
(void)M;
|
||||||
for (size_t num : {
|
for (size_t num_keys : {
|
||||||
#if HAVE_PARALLEL_IPS4O
|
#if HAVE_PARALLEL_IPS4O || SORT_100M
|
||||||
100 * M,
|
100 * M,
|
||||||
#else
|
#else
|
||||||
AdjustedReps(1 * M),
|
1 * M,
|
||||||
#endif
|
#endif
|
||||||
}) {
|
}) {
|
||||||
// BenchSort<LaneTraits<OrderAscending>, float>(num);
|
BenchSort<TraitsLane<OrderAscending<float>>>(num_keys);
|
||||||
// BenchSort<LaneTraits<OrderDescending>, double>(num);
|
// BenchSort<TraitsLane<OrderDescending<double>>>(num_keys);
|
||||||
// BenchSort<LaneTraits<OrderAscending>, int16_t>(num);
|
// BenchSort<TraitsLane<OrderAscending<int16_t>>>(num_keys);
|
||||||
BenchSort<LaneTraits<OrderDescending>, int32_t>(num);
|
BenchSort<TraitsLane<OrderDescending<int32_t>>>(num_keys);
|
||||||
BenchSort<LaneTraits<OrderAscending>, int64_t>(num);
|
BenchSort<TraitsLane<OrderAscending<int64_t>>>(num_keys);
|
||||||
// BenchSort<LaneTraits<OrderDescending>, uint16_t>(num);
|
// BenchSort<TraitsLane<OrderDescending<uint16_t>>>(num_keys);
|
||||||
// BenchSort<LaneTraits<OrderDescending>, uint32_t>(num);
|
// BenchSort<TraitsLane<OrderDescending<uint32_t>>>(num_keys);
|
||||||
// BenchSort<LaneTraits<OrderAscending>, uint64_t>(num);
|
// BenchSort<TraitsLane<OrderAscending<uint64_t>>>(num_keys);
|
||||||
|
|
||||||
BenchSort<Traits128<OrderAscending128>, uint64_t>(num);
|
#if !HAVE_VXSORT && VQSORT_ENABLED
|
||||||
// BenchSort<Traits128<OrderAscending128>, uint64_t>(num);
|
BenchSort<Traits128<OrderAscending128>>(num_keys);
|
||||||
|
// BenchSort<Traits128<OrderAscendingKV128>>(num_keys);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#else
|
|
||||||
void BenchAllPartition() {}
|
|
||||||
void BenchAllBase() {}
|
|
||||||
void BenchAllSort() {}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
} // namespace HWY_NAMESPACE
|
} // namespace HWY_NAMESPACE
|
||||||
|
@ -242,6 +301,7 @@ HWY_AFTER_NAMESPACE();
|
||||||
#if HWY_ONCE
|
#if HWY_ONCE
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
|
int64_t first_sort_target = 0; // none run yet
|
||||||
namespace {
|
namespace {
|
||||||
HWY_BEFORE_TEST(BenchSort);
|
HWY_BEFORE_TEST(BenchSort);
|
||||||
HWY_EXPORT_AND_TEST_P(BenchSort, BenchAllPartition);
|
HWY_EXPORT_AND_TEST_P(BenchSort, BenchAllPartition);
|
||||||
|
@ -250,10 +310,4 @@ HWY_EXPORT_AND_TEST_P(BenchSort, BenchAllSort);
|
||||||
} // namespace
|
} // namespace
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
|
@ -1,30 +0,0 @@
|
||||||
// Copyright 2022 Google LLC
|
|
||||||
//
|
|
||||||
// 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.
|
|
||||||
|
|
||||||
// Speed up MSVC builds by building fewer targets. This header must be included
|
|
||||||
// from all TUs that contain a HWY_DYNAMIC_DISPATCH to vqsort, i.e. vqsort_*.cc.
|
|
||||||
// However, users of vqsort.h are unaffected.
|
|
||||||
|
|
||||||
#ifndef HIGHWAY_HWY_CONTRIB_SORT_DISABLED_TARGETS_H_
|
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_DISABLED_TARGETS_H_
|
|
||||||
|
|
||||||
#include "hwy/base.h"
|
|
||||||
|
|
||||||
#if HWY_COMPILER_MSVC
|
|
||||||
#undef HWY_DISABLED_TARGETS
|
|
||||||
// HWY_SCALAR remains, so there will still be a valid target to call.
|
|
||||||
#define HWY_DISABLED_TARGETS (HWY_SSSE3 | HWY_SSE4)
|
|
||||||
#endif // HWY_COMPILER_MSVC
|
|
||||||
|
|
||||||
#endif // HIGHWAY_HWY_CONTRIB_SORT_DISABLED_TARGETS_H_
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -33,20 +34,19 @@ struct Timestamp {
|
||||||
double t;
|
double t;
|
||||||
};
|
};
|
||||||
|
|
||||||
double SecondsSince(const Timestamp& t0) {
|
static inline double SecondsSince(const Timestamp& t0) {
|
||||||
const Timestamp t1;
|
const Timestamp t1;
|
||||||
return t1.t - t0.t;
|
return t1.t - t0.t;
|
||||||
}
|
}
|
||||||
|
|
||||||
constexpr size_t kReps = 30;
|
|
||||||
|
|
||||||
// Returns trimmed mean (we don't want to run an out-of-L3-cache sort often
|
// Returns trimmed mean (we don't want to run an out-of-L3-cache sort often
|
||||||
// enough for the mode to be reliable).
|
// enough for the mode to be reliable).
|
||||||
double SummarizeMeasurements(std::vector<double>& seconds) {
|
static inline double SummarizeMeasurements(std::vector<double>& seconds) {
|
||||||
std::sort(seconds.begin(), seconds.end());
|
std::sort(seconds.begin(), seconds.end());
|
||||||
double sum = 0;
|
double sum = 0;
|
||||||
int count = 0;
|
int count = 0;
|
||||||
for (size_t i = kReps / 4; i < seconds.size() - kReps / 2; ++i) {
|
const size_t num = seconds.size();
|
||||||
|
for (size_t i = num / 4; i < num / 2; ++i) {
|
||||||
sum += seconds[i];
|
sum += seconds[i];
|
||||||
count += 1;
|
count += 1;
|
||||||
}
|
}
|
||||||
|
@ -71,72 +71,62 @@ namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
struct Result {
|
struct Result {
|
||||||
Result() {}
|
Result() {}
|
||||||
Result(const uint32_t target, const Algo algo, Dist dist, bool is128,
|
Result(const Algo algo, Dist dist, size_t num_keys, size_t num_threads,
|
||||||
size_t num, size_t num_threads, double sec, size_t sizeof_t,
|
double sec, size_t sizeof_key, const std::string& key_name)
|
||||||
const char* type_name)
|
: target(HWY_TARGET),
|
||||||
: target(target),
|
|
||||||
algo(algo),
|
algo(algo),
|
||||||
dist(dist),
|
dist(dist),
|
||||||
is128(is128),
|
num_keys(num_keys),
|
||||||
num(num),
|
|
||||||
num_threads(num_threads),
|
num_threads(num_threads),
|
||||||
sec(sec),
|
sec(sec),
|
||||||
sizeof_t(sizeof_t),
|
sizeof_key(sizeof_key),
|
||||||
type_name(type_name) {}
|
key_name(key_name) {}
|
||||||
|
|
||||||
void Print() const {
|
void Print() const {
|
||||||
const double bytes = static_cast<double>(num) *
|
const double bytes = static_cast<double>(num_keys) *
|
||||||
static_cast<double>(num_threads) *
|
static_cast<double>(num_threads) *
|
||||||
static_cast<double>(sizeof_t);
|
static_cast<double>(sizeof_key);
|
||||||
printf("%10s: %12s: %7s: %9s: %.2E %4.0f MB/s (%2zu threads)\n",
|
printf("%10s: %12s: %7s: %9s: %.2E %4.0f MB/s (%2zu threads)\n",
|
||||||
hwy::TargetName(target), AlgoName(algo),
|
hwy::TargetName(target), AlgoName(algo), key_name.c_str(),
|
||||||
is128 ? "u128" : type_name.c_str(), DistName(dist),
|
DistName(dist), static_cast<double>(num_keys), bytes * 1E-6 / sec,
|
||||||
static_cast<double>(num), bytes * 1E-6 / sec, num_threads);
|
num_threads);
|
||||||
}
|
}
|
||||||
|
|
||||||
uint32_t target;
|
int64_t target;
|
||||||
Algo algo;
|
Algo algo;
|
||||||
Dist dist;
|
Dist dist;
|
||||||
bool is128;
|
size_t num_keys = 0;
|
||||||
size_t num = 0;
|
|
||||||
size_t num_threads = 0;
|
size_t num_threads = 0;
|
||||||
double sec = 0.0;
|
double sec = 0.0;
|
||||||
size_t sizeof_t = 0;
|
size_t sizeof_key = 0;
|
||||||
std::string type_name;
|
std::string key_name;
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename T, class Traits>
|
template <class Traits, typename LaneType>
|
||||||
Result MakeResult(const Algo algo, Dist dist, Traits st, size_t num,
|
bool VerifySort(Traits st, const InputStats<LaneType>& input_stats,
|
||||||
size_t num_threads, double sec) {
|
const LaneType* out, size_t num_lanes, const char* caller) {
|
||||||
char string100[100];
|
constexpr size_t N1 = st.LanesPerKey();
|
||||||
hwy::detail::TypeName(hwy::detail::MakeTypeInfo<T>(), 1, string100);
|
HWY_ASSERT(num_lanes >= N1);
|
||||||
return Result(HWY_TARGET, algo, dist, st.Is128(), num, num_threads, sec,
|
|
||||||
sizeof(T), string100);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class Traits, typename T>
|
InputStats<LaneType> output_stats;
|
||||||
bool VerifySort(Traits st, const InputStats<T>& input_stats, const T* out,
|
|
||||||
size_t num, const char* caller) {
|
|
||||||
constexpr size_t N1 = st.Is128() ? 2 : 1;
|
|
||||||
HWY_ASSERT(num >= N1);
|
|
||||||
|
|
||||||
InputStats<T> output_stats;
|
|
||||||
// Ensure it matches the sort order
|
// Ensure it matches the sort order
|
||||||
for (size_t i = 0; i < num - N1; i += N1) {
|
for (size_t i = 0; i < num_lanes - N1; i += N1) {
|
||||||
output_stats.Notify(out[i]);
|
output_stats.Notify(out[i]);
|
||||||
if (N1 == 2) output_stats.Notify(out[i + 1]);
|
if (N1 == 2) output_stats.Notify(out[i + 1]);
|
||||||
// Reverse order instead of checking !Compare1 so we accept equal keys.
|
// Reverse order instead of checking !Compare1 so we accept equal keys.
|
||||||
if (st.Compare1(out + i + N1, out + i)) {
|
if (st.Compare1(out + i + N1, out + i)) {
|
||||||
printf("%s: i=%d of %d: N1=%d %5.0f %5.0f vs. %5.0f %5.0f\n\n", caller,
|
printf("%s: i=%d of %d lanes: N1=%d %5.0f %5.0f vs. %5.0f %5.0f\n\n",
|
||||||
static_cast<int>(i), static_cast<int>(num), static_cast<int>(N1),
|
caller, static_cast<int>(i), static_cast<int>(num_lanes),
|
||||||
double(out[i + 1]), double(out[i + 0]), double(out[i + N1 + 1]),
|
static_cast<int>(N1), static_cast<double>(out[i + 1]),
|
||||||
double(out[i + N1]));
|
static_cast<double>(out[i + 0]),
|
||||||
|
static_cast<double>(out[i + N1 + 1]),
|
||||||
|
static_cast<double>(out[i + N1]));
|
||||||
HWY_ABORT("%d-bit sort is incorrect\n",
|
HWY_ABORT("%d-bit sort is incorrect\n",
|
||||||
static_cast<int>(sizeof(T) * 8 * N1));
|
static_cast<int>(sizeof(LaneType) * 8 * N1));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
output_stats.Notify(out[num - N1]);
|
output_stats.Notify(out[num_lanes - N1]);
|
||||||
if (N1 == 2) output_stats.Notify(out[num - N1 + 1]);
|
if (N1 == 2) output_stats.Notify(out[num_lanes - N1 + 1]);
|
||||||
|
|
||||||
return input_stats == output_stats;
|
return input_stats == output_stats;
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -27,8 +28,8 @@ namespace hwy {
|
||||||
struct SortConstants {
|
struct SortConstants {
|
||||||
// SortingNetwork reshapes its input into a matrix. This is the maximum number
|
// SortingNetwork reshapes its input into a matrix. This is the maximum number
|
||||||
// of *keys* per vector.
|
// of *keys* per vector.
|
||||||
#if HWY_COMPILER_MSVC
|
#if HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
||||||
static constexpr size_t kMaxCols = 8; // avoids build timeout
|
static constexpr size_t kMaxCols = 8; // avoid build timeout/stack overflow
|
||||||
#else
|
#else
|
||||||
static constexpr size_t kMaxCols = 16; // enough for u32 in 512-bit vector
|
static constexpr size_t kMaxCols = 16; // enough for u32 in 512-bit vector
|
||||||
#endif
|
#endif
|
||||||
|
@ -41,7 +42,7 @@ struct SortConstants {
|
||||||
static constexpr size_t kMaxRowsLog2 = 4;
|
static constexpr size_t kMaxRowsLog2 = 4;
|
||||||
static constexpr size_t kMaxRows = size_t{1} << kMaxRowsLog2;
|
static constexpr size_t kMaxRows = size_t{1} << kMaxRowsLog2;
|
||||||
|
|
||||||
static HWY_INLINE size_t BaseCaseNum(size_t N) {
|
static constexpr HWY_INLINE size_t BaseCaseNum(size_t N) {
|
||||||
return kMaxRows * HWY_MIN(N, kMaxCols);
|
return kMaxRows * HWY_MIN(N, kMaxCols);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -52,7 +53,7 @@ struct SortConstants {
|
||||||
// To change, must also update left + 3 * N etc. in the loop.
|
// To change, must also update left + 3 * N etc. in the loop.
|
||||||
static constexpr size_t kPartitionUnroll = 4;
|
static constexpr size_t kPartitionUnroll = 4;
|
||||||
|
|
||||||
static HWY_INLINE size_t PartitionBufNum(size_t N) {
|
static constexpr HWY_INLINE size_t PartitionBufNum(size_t N) {
|
||||||
// The main loop reads kPartitionUnroll vectors, and first loads from
|
// The main loop reads kPartitionUnroll vectors, and first loads from
|
||||||
// both left and right beforehand, so it requires min = 2 *
|
// both left and right beforehand, so it requires min = 2 *
|
||||||
// kPartitionUnroll vectors. To handle smaller amounts (only guaranteed
|
// kPartitionUnroll vectors. To handle smaller amounts (only guaranteed
|
||||||
|
@ -64,9 +65,26 @@ struct SortConstants {
|
||||||
// Chunk := group of keys loaded for sampling a pivot. Matches the typical
|
// Chunk := group of keys loaded for sampling a pivot. Matches the typical
|
||||||
// cache line size of 64 bytes to get maximum benefit per L2 miss. If vectors
|
// cache line size of 64 bytes to get maximum benefit per L2 miss. If vectors
|
||||||
// are larger, use entire vectors to ensure we do not overrun the array.
|
// are larger, use entire vectors to ensure we do not overrun the array.
|
||||||
static HWY_INLINE size_t LanesPerChunk(size_t sizeof_t, size_t N) {
|
static constexpr HWY_INLINE size_t LanesPerChunk(size_t sizeof_t, size_t N) {
|
||||||
return HWY_MAX(64 / sizeof_t, N);
|
return HWY_MAX(64 / sizeof_t, N);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static constexpr HWY_INLINE size_t PivotBufNum(size_t sizeof_t, size_t N) {
|
||||||
|
// 3 chunks of medians, 1 chunk of median medians plus two padding vectors.
|
||||||
|
return (3 + 1) * LanesPerChunk(sizeof_t, N) + 2 * N;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
static constexpr HWY_INLINE size_t BufNum(size_t N) {
|
||||||
|
// One extra for padding plus another for full-vector loads.
|
||||||
|
return HWY_MAX(BaseCaseNum(N) + 2 * N,
|
||||||
|
HWY_MAX(PartitionBufNum(N), PivotBufNum(sizeof(T), N)));
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
static constexpr HWY_INLINE size_t BufBytes(size_t vector_size) {
|
||||||
|
return sizeof(T) * BufNum<T>(vector_size / sizeof(T));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
@ -84,12 +102,23 @@ struct SortConstants {
|
||||||
|
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
// vqsort isn't available on HWY_SCALAR, and builds time out on MSVC opt and
|
||||||
|
// Arm v7 debug.
|
||||||
|
#undef VQSORT_ENABLED
|
||||||
|
#if (HWY_TARGET == HWY_SCALAR) || \
|
||||||
|
(HWY_COMPILER_MSVC && !HWY_IS_DEBUG_BUILD) || \
|
||||||
|
(HWY_ARCH_ARM_V7 && HWY_IS_DEBUG_BUILD)
|
||||||
|
#define VQSORT_ENABLED 0
|
||||||
|
#else
|
||||||
|
#define VQSORT_ENABLED 1
|
||||||
|
#endif
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
// Default tag / vector width selector.
|
// Default tag / vector width selector.
|
||||||
// TODO(janwas): enable once LMUL < 1 is supported.
|
#if HWY_TARGET == HWY_RVV
|
||||||
#if HWY_TARGET == HWY_RVV && 0
|
// Use LMUL = 1/2; for SEW=64 this ends up emulated via vsetvl.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
using SortTag = ScalableTag<T, -1>;
|
using SortTag = ScalableTag<T, -1>;
|
||||||
#else
|
#else
|
||||||
|
|
|
@ -1,909 +0,0 @@
|
||||||
// Copyright 2021 Google LLC
|
|
||||||
//
|
|
||||||
// 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.
|
|
||||||
|
|
||||||
// Per-target include guard
|
|
||||||
|
|
||||||
#if defined(HIGHWAY_HWY_CONTRIB_SORT_SORT_INL_H_) == \
|
|
||||||
defined(HWY_TARGET_TOGGLE)
|
|
||||||
#ifdef HIGHWAY_HWY_CONTRIB_SORT_SORT_INL_H_
|
|
||||||
#undef HIGHWAY_HWY_CONTRIB_SORT_SORT_INL_H_
|
|
||||||
#else
|
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_SORT_INL_H_
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <inttypes.h>
|
|
||||||
|
|
||||||
#include "hwy/aligned_allocator.h"
|
|
||||||
#include "hwy/highway.h"
|
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
|
||||||
namespace hwy {
|
|
||||||
namespace HWY_NAMESPACE {
|
|
||||||
|
|
||||||
enum class SortOrder { kAscending, kDescending };
|
|
||||||
|
|
||||||
#if HWY_TARGET != HWY_SCALAR && HWY_ARCH_X86
|
|
||||||
|
|
||||||
#define HWY_SORT_VERIFY 1
|
|
||||||
|
|
||||||
constexpr inline SortOrder Reverse(SortOrder order) {
|
|
||||||
return (order == SortOrder::kAscending) ? SortOrder::kDescending
|
|
||||||
: SortOrder::kAscending;
|
|
||||||
}
|
|
||||||
|
|
||||||
namespace verify {
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
bool Compare(T a, T b, SortOrder kOrder) {
|
|
||||||
if (kOrder == SortOrder::kAscending) return a <= b;
|
|
||||||
return a >= b;
|
|
||||||
}
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
class Runs {
|
|
||||||
using T = TFromD<D>;
|
|
||||||
|
|
||||||
public:
|
|
||||||
Runs(D d, size_t num_regs, size_t run_length = 0, bool alternating = false) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
|
|
||||||
buf_ = AllocateAligned<T>(N);
|
|
||||||
consecutive_ = AllocateAligned<T>(num_regs * N);
|
|
||||||
|
|
||||||
num_regs_ = num_regs;
|
|
||||||
if (run_length) {
|
|
||||||
run_length_ = run_length;
|
|
||||||
num_runs_ = num_regs * N / run_length;
|
|
||||||
is_vector_ = true;
|
|
||||||
alternating_ = alternating;
|
|
||||||
} else {
|
|
||||||
run_length_ = num_regs * 4;
|
|
||||||
num_runs_ = N / 4;
|
|
||||||
is_vector_ = false;
|
|
||||||
alternating_ = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void ScatterQuartets(D d, const size_t idx_reg, Vec<D> v) {
|
|
||||||
HWY_ASSERT(idx_reg < num_regs_);
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
for (size_t i = 0; i < N; i += 4) {
|
|
||||||
Store(v, d, buf_.get());
|
|
||||||
const size_t idx_q = (i / 4) * num_regs_ + idx_reg;
|
|
||||||
CopyBytes<16>(buf_.get() + i, consecutive_.get() + idx_q * 4);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void StoreVector(D d, const size_t idx_reg, Vec<D> v) {
|
|
||||||
HWY_ASSERT(idx_reg < num_regs_);
|
|
||||||
Store(v, d, &consecutive_[idx_reg * Lanes(d)]);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool IsBitonic() const {
|
|
||||||
HWY_ASSERT(!alternating_);
|
|
||||||
for (size_t ir = 0; ir < num_runs_; ++ir) {
|
|
||||||
const T* p = &consecutive_[ir * run_length_];
|
|
||||||
bool is_asc = true;
|
|
||||||
bool is_desc = true;
|
|
||||||
bool is_zero = true;
|
|
||||||
|
|
||||||
for (size_t i = 0; i < run_length_ / 2 - 1; ++i) {
|
|
||||||
is_asc &= (p[i] <= p[i + 1]);
|
|
||||||
is_desc &= (p[i] >= p[i + 1]);
|
|
||||||
}
|
|
||||||
for (size_t i = 0; i < run_length_; ++i) {
|
|
||||||
is_zero &= (p[i] == 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool is_asc2 = true;
|
|
||||||
bool is_desc2 = true;
|
|
||||||
for (size_t i = run_length_ / 2; i < run_length_ - 1; ++i) {
|
|
||||||
is_asc2 &= (p[i] <= p[i + 1]);
|
|
||||||
is_desc2 &= (p[i] >= p[i + 1]);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (is_zero) continue;
|
|
||||||
if (is_asc && is_desc2) continue;
|
|
||||||
if (is_desc && is_asc2) continue;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void CheckBitonic(int line, int caller) const {
|
|
||||||
if (IsBitonic()) return;
|
|
||||||
for (size_t ir = 0; ir < num_runs_; ++ir) {
|
|
||||||
const T* p = &consecutive_[ir * run_length_];
|
|
||||||
printf("run %" PRIu64 " (len %" PRIu64 ")\n", static_cast<uint64_t>(ir),
|
|
||||||
static_cast<uint64_t>(run_length_));
|
|
||||||
for (size_t i = 0; i < run_length_; ++i) {
|
|
||||||
printf("%.0f\n", static_cast<float>(p[i]));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
printf("caller %d\n", caller);
|
|
||||||
hwy::Abort("", line, "not bitonic");
|
|
||||||
}
|
|
||||||
|
|
||||||
void CheckSorted(SortOrder kOrder, int line, int caller) const {
|
|
||||||
for (size_t ir = 0; ir < num_runs_; ++ir) {
|
|
||||||
const SortOrder order =
|
|
||||||
(alternating_ && (ir & 1)) ? Reverse(kOrder) : kOrder;
|
|
||||||
const T* p = &consecutive_[ir * run_length_];
|
|
||||||
|
|
||||||
for (size_t i = 0; i < run_length_ - 1; ++i) {
|
|
||||||
if (!Compare(p[i], p[i + 1], order)) {
|
|
||||||
printf("ir%" PRIu64 " run_length=%" PRIu64
|
|
||||||
" alt=%d original order=%d this order=%d\n",
|
|
||||||
static_cast<uint64_t>(ir), static_cast<uint64_t>(run_length_),
|
|
||||||
alternating_, static_cast<int>(kOrder),
|
|
||||||
static_cast<int>(order));
|
|
||||||
for (size_t i = 0; i < run_length_; ++i) {
|
|
||||||
printf(" %.0f\n", static_cast<float>(p[i]));
|
|
||||||
}
|
|
||||||
printf("caller %d\n", caller);
|
|
||||||
hwy::Abort("", line, "not sorted");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
AlignedFreeUniquePtr<T[]> buf_;
|
|
||||||
AlignedFreeUniquePtr<T[]> consecutive_;
|
|
||||||
size_t num_regs_;
|
|
||||||
size_t run_length_;
|
|
||||||
size_t num_runs_;
|
|
||||||
bool is_vector_;
|
|
||||||
bool alternating_;
|
|
||||||
};
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreDeinterleavedQuartets(D d, Vec<D> v0) {
|
|
||||||
Runs<D> runs(d, 1);
|
|
||||||
runs.ScatterQuartets(d, 0, v0);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreDeinterleavedQuartets(D d, Vec<D> v0, Vec<D> v1) {
|
|
||||||
Runs<D> runs(d, 2);
|
|
||||||
runs.ScatterQuartets(d, 0, v0);
|
|
||||||
runs.ScatterQuartets(d, 1, v1);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreDeinterleavedQuartets(D d, Vec<D> v0, Vec<D> v1, Vec<D> v2,
|
|
||||||
Vec<D> v3) {
|
|
||||||
Runs<D> runs(d, 4);
|
|
||||||
runs.ScatterQuartets(d, 0, v0);
|
|
||||||
runs.ScatterQuartets(d, 1, v1);
|
|
||||||
runs.ScatterQuartets(d, 2, v2);
|
|
||||||
runs.ScatterQuartets(d, 3, v3);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreDeinterleavedQuartets(D d, Vec<D> v0, Vec<D> v1, Vec<D> v2,
|
|
||||||
Vec<D> v3, Vec<D> v4, Vec<D> v5, Vec<D> v6,
|
|
||||||
Vec<D> v7) {
|
|
||||||
Runs<D> runs(d, 8);
|
|
||||||
runs.ScatterQuartets(d, 0, v0);
|
|
||||||
runs.ScatterQuartets(d, 1, v1);
|
|
||||||
runs.ScatterQuartets(d, 2, v2);
|
|
||||||
runs.ScatterQuartets(d, 3, v3);
|
|
||||||
runs.ScatterQuartets(d, 4, v4);
|
|
||||||
runs.ScatterQuartets(d, 5, v5);
|
|
||||||
runs.ScatterQuartets(d, 6, v6);
|
|
||||||
runs.ScatterQuartets(d, 7, v7);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreDeinterleavedQuartets(D d, Vec<D> v0, Vec<D> v1, Vec<D> v2,
|
|
||||||
Vec<D> v3, Vec<D> v4, Vec<D> v5, Vec<D> v6,
|
|
||||||
Vec<D> v7, Vec<D> v8, Vec<D> v9, Vec<D> vA,
|
|
||||||
Vec<D> vB, Vec<D> vC, Vec<D> vD, Vec<D> vE,
|
|
||||||
Vec<D> vF) {
|
|
||||||
Runs<D> runs(d, 16);
|
|
||||||
runs.ScatterQuartets(d, 0x0, v0);
|
|
||||||
runs.ScatterQuartets(d, 0x1, v1);
|
|
||||||
runs.ScatterQuartets(d, 0x2, v2);
|
|
||||||
runs.ScatterQuartets(d, 0x3, v3);
|
|
||||||
runs.ScatterQuartets(d, 0x4, v4);
|
|
||||||
runs.ScatterQuartets(d, 0x5, v5);
|
|
||||||
runs.ScatterQuartets(d, 0x6, v6);
|
|
||||||
runs.ScatterQuartets(d, 0x7, v7);
|
|
||||||
runs.ScatterQuartets(d, 0x8, v8);
|
|
||||||
runs.ScatterQuartets(d, 0x9, v9);
|
|
||||||
runs.ScatterQuartets(d, 0xA, vA);
|
|
||||||
runs.ScatterQuartets(d, 0xB, vB);
|
|
||||||
runs.ScatterQuartets(d, 0xC, vC);
|
|
||||||
runs.ScatterQuartets(d, 0xD, vD);
|
|
||||||
runs.ScatterQuartets(d, 0xE, vE);
|
|
||||||
runs.ScatterQuartets(d, 0xF, vF);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreDeinterleavedQuartets(
|
|
||||||
D d, const Vec<D>& v00, const Vec<D>& v01, const Vec<D>& v02,
|
|
||||||
const Vec<D>& v03, const Vec<D>& v04, const Vec<D>& v05, const Vec<D>& v06,
|
|
||||||
const Vec<D>& v07, const Vec<D>& v08, const Vec<D>& v09, const Vec<D>& v0A,
|
|
||||||
const Vec<D>& v0B, const Vec<D>& v0C, const Vec<D>& v0D, const Vec<D>& v0E,
|
|
||||||
const Vec<D>& v0F, const Vec<D>& v10, const Vec<D>& v11, const Vec<D>& v12,
|
|
||||||
const Vec<D>& v13, const Vec<D>& v14, const Vec<D>& v15, const Vec<D>& v16,
|
|
||||||
const Vec<D>& v17, const Vec<D>& v18, const Vec<D>& v19, const Vec<D>& v1A,
|
|
||||||
const Vec<D>& v1B, const Vec<D>& v1C, const Vec<D>& v1D, const Vec<D>& v1E,
|
|
||||||
const Vec<D>& v1F) {
|
|
||||||
Runs<D> runs(d, 32);
|
|
||||||
runs.ScatterQuartets(d, 0x00, v00);
|
|
||||||
runs.ScatterQuartets(d, 0x01, v01);
|
|
||||||
runs.ScatterQuartets(d, 0x02, v02);
|
|
||||||
runs.ScatterQuartets(d, 0x03, v03);
|
|
||||||
runs.ScatterQuartets(d, 0x04, v04);
|
|
||||||
runs.ScatterQuartets(d, 0x05, v05);
|
|
||||||
runs.ScatterQuartets(d, 0x06, v06);
|
|
||||||
runs.ScatterQuartets(d, 0x07, v07);
|
|
||||||
runs.ScatterQuartets(d, 0x08, v08);
|
|
||||||
runs.ScatterQuartets(d, 0x09, v09);
|
|
||||||
runs.ScatterQuartets(d, 0x0A, v0A);
|
|
||||||
runs.ScatterQuartets(d, 0x0B, v0B);
|
|
||||||
runs.ScatterQuartets(d, 0x0C, v0C);
|
|
||||||
runs.ScatterQuartets(d, 0x0D, v0D);
|
|
||||||
runs.ScatterQuartets(d, 0x0E, v0E);
|
|
||||||
runs.ScatterQuartets(d, 0x0F, v0F);
|
|
||||||
runs.ScatterQuartets(d, 0x10, v10);
|
|
||||||
runs.ScatterQuartets(d, 0x11, v11);
|
|
||||||
runs.ScatterQuartets(d, 0x12, v12);
|
|
||||||
runs.ScatterQuartets(d, 0x13, v13);
|
|
||||||
runs.ScatterQuartets(d, 0x14, v14);
|
|
||||||
runs.ScatterQuartets(d, 0x15, v15);
|
|
||||||
runs.ScatterQuartets(d, 0x16, v16);
|
|
||||||
runs.ScatterQuartets(d, 0x17, v17);
|
|
||||||
runs.ScatterQuartets(d, 0x18, v18);
|
|
||||||
runs.ScatterQuartets(d, 0x19, v19);
|
|
||||||
runs.ScatterQuartets(d, 0x1A, v1A);
|
|
||||||
runs.ScatterQuartets(d, 0x1B, v1B);
|
|
||||||
runs.ScatterQuartets(d, 0x1C, v1C);
|
|
||||||
runs.ScatterQuartets(d, 0x1D, v1D);
|
|
||||||
runs.ScatterQuartets(d, 0x1E, v1E);
|
|
||||||
runs.ScatterQuartets(d, 0x1F, v1F);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreVectors(D d, Vec<D> v0, size_t run_length, bool alternating) {
|
|
||||||
Runs<D> runs(d, 1, run_length, alternating);
|
|
||||||
runs.StoreVector(d, 0, v0);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreVectors(D d, Vec<D> v0, Vec<D> v1) {
|
|
||||||
constexpr size_t kRegs = 2;
|
|
||||||
Runs<D> runs(d, kRegs, /*run_length=*/kRegs * Lanes(d), /*alternating=*/false);
|
|
||||||
runs.StoreVector(d, 0, v0);
|
|
||||||
runs.StoreVector(d, 1, v1);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreVectors(D d, Vec<D> v0, Vec<D> v1, Vec<D> v2, Vec<D> v3) {
|
|
||||||
constexpr size_t kRegs = 4;
|
|
||||||
Runs<D> runs(d, kRegs, /*run_length=*/kRegs * Lanes(d), /*alternating=*/false);
|
|
||||||
runs.StoreVector(d, 0, v0);
|
|
||||||
runs.StoreVector(d, 1, v1);
|
|
||||||
runs.StoreVector(d, 2, v2);
|
|
||||||
runs.StoreVector(d, 3, v3);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
Runs<D> StoreVectors(D d, Vec<D> v0, Vec<D> v1, Vec<D> v2, Vec<D> v3, Vec<D> v4,
|
|
||||||
Vec<D> v5, Vec<D> v6, Vec<D> v7) {
|
|
||||||
constexpr size_t kRegs = 8;
|
|
||||||
Runs<D> runs(d, kRegs, /*run_length=*/kRegs * Lanes(d), /*alternating=*/false);
|
|
||||||
runs.StoreVector(d, 0, v0);
|
|
||||||
runs.StoreVector(d, 1, v1);
|
|
||||||
runs.StoreVector(d, 2, v2);
|
|
||||||
runs.StoreVector(d, 3, v3);
|
|
||||||
runs.StoreVector(d, 4, v4);
|
|
||||||
runs.StoreVector(d, 5, v5);
|
|
||||||
runs.StoreVector(d, 6, v6);
|
|
||||||
runs.StoreVector(d, 7, v7);
|
|
||||||
return runs;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // HWY_SORT_VERIFY
|
|
||||||
} // namespace verify
|
|
||||||
|
|
||||||
namespace detail {
|
|
||||||
|
|
||||||
// ------------------------------ Vector-length agnostic (quartets)
|
|
||||||
|
|
||||||
// For each lane i: replaces a[i] with the first and b[i] with the second
|
|
||||||
// according to kOrder.
|
|
||||||
// Corresponds to a conditional swap, which is one "node" of a sorting network.
|
|
||||||
// Min/Max are cheaper than compare + blend at least for integers.
|
|
||||||
template <SortOrder kOrder, class V>
|
|
||||||
HWY_INLINE void SortLanesIn2Vectors(V& a, V& b) {
|
|
||||||
V temp = a;
|
|
||||||
a = (kOrder == SortOrder::kAscending) ? Min(a, b) : Max(a, b);
|
|
||||||
b = (kOrder == SortOrder::kAscending) ? Max(temp, b) : Min(temp, b);
|
|
||||||
}
|
|
||||||
|
|
||||||
// For each lane: sorts the four values in the that lane of the four vectors.
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void SortLanesIn4Vectors(D d, const TFromD<D>* in, V& v0, V& v1,
|
|
||||||
V& v2, V& v3) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
|
|
||||||
// Bitonic and odd-even sorters both have 5 nodes. This one is from
|
|
||||||
// http://users.telenet.be/bertdobbelaere/SorterHunter/sorting_networks.html
|
|
||||||
|
|
||||||
// layer 1
|
|
||||||
v0 = Load(d, in + 0 * N);
|
|
||||||
v2 = Load(d, in + 2 * N);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v2);
|
|
||||||
v1 = Load(d, in + 1 * N);
|
|
||||||
v3 = Load(d, in + 3 * N);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v1, v3);
|
|
||||||
|
|
||||||
// layer 2
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v2, v3);
|
|
||||||
|
|
||||||
// layer 3
|
|
||||||
SortLanesIn2Vectors<kOrder>(v1, v2);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Inputs are vectors with columns in sorted order (from SortLanesIn4Vectors).
|
|
||||||
// Transposes so that output vectors are sorted quartets (128-bit blocks),
|
|
||||||
// and a quartet in v0 comes before its counterpart in v1, etc.
|
|
||||||
template <class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void Transpose4x4(D d, V& v0, V& v1, V& v2, V& v3) {
|
|
||||||
const RepartitionToWide<decltype(d)> dw;
|
|
||||||
|
|
||||||
// Input: first number is reg, second is lane (0 is lowest)
|
|
||||||
// 03 02 01 00 |
|
|
||||||
// 13 12 11 10 | columns are sorted
|
|
||||||
// 23 22 21 20 | (in this order)
|
|
||||||
// 33 32 31 30 V
|
|
||||||
const V t0 = InterleaveLower(d, v0, v1); // 11 01 10 00
|
|
||||||
const V t1 = InterleaveLower(d, v2, v3); // 31 21 30 20
|
|
||||||
const V t2 = InterleaveUpper(d, v0, v1); // 13 03 12 02
|
|
||||||
const V t3 = InterleaveUpper(d, v2, v3); // 33 23 32 22
|
|
||||||
|
|
||||||
// 30 20 10 00
|
|
||||||
v0 = BitCast(d, InterleaveLower(BitCast(dw, t0), BitCast(dw, t1)));
|
|
||||||
// 31 21 11 01
|
|
||||||
v1 = BitCast(d, InterleaveUpper(BitCast(dw, t0), BitCast(dw, t1)));
|
|
||||||
// 32 22 12 02
|
|
||||||
v2 = BitCast(d, InterleaveLower(BitCast(dw, t2), BitCast(dw, t3)));
|
|
||||||
// 33 23 13 03 --> sorted in descending order (03=smallest in lane 0).
|
|
||||||
v3 = BitCast(d, InterleaveUpper(BitCast(dw, t2), BitCast(dw, t3)));
|
|
||||||
}
|
|
||||||
|
|
||||||
// 12 ops (including 4 swizzle)
|
|
||||||
// Precondition: v0 and v1 are already sorted according to kOrder.
|
|
||||||
// Postcondition: concatenate(v0, v1) is sorted and v0 is the lower half.
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void Merge2SortedQuartets(D d, V& v0, V& v1, int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input0 = verify::StoreDeinterleavedQuartets(d, v0);
|
|
||||||
const verify::Runs<D> input1 = verify::StoreDeinterleavedQuartets(d, v1);
|
|
||||||
input0.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
input1.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// See figure 5 from https://www.vldb.org/pvldb/vol8/p1274-inoue.pdf.
|
|
||||||
// This requires 8 min/max vs 6 for bitonic merge (see Figure 2 in
|
|
||||||
// http://www.vldb.org/pvldb/vol1/1454171.pdf), but is faster overall because
|
|
||||||
// it needs less shuffling, and does not need a bitonic input.
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
v0 = Shuffle0321(v0);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
v0 = Shuffle0321(v0);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
v0 = Shuffle0321(v0);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
v0 = Shuffle0321(v0);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
auto output = verify::StoreDeinterleavedQuartets(d, v0, v1);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------ Bitonic merge (quartets)
|
|
||||||
|
|
||||||
// For the last layer of bitonic merge. Conditionally swaps even-numbered lanes
|
|
||||||
// with their odd-numbered neighbor. Works for both quartets and vectors.
|
|
||||||
template <SortOrder kOrder, class D>
|
|
||||||
HWY_INLINE void SortAdjacentLanesQV(D d, Vec<D>& q_or_v) {
|
|
||||||
(void)d;
|
|
||||||
// Optimization for 32-bit integers: swap via Shuffle and 64-bit Min/Max.
|
|
||||||
// (not worthwhile on SSE4/AVX2 because they lack 64-bit Min/Max)
|
|
||||||
#if !HWY_ARCH_X86 || HWY_TARGET <= HWY_AVX3
|
|
||||||
if (sizeof(TFromD<D>) == 4 && !IsFloat<TFromD<D>>()) {
|
|
||||||
const RepartitionToWide<decltype(d)> dw;
|
|
||||||
const auto wide = BitCast(dw, q_or_v);
|
|
||||||
const auto swap = BitCast(dw, Shuffle2301(q_or_v));
|
|
||||||
if (kOrder == SortOrder::kAscending) {
|
|
||||||
q_or_v = BitCast(d, Max(wide, swap));
|
|
||||||
} else {
|
|
||||||
q_or_v = BitCast(d, Min(wide, swap));
|
|
||||||
}
|
|
||||||
} else
|
|
||||||
#endif
|
|
||||||
{
|
|
||||||
Vec<D> swapped = Shuffle2301(q_or_v);
|
|
||||||
SortLanesIn2Vectors<kOrder>(q_or_v, swapped);
|
|
||||||
q_or_v = OddEven(swapped, q_or_v);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Lane 0 with 2, 1 with 3 etc. Works for both quartets and vectors.
|
|
||||||
template <SortOrder kOrder, class D>
|
|
||||||
HWY_INLINE void SortDistance2LanesQV(D d, Vec<D>& q_or_v) {
|
|
||||||
const RepartitionToWide<decltype(d)> dw;
|
|
||||||
Vec<D> swapped = Shuffle1032(q_or_v);
|
|
||||||
SortLanesIn2Vectors<kOrder>(q_or_v, swapped);
|
|
||||||
q_or_v = BitCast(d, OddEven(BitCast(dw, swapped), BitCast(dw, q_or_v)));
|
|
||||||
}
|
|
||||||
|
|
||||||
// For all BitonicMerge*, and each block, the concatenation of those blocks from
|
|
||||||
// the first half and second half of the input vectors must be sorted in
|
|
||||||
// opposite orders.
|
|
||||||
|
|
||||||
// 14 ops (including 4 swizzle)
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void BitonicMerge2Quartets(D d, V& q0, V& q1, int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input = verify::StoreDeinterleavedQuartets(d, q0, q1);
|
|
||||||
if (caller == -1) input.CheckBitonic(__LINE__, __LINE__);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layer 1: lane stride 4 (2 ops)
|
|
||||||
SortLanesIn2Vectors<kOrder>(q0, q1);
|
|
||||||
|
|
||||||
// Layer 2: lane stride 2 (6 ops)
|
|
||||||
SortDistance2LanesQV<kOrder>(d, q0);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, q1);
|
|
||||||
|
|
||||||
// Layer 3: lane stride 1 (4 ops)
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, q0);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, q1);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> output = verify::StoreDeinterleavedQuartets(d, q0, q1);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// 32 ops, more efficient than three 4+4 merges (36 ops).
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void BitonicMerge4Quartets(D d, V& q0, V& q1, V& q2, V& q3,
|
|
||||||
int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input =
|
|
||||||
verify::StoreDeinterleavedQuartets(d, q0, q1, q2, q3);
|
|
||||||
if (caller == -1) input.CheckBitonic(__LINE__, __LINE__);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layer 1: lane stride 8
|
|
||||||
SortLanesIn2Vectors<kOrder>(q0, q2);
|
|
||||||
SortLanesIn2Vectors<kOrder>(q1, q3);
|
|
||||||
|
|
||||||
// Layers 2 to 4
|
|
||||||
// Inputs are not fully sorted, so cannot use Merge2SortedQuartets.
|
|
||||||
BitonicMerge2Quartets<kOrder>(d, q0, q1, __LINE__);
|
|
||||||
BitonicMerge2Quartets<kOrder>(d, q2, q3, __LINE__);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> output =
|
|
||||||
verify::StoreDeinterleavedQuartets(d, q0, q1, q2, q3);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// 72 ops.
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void BitonicMerge8Quartets(D d, V& q0, V& q1, V& q2, V& q3, V& q4,
|
|
||||||
V& q5, V& q6, V& q7, int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input =
|
|
||||||
verify::StoreDeinterleavedQuartets(d, q0, q1, q2, q3, q4, q5, q6, q7);
|
|
||||||
if (caller == -1) input.CheckBitonic(__LINE__, __LINE__);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layer 1: lane stride 16
|
|
||||||
SortLanesIn2Vectors<kOrder>(q0, q4);
|
|
||||||
SortLanesIn2Vectors<kOrder>(q1, q5);
|
|
||||||
SortLanesIn2Vectors<kOrder>(q2, q6);
|
|
||||||
SortLanesIn2Vectors<kOrder>(q3, q7);
|
|
||||||
|
|
||||||
// Layers 2 to 5
|
|
||||||
BitonicMerge4Quartets<kOrder>(d, q0, q1, q2, q3, __LINE__);
|
|
||||||
BitonicMerge4Quartets<kOrder>(d, q4, q5, q6, q7, __LINE__);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> output =
|
|
||||||
verify::StoreDeinterleavedQuartets(d, q0, q1, q2, q3, q4, q5, q6, q7);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------ Bitonic merge (vectors)
|
|
||||||
|
|
||||||
// Lane 0 with 4, 1 with 5 etc. Only used for vectors with at least 8 lanes.
|
|
||||||
#if HWY_TARGET <= HWY_AVX3
|
|
||||||
|
|
||||||
// TODO(janwas): move to op
|
|
||||||
template <typename T>
|
|
||||||
Vec512<T> Shuffle128_2020(Vec512<T> a, Vec512<T> b) {
|
|
||||||
return Vec512<T>{_mm512_shuffle_i32x4(a.raw, b.raw, _MM_SHUFFLE(2, 0, 2, 0))};
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
Vec512<T> Shuffle128_3131(Vec512<T> a, Vec512<T> b) {
|
|
||||||
return Vec512<T>{_mm512_shuffle_i32x4(a.raw, b.raw, _MM_SHUFFLE(3, 1, 3, 1))};
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
Vec512<T> Shuffle128_2301(Vec512<T> a, Vec512<T> b) {
|
|
||||||
return Vec512<T>{_mm512_shuffle_i32x4(a.raw, b.raw, _MM_SHUFFLE(2, 3, 0, 1))};
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
Vec512<T> OddEven128(Vec512<T> odd, Vec512<T> even) {
|
|
||||||
return Vec512<T>{_mm512_mask_blend_epi64(__mmask8{0x33u}, odd.raw, even.raw)};
|
|
||||||
}
|
|
||||||
|
|
||||||
template <SortOrder kOrder, class T>
|
|
||||||
HWY_INLINE void SortDistance4LanesV(Simd<T, 16> d, Vec<decltype(d)>& v) {
|
|
||||||
// In: FEDCBA98 76543210
|
|
||||||
// Swap 128-bit halves of each 256 bits => BA98FEDC 32107654
|
|
||||||
Vec512<T> swapped = Shuffle128_2301(v, v);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v, swapped);
|
|
||||||
v = OddEven128(swapped, v);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
template <SortOrder kOrder, typename T>
|
|
||||||
HWY_INLINE void SortDistance4LanesV(Simd<T, 8> d, Vec<decltype(d)>& v) {
|
|
||||||
Vec<decltype(d)> swapped = ConcatLowerUpper(d, v, v);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v, swapped);
|
|
||||||
v = ConcatUpperLower(swapped, v);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <SortOrder kOrder, typename T>
|
|
||||||
HWY_INLINE void SortDistance4LanesV(Simd<T, 4> /* tag */, ...) {}
|
|
||||||
|
|
||||||
// Only used for vectors with at least 16 lanes.
|
|
||||||
template <SortOrder kOrder, class D>
|
|
||||||
HWY_INLINE void SortDistance8LanesV(D d, Vec<D>& v) {
|
|
||||||
Vec<D> swapped = ConcatLowerUpper(d, v, v);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v, swapped);
|
|
||||||
v = ConcatUpperLower(swapped, v);
|
|
||||||
}
|
|
||||||
|
|
||||||
// 120 ops. Only used if vectors are at least 8 lanes.
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void BitonicMergeTo64(D d, V& v0, V& v1, V& v2, V& v3, V& v4, V& v5,
|
|
||||||
V& v6, V& v7, int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input =
|
|
||||||
verify::StoreVectors(d, v0, v1, v2, v3, v4, v5, v6, v7);
|
|
||||||
if (caller == -1) input.CheckBitonic(__LINE__, __LINE__);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layer 1: lane stride 32
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v4);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v1, v5);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v2, v6);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v3, v7);
|
|
||||||
|
|
||||||
// Layer 2: lane stride 16
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v2);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v1, v3);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v4, v6);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v5, v7);
|
|
||||||
|
|
||||||
// Layer 3: lane stride 8
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v2, v3);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v4, v5);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v6, v7);
|
|
||||||
|
|
||||||
// Layer 4: lane stride 4
|
|
||||||
SortDistance4LanesV<kOrder>(d, v0);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v1);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v2);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v3);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v4);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v5);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v6);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v7);
|
|
||||||
|
|
||||||
// Layer 5: lane stride 2
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v0);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v1);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v2);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v3);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v4);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v5);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v6);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v7);
|
|
||||||
|
|
||||||
// Layer 6: lane stride 1
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v0);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v1);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v2);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v3);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v4);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v5);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v6);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v7);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> output =
|
|
||||||
verify::StoreVectors(d, v0, v1, v2, v3, v4, v5, v6, v7);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// 60 ops. Only used if vectors are at least 16 lanes.
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void BitonicMergeTo64(D d, V& v0, V& v1, V& v2, V& v3, int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input = verify::StoreVectors(d, v0, v1, v2, v3);
|
|
||||||
if (caller == -1) input.CheckBitonic(__LINE__, __LINE__);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layer 1: lane stride 32
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v2);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v1, v3);
|
|
||||||
|
|
||||||
// Layer 2: lane stride 16
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v1);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v2, v3);
|
|
||||||
|
|
||||||
// Layer 3: lane stride 8
|
|
||||||
SortDistance8LanesV<kOrder>(d, v0);
|
|
||||||
SortDistance8LanesV<kOrder>(d, v1);
|
|
||||||
SortDistance8LanesV<kOrder>(d, v2);
|
|
||||||
SortDistance8LanesV<kOrder>(d, v3);
|
|
||||||
|
|
||||||
// Layer 4: lane stride 4
|
|
||||||
SortDistance4LanesV<kOrder>(d, v0);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v1);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v2);
|
|
||||||
SortDistance4LanesV<kOrder>(d, v3);
|
|
||||||
|
|
||||||
// Layer 5: lane stride 2
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v0);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v1);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v2);
|
|
||||||
SortDistance2LanesQV<kOrder>(d, v3);
|
|
||||||
|
|
||||||
// Layer 6: lane stride 1
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v0);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v1);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v2);
|
|
||||||
SortAdjacentLanesQV<kOrder>(d, v3);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> output = verify::StoreVectors(d, v0, v1, v2, v3);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// 128 ops. Only used if vectors are at least 16 lanes.
|
|
||||||
template <SortOrder kOrder, class D, class V = Vec<D>>
|
|
||||||
HWY_INLINE void BitonicMergeTo128(D d, V& v0, V& v1, V& v2, V& v3, V& v4, V& v5,
|
|
||||||
V& v6, V& v7, int caller) {
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> input =
|
|
||||||
verify::StoreVectors(d, v0, v1, v2, v3, v4, v5, v6, v7);
|
|
||||||
if (caller == -1) input.CheckBitonic(__LINE__, __LINE__);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Layer 1: lane stride 64
|
|
||||||
SortLanesIn2Vectors<kOrder>(v0, v4);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v1, v5);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v2, v6);
|
|
||||||
SortLanesIn2Vectors<kOrder>(v3, v7);
|
|
||||||
|
|
||||||
BitonicMergeTo64<kOrder>(d, v0, v1, v2, v3, __LINE__);
|
|
||||||
BitonicMergeTo64<kOrder>(d, v4, v5, v6, v7, __LINE__);
|
|
||||||
|
|
||||||
#if HWY_SORT_VERIFY
|
|
||||||
const verify::Runs<D> output =
|
|
||||||
verify::StoreVectors(d, v0, v1, v2, v3, v4, v5, v6, v7);
|
|
||||||
output.CheckSorted(kOrder, __LINE__, caller);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------ Vector-length dependent
|
|
||||||
|
|
||||||
// Only called when N=4 (single block, so quartets can just be stored).
|
|
||||||
template <SortOrder kOrder, class D, class V>
|
|
||||||
HWY_API size_t SingleQuartetPerVector(D d, V& q0, V& q1, V& q2, V& q3, V& q4,
|
|
||||||
V& q5, V& q6, V& q7, TFromD<D>* inout) {
|
|
||||||
Store(q0, d, inout + 0 * 4);
|
|
||||||
Store(q1, d, inout + 1 * 4);
|
|
||||||
Store(q2, d, inout + 2 * 4);
|
|
||||||
Store(q3, d, inout + 3 * 4);
|
|
||||||
Store(q4, d, inout + 4 * 4);
|
|
||||||
Store(q5, d, inout + 5 * 4);
|
|
||||||
Store(q6, d, inout + 6 * 4);
|
|
||||||
Store(q7, d, inout + 7 * 4);
|
|
||||||
return 8 * 4;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Only called when N=8.
|
|
||||||
template <SortOrder kOrder, class D, class V>
|
|
||||||
HWY_API size_t TwoQuartetsPerVector(D d, V& q0, V& q1, V& q2, V& q3, V& q4,
|
|
||||||
V& q5, V& q6, V& q7, TFromD<D>* inout) {
|
|
||||||
V v0 = ConcatLowerLower(d, q1, q0);
|
|
||||||
V v1 = ConcatLowerLower(d, q3, q2);
|
|
||||||
V v2 = ConcatLowerLower(d, q5, q4);
|
|
||||||
V v3 = ConcatLowerLower(d, q7, q6);
|
|
||||||
// TODO(janwas): merge into single table
|
|
||||||
V v4 = Reverse(d, ConcatUpperUpper(d, q7, q6));
|
|
||||||
V v5 = Reverse(d, ConcatUpperUpper(d, q5, q4));
|
|
||||||
V v6 = Reverse(d, ConcatUpperUpper(d, q3, q2));
|
|
||||||
V v7 = Reverse(d, ConcatUpperUpper(d, q1, q0));
|
|
||||||
detail::BitonicMergeTo64<kOrder>(d, v0, v1, v2, v3, v4, v5, v6, v7, -1);
|
|
||||||
|
|
||||||
Store(v0, d, inout + 0 * 8);
|
|
||||||
Store(v1, d, inout + 1 * 8);
|
|
||||||
Store(v2, d, inout + 2 * 8);
|
|
||||||
Store(v3, d, inout + 3 * 8);
|
|
||||||
Store(v4, d, inout + 4 * 8);
|
|
||||||
Store(v5, d, inout + 5 * 8);
|
|
||||||
Store(v6, d, inout + 6 * 8);
|
|
||||||
Store(v7, d, inout + 7 * 8);
|
|
||||||
return 8 * 8;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Only called when N=16.
|
|
||||||
template <SortOrder kOrder, typename T, class V>
|
|
||||||
HWY_API size_t FourQuartetsPerVector(Simd<T, 16> d, V& q0, V& q1, V& q2, V& q3,
|
|
||||||
V& q4, V& q5, V& q6, V& q7, T* inout) {
|
|
||||||
const V q11_01_10_00 = Shuffle128_2020(q0, q1);
|
|
||||||
const V q13_03_12_02 = Shuffle128_2020(q2, q3);
|
|
||||||
V v0 = Shuffle128_2020(q11_01_10_00, q13_03_12_02); // 3..0
|
|
||||||
|
|
||||||
const V q15_05_14_04 = Shuffle128_2020(q4, q5);
|
|
||||||
const V q17_07_16_06 = Shuffle128_2020(q6, q7);
|
|
||||||
V v1 = Shuffle128_2020(q15_05_14_04, q17_07_16_06); // 7..4
|
|
||||||
|
|
||||||
const V q19_09_18_08 = Shuffle128_3131(q0, q1);
|
|
||||||
const V q1b_0b_1a_0a = Shuffle128_3131(q2, q3);
|
|
||||||
V v3 = Reverse(d, Shuffle128_2020(q19_09_18_08, q1b_0b_1a_0a)); // b..8
|
|
||||||
|
|
||||||
const V q1d_0d_1c_0c = Shuffle128_3131(q4, q5);
|
|
||||||
const V q1f_0f_1e_0e = Shuffle128_3131(q6, q7);
|
|
||||||
V v2 = Reverse(d, Shuffle128_2020(q1d_0d_1c_0c, q1f_0f_1e_0e)); // f..c
|
|
||||||
|
|
||||||
detail::BitonicMergeTo64<kOrder>(d, v0, v1, v2, v3, -1);
|
|
||||||
|
|
||||||
// TODO(janwas): merge into single table
|
|
||||||
V v4 = Shuffle128_3131(q11_01_10_00, q13_03_12_02); // 13..10
|
|
||||||
V v5 = Shuffle128_3131(q15_05_14_04, q17_07_16_06); // 17..14
|
|
||||||
V v7 = Reverse(d, Shuffle128_3131(q19_09_18_08, q1b_0b_1a_0a)); // 1b..18
|
|
||||||
V v6 = Reverse(d, Shuffle128_3131(q1d_0d_1c_0c, q1f_0f_1e_0e)); // 1f..1c
|
|
||||||
|
|
||||||
detail::BitonicMergeTo64<Reverse(kOrder)>(d, v4, v5, v6, v7, -1);
|
|
||||||
|
|
||||||
detail::BitonicMergeTo128<kOrder>(d, v0, v1, v2, v3, v4, v5, v6, v7, -1);
|
|
||||||
|
|
||||||
Store(v0, d, inout + 0 * 16);
|
|
||||||
Store(v1, d, inout + 1 * 16);
|
|
||||||
Store(v2, d, inout + 2 * 16);
|
|
||||||
Store(v3, d, inout + 3 * 16);
|
|
||||||
Store(v4, d, inout + 4 * 16);
|
|
||||||
Store(v5, d, inout + 5 * 16);
|
|
||||||
Store(v6, d, inout + 6 * 16);
|
|
||||||
Store(v7, d, inout + 7 * 16);
|
|
||||||
return 8 * 16;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Avoid needing #if at the call sites.
|
|
||||||
template <SortOrder kOrder, typename T>
|
|
||||||
HWY_API size_t TwoQuartetsPerVector(Simd<T, 4> /* tag */, ...) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <SortOrder kOrder, typename T>
|
|
||||||
HWY_API size_t FourQuartetsPerVector(Simd<T, 4> /* tag */, ...) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
template <SortOrder kOrder, typename T>
|
|
||||||
HWY_API size_t FourQuartetsPerVector(Simd<T, 8> /* tag */, ...) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace detail
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
HWY_API size_t SortBatchSize(D d) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
if (N == 4) return 32;
|
|
||||||
if (N == 8) return 64;
|
|
||||||
if (N == 16) return 128;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <SortOrder kOrder, class D>
|
|
||||||
HWY_API size_t SortBatch(D d, TFromD<D>* inout) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
|
|
||||||
Vec<D> q0, q1, q2, q3;
|
|
||||||
detail::SortLanesIn4Vectors<kOrder>(d, inout, q0, q1, q2, q3);
|
|
||||||
detail::Transpose4x4(d, q0, q1, q2, q3);
|
|
||||||
detail::Merge2SortedQuartets<kOrder>(d, q0, q1, -1);
|
|
||||||
detail::Merge2SortedQuartets<kOrder>(d, q2, q3, -1);
|
|
||||||
|
|
||||||
// Bitonic merges require one input to be in reverse order.
|
|
||||||
constexpr SortOrder kReverse = Reverse(kOrder);
|
|
||||||
|
|
||||||
Vec<D> q4, q5, q6, q7;
|
|
||||||
detail::SortLanesIn4Vectors<kReverse>(d, inout + 4 * N, q4, q5, q6, q7);
|
|
||||||
detail::Transpose4x4(d, q4, q5, q6, q7);
|
|
||||||
detail::Merge2SortedQuartets<kReverse>(d, q4, q5, -1);
|
|
||||||
detail::Merge2SortedQuartets<kReverse>(d, q6, q7, -1);
|
|
||||||
|
|
||||||
detail::BitonicMerge4Quartets<kOrder>(d, q0, q1, q4, q5, -1);
|
|
||||||
detail::BitonicMerge4Quartets<kReverse>(d, q2, q3, q6, q7, -1);
|
|
||||||
|
|
||||||
detail::BitonicMerge8Quartets<kOrder>(d, q0, q1, q4, q5, q2, q3, q6, q7,
|
|
||||||
__LINE__);
|
|
||||||
|
|
||||||
if (N == 4) {
|
|
||||||
return detail::SingleQuartetPerVector<kOrder>(d, q0, q1, q4, q5, q2, q3, q6,
|
|
||||||
q7, inout);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (N == 8) {
|
|
||||||
return detail::TwoQuartetsPerVector<kOrder>(d, q0, q1, q4, q5, q2, q3, q6,
|
|
||||||
q7, inout);
|
|
||||||
}
|
|
||||||
|
|
||||||
return detail::FourQuartetsPerVector<kOrder>(d, q0, q1, q4, q5, q2, q3, q6,
|
|
||||||
q7, inout);
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
// Avoids unused attribute warning
|
|
||||||
template <SortOrder kOrder, class D>
|
|
||||||
HWY_API size_t SortBatch(D /* tag */, TFromD<D>* /* inout */) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // HWY_TARGET != HWY_SCALAR && HWY_ARCH_X86
|
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
|
||||||
} // namespace HWY_NAMESPACE
|
|
||||||
} // namespace hwy
|
|
||||||
HWY_AFTER_NAMESPACE();
|
|
||||||
|
|
||||||
#endif // HIGHWAY_HWY_CONTRIB_SORT_SORT_INL_H_
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,97 +13,76 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
// clang-format off
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/sort_test.cc"
|
|
||||||
#include "hwy/foreach_target.h"
|
|
||||||
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
|
||||||
// After foreach_target
|
|
||||||
#include "hwy/contrib/sort/algo-inl.h"
|
|
||||||
#include "hwy/contrib/sort/result-inl.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort-inl.h" // BaseCase
|
|
||||||
#include "hwy/tests/test_util-inl.h"
|
|
||||||
// clang-format on
|
|
||||||
|
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <string.h> // memcpy
|
#include <string.h> // memcpy
|
||||||
|
|
||||||
#include <algorithm> // std::max
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#undef VQSORT_TEST_IMPL
|
// clang-format off
|
||||||
#if (HWY_TARGET == HWY_SCALAR) || (defined(_MSC_VER) && !HWY_IS_DEBUG_BUILD)
|
#undef HWY_TARGET_INCLUDE
|
||||||
// Scalar does not implement these, and MSVC non-debug builds time out.
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/sort_test.cc"
|
||||||
#define VQSORT_TEST_IMPL 0
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
#else
|
|
||||||
#define VQSORT_TEST_IMPL 1
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#undef VQSORT_TEST_SORT
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
// MSVC non-debug builds time out.
|
// After foreach_target
|
||||||
#if defined(_MSC_VER) && !HWY_IS_DEBUG_BUILD
|
#include "hwy/contrib/sort/algo-inl.h"
|
||||||
#define VQSORT_TEST_SORT 0
|
#include "hwy/contrib/sort/traits128-inl.h"
|
||||||
#else
|
#include "hwy/contrib/sort/result-inl.h"
|
||||||
#define VQSORT_TEST_SORT 1
|
#include "hwy/contrib/sort/vqsort-inl.h" // BaseCase
|
||||||
#endif
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
// clang-format on
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
#if VQSORT_TEST_IMPL || VQSORT_TEST_SORT
|
|
||||||
using detail::LaneTraits;
|
|
||||||
using detail::OrderAscending;
|
using detail::OrderAscending;
|
||||||
using detail::OrderAscending128;
|
|
||||||
using detail::OrderDescending;
|
using detail::OrderDescending;
|
||||||
using detail::OrderDescending128;
|
|
||||||
using detail::SharedTraits;
|
using detail::SharedTraits;
|
||||||
|
using detail::TraitsLane;
|
||||||
|
#if VQSORT_ENABLED || HWY_IDE
|
||||||
|
using detail::OrderAscending128;
|
||||||
|
using detail::OrderAscendingKV128;
|
||||||
|
using detail::OrderDescending128;
|
||||||
|
using detail::OrderDescendingKV128;
|
||||||
using detail::Traits128;
|
using detail::Traits128;
|
||||||
#endif
|
|
||||||
|
|
||||||
#if !VQSORT_TEST_IMPL
|
|
||||||
static void TestAllMedian() {}
|
|
||||||
static void TestAllBaseCase() {}
|
|
||||||
static void TestAllPartition() {}
|
|
||||||
static void TestAllGenerator() {}
|
|
||||||
#else
|
|
||||||
|
|
||||||
template <class Traits>
|
template <class Traits>
|
||||||
static HWY_NOINLINE void TestMedian3() {
|
static HWY_NOINLINE void TestMedian3() {
|
||||||
using T = uint64_t;
|
using LaneType = typename Traits::LaneType;
|
||||||
using D = CappedTag<T, 1>;
|
using D = CappedTag<LaneType, 1>;
|
||||||
SharedTraits<Traits> st;
|
SharedTraits<Traits> st;
|
||||||
const D d;
|
const D d;
|
||||||
using V = Vec<D>;
|
using V = Vec<D>;
|
||||||
for (uint32_t bits = 0; bits < 8; ++bits) {
|
for (uint32_t bits = 0; bits < 8; ++bits) {
|
||||||
const V v0 = Set(d, T{(bits & (1u << 0)) ? 1u : 0u});
|
const V v0 = Set(d, LaneType{(bits & (1u << 0)) ? 1u : 0u});
|
||||||
const V v1 = Set(d, T{(bits & (1u << 1)) ? 1u : 0u});
|
const V v1 = Set(d, LaneType{(bits & (1u << 1)) ? 1u : 0u});
|
||||||
const V v2 = Set(d, T{(bits & (1u << 2)) ? 1u : 0u});
|
const V v2 = Set(d, LaneType{(bits & (1u << 2)) ? 1u : 0u});
|
||||||
const T m = GetLane(detail::MedianOf3(st, v0, v1, v2));
|
const LaneType m = GetLane(detail::MedianOf3(st, v0, v1, v2));
|
||||||
// If at least half(rounded up) of bits are 1, so is the median.
|
// If at least half(rounded up) of bits are 1, so is the median.
|
||||||
const size_t count = PopCount(bits);
|
const size_t count = PopCount(bits);
|
||||||
HWY_ASSERT_EQ((count >= 2) ? static_cast<T>(1) : 0, m);
|
HWY_ASSERT_EQ((count >= 2) ? static_cast<LaneType>(1) : 0, m);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllMedian() {
|
HWY_NOINLINE void TestAllMedian() {
|
||||||
TestMedian3<LaneTraits<OrderAscending> >();
|
TestMedian3<TraitsLane<OrderAscending<uint64_t> > >();
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
static HWY_NOINLINE void TestBaseCaseAscDesc() {
|
static HWY_NOINLINE void TestBaseCaseAscDesc() {
|
||||||
|
using LaneType = typename Traits::LaneType;
|
||||||
SharedTraits<Traits> st;
|
SharedTraits<Traits> st;
|
||||||
const SortTag<T> d;
|
const SortTag<LaneType> d;
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
const size_t base_case_num = SortConstants::BaseCaseNum(N);
|
const size_t base_case_num = SortConstants::BaseCaseNum(N);
|
||||||
const size_t N1 = st.LanesPerKey();
|
const size_t N1 = st.LanesPerKey();
|
||||||
|
|
||||||
constexpr int kDebug = 0;
|
constexpr int kDebug = 0;
|
||||||
auto aligned_keys = hwy::AllocateAligned<T>(N + base_case_num + N);
|
auto aligned_lanes = hwy::AllocateAligned<LaneType>(N + base_case_num + N);
|
||||||
auto buf = hwy::AllocateAligned<T>(base_case_num + 2 * N);
|
auto buf = hwy::AllocateAligned<LaneType>(base_case_num + 2 * N);
|
||||||
|
|
||||||
std::vector<size_t> lengths;
|
std::vector<size_t> lengths;
|
||||||
lengths.push_back(HWY_MAX(1, N1));
|
lengths.push_back(HWY_MAX(1, N1));
|
||||||
|
@ -123,43 +103,45 @@ static HWY_NOINLINE void TestBaseCaseAscDesc() {
|
||||||
for (bool asc : {false, true}) {
|
for (bool asc : {false, true}) {
|
||||||
for (size_t len : lengths) {
|
for (size_t len : lengths) {
|
||||||
for (size_t misalign : misalignments) {
|
for (size_t misalign : misalignments) {
|
||||||
T* HWY_RESTRICT keys = aligned_keys.get() + misalign;
|
LaneType* HWY_RESTRICT lanes = aligned_lanes.get() + misalign;
|
||||||
if (kDebug) {
|
if (kDebug) {
|
||||||
printf("============%s asc %d N1 %d len %d misalign %d\n",
|
printf("============%s asc %d N1 %d len %d misalign %d\n",
|
||||||
hwy::TypeName(T(), 1).c_str(), asc, static_cast<int>(N1),
|
st.KeyString().c_str(), asc, static_cast<int>(N1),
|
||||||
static_cast<int>(len), static_cast<int>(misalign));
|
static_cast<int>(len), static_cast<int>(misalign));
|
||||||
}
|
}
|
||||||
|
|
||||||
for (size_t i = 0; i < misalign; ++i) {
|
for (size_t i = 0; i < misalign; ++i) {
|
||||||
aligned_keys[i] = hwy::LowestValue<T>();
|
aligned_lanes[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
InputStats<T> input_stats;
|
InputStats<LaneType> input_stats;
|
||||||
for (size_t i = 0; i < len; ++i) {
|
for (size_t i = 0; i < len; ++i) {
|
||||||
keys[i] =
|
lanes[i] = asc ? static_cast<LaneType>(LaneType(i) + 1)
|
||||||
asc ? static_cast<T>(T(i) + 1) : static_cast<T>(T(len) - T(i));
|
: static_cast<LaneType>(LaneType(len) - LaneType(i));
|
||||||
input_stats.Notify(keys[i]);
|
input_stats.Notify(lanes[i]);
|
||||||
if (kDebug >= 2) printf("%3zu: %f\n", i, double(keys[i]));
|
if (kDebug >= 2) {
|
||||||
|
printf("%3zu: %f\n", i, static_cast<double>(lanes[i]));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
for (size_t i = len; i < base_case_num + N; ++i) {
|
for (size_t i = len; i < base_case_num + N; ++i) {
|
||||||
keys[i] = hwy::LowestValue<T>();
|
lanes[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
|
|
||||||
detail::BaseCase(d, st, keys, len, buf.get());
|
detail::BaseCase(d, st, lanes, lanes + len, len, buf.get());
|
||||||
|
|
||||||
if (kDebug >= 2) {
|
if (kDebug >= 2) {
|
||||||
printf("out>>>>>>\n");
|
printf("out>>>>>>\n");
|
||||||
for (size_t i = 0; i < len; ++i) {
|
for (size_t i = 0; i < len; ++i) {
|
||||||
printf("%3zu: %f\n", i, double(keys[i]));
|
printf("%3zu: %f\n", i, static_cast<double>(lanes[i]));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_ASSERT(VerifySort(st, input_stats, keys, len, "BaseAscDesc"));
|
HWY_ASSERT(VerifySort(st, input_stats, lanes, len, "BaseAscDesc"));
|
||||||
for (size_t i = 0; i < misalign; ++i) {
|
for (size_t i = 0; i < misalign; ++i) {
|
||||||
if (aligned_keys[i] != hwy::LowestValue<T>())
|
if (aligned_lanes[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun misalign at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun misalign at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
for (size_t i = len; i < base_case_num + N; ++i) {
|
for (size_t i = len; i < base_case_num + N; ++i) {
|
||||||
if (keys[i] != hwy::LowestValue<T>())
|
if (lanes[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
} // misalign
|
} // misalign
|
||||||
|
@ -167,17 +149,18 @@ static HWY_NOINLINE void TestBaseCaseAscDesc() {
|
||||||
} // asc
|
} // asc
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
static HWY_NOINLINE void TestBaseCase01() {
|
static HWY_NOINLINE void TestBaseCase01() {
|
||||||
|
using LaneType = typename Traits::LaneType;
|
||||||
SharedTraits<Traits> st;
|
SharedTraits<Traits> st;
|
||||||
const SortTag<T> d;
|
const SortTag<LaneType> d;
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
const size_t base_case_num = SortConstants::BaseCaseNum(N);
|
const size_t base_case_num = SortConstants::BaseCaseNum(N);
|
||||||
const size_t N1 = st.LanesPerKey();
|
const size_t N1 = st.LanesPerKey();
|
||||||
|
|
||||||
constexpr int kDebug = 0;
|
constexpr int kDebug = 0;
|
||||||
auto keys = hwy::AllocateAligned<T>(base_case_num + N);
|
auto lanes = hwy::AllocateAligned<LaneType>(base_case_num + N);
|
||||||
auto buf = hwy::AllocateAligned<T>(base_case_num + 2 * N);
|
auto buf = hwy::AllocateAligned<LaneType>(base_case_num + 2 * N);
|
||||||
|
|
||||||
std::vector<size_t> lengths;
|
std::vector<size_t> lengths;
|
||||||
lengths.push_back(HWY_MAX(1, N1));
|
lengths.push_back(HWY_MAX(1, N1));
|
||||||
|
@ -189,65 +172,69 @@ static HWY_NOINLINE void TestBaseCase01() {
|
||||||
|
|
||||||
for (size_t len : lengths) {
|
for (size_t len : lengths) {
|
||||||
if (kDebug) {
|
if (kDebug) {
|
||||||
printf("============%s 01 N1 %d len %d\n", hwy::TypeName(T(), 1).c_str(),
|
printf("============%s 01 N1 %d len %d\n", st.KeyString().c_str(),
|
||||||
static_cast<int>(N1), static_cast<int>(len));
|
static_cast<int>(N1), static_cast<int>(len));
|
||||||
}
|
}
|
||||||
const uint64_t kMaxBits = AdjustedLog2Reps(HWY_MIN(len, size_t{14}));
|
const uint64_t kMaxBits = AdjustedLog2Reps(HWY_MIN(len, size_t{14}));
|
||||||
for (uint64_t bits = 0; bits < ((1ull << kMaxBits) - 1); ++bits) {
|
for (uint64_t bits = 0; bits < ((1ull << kMaxBits) - 1); ++bits) {
|
||||||
InputStats<T> input_stats;
|
InputStats<LaneType> input_stats;
|
||||||
for (size_t i = 0; i < len; ++i) {
|
for (size_t i = 0; i < len; ++i) {
|
||||||
keys[i] = (i < 64 && (bits & (1ull << i))) ? 1 : 0;
|
lanes[i] = (i < 64 && (bits & (1ull << i))) ? 1 : 0;
|
||||||
input_stats.Notify(keys[i]);
|
input_stats.Notify(lanes[i]);
|
||||||
if (kDebug >= 2) printf("%3zu: %f\n", i, double(keys[i]));
|
if (kDebug >= 2) {
|
||||||
|
printf("%3zu: %f\n", i, static_cast<double>(lanes[i]));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
for (size_t i = len; i < base_case_num + N; ++i) {
|
for (size_t i = len; i < base_case_num + N; ++i) {
|
||||||
keys[i] = hwy::LowestValue<T>();
|
lanes[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
|
|
||||||
detail::BaseCase(d, st, keys.get(), len, buf.get());
|
detail::BaseCase(d, st, lanes.get(), lanes.get() + len, len, buf.get());
|
||||||
|
|
||||||
if (kDebug >= 2) {
|
if (kDebug >= 2) {
|
||||||
printf("out>>>>>>\n");
|
printf("out>>>>>>\n");
|
||||||
for (size_t i = 0; i < len; ++i) {
|
for (size_t i = 0; i < len; ++i) {
|
||||||
printf("%3zu: %f\n", i, double(keys[i]));
|
printf("%3zu: %f\n", i, static_cast<double>(lanes[i]));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_ASSERT(VerifySort(st, input_stats, keys.get(), len, "Base01"));
|
HWY_ASSERT(VerifySort(st, input_stats, lanes.get(), len, "Base01"));
|
||||||
for (size_t i = len; i < base_case_num + N; ++i) {
|
for (size_t i = len; i < base_case_num + N; ++i) {
|
||||||
if (keys[i] != hwy::LowestValue<T>())
|
if (lanes[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
} // bits
|
} // bits
|
||||||
} // len
|
} // len
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
static HWY_NOINLINE void TestBaseCase() {
|
static HWY_NOINLINE void TestBaseCase() {
|
||||||
TestBaseCaseAscDesc<Traits, T>();
|
TestBaseCaseAscDesc<Traits>();
|
||||||
TestBaseCase01<Traits, T>();
|
TestBaseCase01<Traits>();
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllBaseCase() {
|
HWY_NOINLINE void TestAllBaseCase() {
|
||||||
// Workaround for stack overflow on MSVC debug.
|
// Workaround for stack overflow on MSVC debug.
|
||||||
#if defined(_MSC_VER) && HWY_IS_DEBUG_BUILD && (HWY_TARGET == HWY_AVX3)
|
#if defined(_MSC_VER)
|
||||||
return;
|
return;
|
||||||
#endif
|
#endif
|
||||||
|
// Only enable EMU128 on x86 - it's slow on emulators.
|
||||||
|
if (!HWY_ARCH_X86 && (HWY_TARGET == HWY_EMU128)) return;
|
||||||
|
|
||||||
TestBaseCase<LaneTraits<OrderAscending>, int32_t>();
|
TestBaseCase<TraitsLane<OrderAscending<int32_t> > >();
|
||||||
TestBaseCase<LaneTraits<OrderDescending>, int64_t>();
|
TestBaseCase<TraitsLane<OrderDescending<int64_t> > >();
|
||||||
TestBaseCase<Traits128<OrderAscending128>, uint64_t>();
|
TestBaseCase<Traits128<OrderAscending128> >();
|
||||||
TestBaseCase<Traits128<OrderDescending128>, uint64_t>();
|
TestBaseCase<Traits128<OrderDescending128> >();
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
static HWY_NOINLINE void VerifyPartition(Traits st, T* HWY_RESTRICT keys,
|
static HWY_NOINLINE void VerifyPartition(
|
||||||
size_t left, size_t border,
|
Traits st, typename Traits::LaneType* HWY_RESTRICT lanes, size_t left,
|
||||||
size_t right, const size_t N1,
|
size_t border, size_t right, const size_t N1,
|
||||||
const T* pivot) {
|
const typename Traits::LaneType* pivot) {
|
||||||
/* for (size_t i = left; i < right; ++i) {
|
/* for (size_t i = left; i < right; ++i) {
|
||||||
if (i == border) printf("--\n");
|
if (i == border) printf("--\n");
|
||||||
printf("%4zu: %3d\n", i, keys[i]);
|
printf("%4zu: %3d\n", i, lanes[i]);
|
||||||
}*/
|
}*/
|
||||||
|
|
||||||
HWY_ASSERT(left % N1 == 0);
|
HWY_ASSERT(left % N1 == 0);
|
||||||
|
@ -255,30 +242,33 @@ static HWY_NOINLINE void VerifyPartition(Traits st, T* HWY_RESTRICT keys,
|
||||||
HWY_ASSERT(right % N1 == 0);
|
HWY_ASSERT(right % N1 == 0);
|
||||||
const bool asc = typename Traits::Order().IsAscending();
|
const bool asc = typename Traits::Order().IsAscending();
|
||||||
for (size_t i = left; i < border; i += N1) {
|
for (size_t i = left; i < border; i += N1) {
|
||||||
if (st.Compare1(pivot, keys + i)) {
|
if (st.Compare1(pivot, lanes + i)) {
|
||||||
HWY_ABORT(
|
HWY_ABORT(
|
||||||
"%s: asc %d left[%d] piv %.0f %.0f compares before %.0f %.0f "
|
"%s: asc %d left[%d] piv %.0f %.0f compares before %.0f %.0f "
|
||||||
"border %d",
|
"border %d",
|
||||||
hwy::TypeName(T(), 1).c_str(), asc, static_cast<int>(i),
|
st.KeyString().c_str(), asc, static_cast<int>(i),
|
||||||
double(pivot[1]), double(pivot[0]), double(keys[i + 1]),
|
static_cast<double>(pivot[1]), static_cast<double>(pivot[0]),
|
||||||
double(keys[i + 0]), static_cast<int>(border));
|
static_cast<double>(lanes[i + 1]), static_cast<double>(lanes[i + 0]),
|
||||||
|
static_cast<int>(border));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for (size_t i = border; i < right; i += N1) {
|
for (size_t i = border; i < right; i += N1) {
|
||||||
if (!st.Compare1(pivot, keys + i)) {
|
if (!st.Compare1(pivot, lanes + i)) {
|
||||||
HWY_ABORT(
|
HWY_ABORT(
|
||||||
"%s: asc %d right[%d] piv %.0f %.0f compares after %.0f %.0f "
|
"%s: asc %d right[%d] piv %.0f %.0f compares after %.0f %.0f "
|
||||||
"border %d",
|
"border %d",
|
||||||
hwy::TypeName(T(), 1).c_str(), asc, static_cast<int>(i),
|
st.KeyString().c_str(), asc, static_cast<int>(i),
|
||||||
double(pivot[1]), double(pivot[0]), double(keys[i + 1]),
|
static_cast<double>(pivot[1]), static_cast<double>(pivot[0]),
|
||||||
double(keys[i]), static_cast<int>(border));
|
static_cast<double>(lanes[i + 1]), static_cast<double>(lanes[i]),
|
||||||
|
static_cast<int>(border));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
static HWY_NOINLINE void TestPartition() {
|
static HWY_NOINLINE void TestPartition() {
|
||||||
const SortTag<T> d;
|
using LaneType = typename Traits::LaneType;
|
||||||
|
const SortTag<LaneType> d;
|
||||||
SharedTraits<Traits> st;
|
SharedTraits<Traits> st;
|
||||||
const bool asc = typename Traits::Order().IsAscending();
|
const bool asc = typename Traits::Order().IsAscending();
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
|
@ -286,8 +276,8 @@ static HWY_NOINLINE void TestPartition() {
|
||||||
const size_t base_case_num = SortConstants::BaseCaseNum(N);
|
const size_t base_case_num = SortConstants::BaseCaseNum(N);
|
||||||
// left + len + align
|
// left + len + align
|
||||||
const size_t total = 32 + (base_case_num + 4 * HWY_MAX(N, 4)) + 2 * N;
|
const size_t total = 32 + (base_case_num + 4 * HWY_MAX(N, 4)) + 2 * N;
|
||||||
auto aligned_keys = hwy::AllocateAligned<T>(total);
|
auto aligned_lanes = hwy::AllocateAligned<LaneType>(total);
|
||||||
auto buf = hwy::AllocateAligned<T>(SortConstants::PartitionBufNum(N));
|
auto buf = hwy::AllocateAligned<LaneType>(SortConstants::PartitionBufNum(N));
|
||||||
|
|
||||||
const size_t N1 = st.LanesPerKey();
|
const size_t N1 = st.LanesPerKey();
|
||||||
for (bool in_asc : {false, true}) {
|
for (bool in_asc : {false, true}) {
|
||||||
|
@ -296,61 +286,66 @@ static HWY_NOINLINE void TestPartition() {
|
||||||
for (size_t ofs : {N, N + 1, N + 2, N + 3, 2 * N, 2 * N + 1, 2 * N + 2,
|
for (size_t ofs : {N, N + 1, N + 2, N + 3, 2 * N, 2 * N + 1, 2 * N + 2,
|
||||||
2 * N + 3, 3 * N - 1, 4 * N - 3, 4 * N - 2}) {
|
2 * N + 3, 3 * N - 1, 4 * N - 3, 4 * N - 2}) {
|
||||||
const size_t len = (base_case_num + ofs) & ~(N1 - 1);
|
const size_t len = (base_case_num + ofs) & ~(N1 - 1);
|
||||||
for (T pivot1 :
|
for (LaneType pivot1 :
|
||||||
{T(0), T(len / 3), T(len / 2), T(2 * len / 3), T(len)}) {
|
{LaneType(0), LaneType(len / 3), LaneType(len / 2),
|
||||||
const T pivot2[2] = {pivot1, 0};
|
LaneType(2 * len / 3), LaneType(len)}) {
|
||||||
|
const LaneType pivot2[2] = {pivot1, 0};
|
||||||
const auto pivot = st.SetKey(d, pivot2);
|
const auto pivot = st.SetKey(d, pivot2);
|
||||||
for (size_t misalign = 0; misalign < N;
|
for (size_t misalign = 0; misalign < N;
|
||||||
misalign += st.LanesPerKey()) {
|
misalign += st.LanesPerKey()) {
|
||||||
T* HWY_RESTRICT keys = aligned_keys.get() + misalign;
|
LaneType* HWY_RESTRICT lanes = aligned_lanes.get() + misalign;
|
||||||
const size_t right = left + len;
|
const size_t right = left + len;
|
||||||
if (kDebug) {
|
if (kDebug) {
|
||||||
printf(
|
printf(
|
||||||
"=========%s asc %d left %d len %d right %d piv %.0f %.0f\n",
|
"=========%s asc %d left %d len %d right %d piv %.0f %.0f\n",
|
||||||
hwy::TypeName(T(), 1).c_str(), asc, static_cast<int>(left),
|
st.KeyString().c_str(), asc, static_cast<int>(left),
|
||||||
static_cast<int>(len), static_cast<int>(right),
|
static_cast<int>(len), static_cast<int>(right),
|
||||||
double(pivot2[1]), double(pivot2[0]));
|
static_cast<double>(pivot2[1]),
|
||||||
|
static_cast<double>(pivot2[0]));
|
||||||
}
|
}
|
||||||
|
|
||||||
for (size_t i = 0; i < misalign; ++i) {
|
for (size_t i = 0; i < misalign; ++i) {
|
||||||
aligned_keys[i] = hwy::LowestValue<T>();
|
aligned_lanes[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
for (size_t i = 0; i < left; ++i) {
|
for (size_t i = 0; i < left; ++i) {
|
||||||
keys[i] = hwy::LowestValue<T>();
|
lanes[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
for (size_t i = left; i < right; ++i) {
|
for (size_t i = left; i < right; ++i) {
|
||||||
keys[i] = static_cast<T>(in_asc ? T(i + 1) - static_cast<T>(left)
|
lanes[i] = static_cast<LaneType>(
|
||||||
: static_cast<T>(right) - T(i));
|
in_asc ? LaneType(i + 1) - static_cast<LaneType>(left)
|
||||||
if (kDebug >= 2) printf("%3zu: %f\n", i, double(keys[i]));
|
: static_cast<LaneType>(right) - LaneType(i));
|
||||||
|
if (kDebug >= 2) {
|
||||||
|
printf("%3zu: %f\n", i, static_cast<double>(lanes[i]));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
for (size_t i = right; i < total - misalign; ++i) {
|
for (size_t i = right; i < total - misalign; ++i) {
|
||||||
keys[i] = hwy::LowestValue<T>();
|
lanes[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t border =
|
size_t border =
|
||||||
detail::Partition(d, st, keys, left, right, pivot, buf.get());
|
detail::Partition(d, st, lanes, left, right, pivot, buf.get());
|
||||||
|
|
||||||
if (kDebug >= 2) {
|
if (kDebug >= 2) {
|
||||||
printf("out>>>>>>\n");
|
printf("out>>>>>>\n");
|
||||||
for (size_t i = left; i < right; ++i) {
|
for (size_t i = left; i < right; ++i) {
|
||||||
printf("%3zu: %f\n", i, double(keys[i]));
|
printf("%3zu: %f\n", i, static_cast<double>(lanes[i]));
|
||||||
}
|
}
|
||||||
for (size_t i = right; i < total - misalign; ++i) {
|
for (size_t i = right; i < total - misalign; ++i) {
|
||||||
printf("%3zu: sentinel %f\n", i, double(keys[i]));
|
printf("%3zu: sentinel %f\n", i, static_cast<double>(lanes[i]));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
VerifyPartition(st, keys, left, border, right, N1, pivot2);
|
VerifyPartition(st, lanes, left, border, right, N1, pivot2);
|
||||||
for (size_t i = 0; i < misalign; ++i) {
|
for (size_t i = 0; i < misalign; ++i) {
|
||||||
if (aligned_keys[i] != hwy::LowestValue<T>())
|
if (aligned_lanes[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun misalign at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun misalign at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
for (size_t i = 0; i < left; ++i) {
|
for (size_t i = 0; i < left; ++i) {
|
||||||
if (keys[i] != hwy::LowestValue<T>())
|
if (lanes[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun left at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun left at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
for (size_t i = right; i < total - misalign; ++i) {
|
for (size_t i = right; i < total - misalign; ++i) {
|
||||||
if (keys[i] != hwy::LowestValue<T>())
|
if (lanes[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
} // misalign
|
} // misalign
|
||||||
|
@ -361,15 +356,18 @@ static HWY_NOINLINE void TestPartition() {
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllPartition() {
|
HWY_NOINLINE void TestAllPartition() {
|
||||||
TestPartition<LaneTraits<OrderAscending>, int16_t>();
|
// Only enable EMU128 on x86 - it's slow on emulators.
|
||||||
TestPartition<LaneTraits<OrderDescending>, int32_t>();
|
if (!HWY_ARCH_X86 && (HWY_TARGET == HWY_EMU128)) return;
|
||||||
TestPartition<LaneTraits<OrderAscending>, int64_t>();
|
|
||||||
TestPartition<LaneTraits<OrderDescending>, float>();
|
TestPartition<TraitsLane<OrderAscending<int16_t> > >();
|
||||||
|
TestPartition<TraitsLane<OrderDescending<int32_t> > >();
|
||||||
|
TestPartition<TraitsLane<OrderAscending<int64_t> > >();
|
||||||
|
TestPartition<TraitsLane<OrderDescending<float> > >();
|
||||||
#if HWY_HAVE_FLOAT64
|
#if HWY_HAVE_FLOAT64
|
||||||
TestPartition<LaneTraits<OrderDescending>, double>();
|
TestPartition<TraitsLane<OrderDescending<double> > >();
|
||||||
#endif
|
#endif
|
||||||
TestPartition<Traits128<OrderAscending128>, uint64_t>();
|
TestPartition<Traits128<OrderAscending128> >();
|
||||||
TestPartition<Traits128<OrderDescending128>, uint64_t>();
|
TestPartition<Traits128<OrderDescending128> >();
|
||||||
}
|
}
|
||||||
|
|
||||||
// (used for sample selection for choosing a pivot)
|
// (used for sample selection for choosing a pivot)
|
||||||
|
@ -399,7 +397,7 @@ static HWY_NOINLINE void TestRandomGenerator() {
|
||||||
|
|
||||||
// Also ensure the mean is near the middle of the range
|
// Also ensure the mean is near the middle of the range
|
||||||
const double expected = (num_blocks - 1) / 2.0;
|
const double expected = (num_blocks - 1) / 2.0;
|
||||||
const double actual = double(sum) / kReps;
|
const double actual = static_cast<double>(sum) / kReps;
|
||||||
HWY_ASSERT(0.9 * expected <= actual && actual <= 1.1 * expected);
|
HWY_ASSERT(0.9 * expected <= actual && actual <= 1.1 * expected);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -409,22 +407,26 @@ HWY_NOINLINE void TestAllGenerator() {
|
||||||
TestRandomGenerator<uint64_t>();
|
TestRandomGenerator<uint64_t>();
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif // VQSORT_TEST_IMPL
|
|
||||||
|
|
||||||
#if !VQSORT_TEST_SORT
|
|
||||||
static void TestAllSort() {}
|
|
||||||
#else
|
#else
|
||||||
|
static void TestAllMedian() {}
|
||||||
|
static void TestAllBaseCase() {}
|
||||||
|
static void TestAllPartition() {}
|
||||||
|
static void TestAllGenerator() {}
|
||||||
|
#endif // VQSORT_ENABLED
|
||||||
|
|
||||||
// Remembers input, and compares results to that of a reference algorithm.
|
// Remembers input, and compares results to that of a reference algorithm.
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
class CompareResults {
|
class CompareResults {
|
||||||
|
using LaneType = typename Traits::LaneType;
|
||||||
|
using KeyType = typename Traits::KeyType;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
void SetInput(const T* in, size_t num) {
|
CompareResults(const LaneType* in, size_t num_lanes) {
|
||||||
copy_.resize(num);
|
copy_.resize(num_lanes);
|
||||||
memcpy(copy_.data(), in, num * sizeof(T));
|
memcpy(copy_.data(), in, num_lanes * sizeof(LaneType));
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Verify(const T* output) {
|
bool Verify(const LaneType* output) {
|
||||||
#if HAVE_PDQSORT
|
#if HAVE_PDQSORT
|
||||||
const Algo reference = Algo::kPDQ;
|
const Algo reference = Algo::kPDQ;
|
||||||
#else
|
#else
|
||||||
|
@ -432,13 +434,28 @@ class CompareResults {
|
||||||
#endif
|
#endif
|
||||||
SharedState shared;
|
SharedState shared;
|
||||||
using Order = typename Traits::Order;
|
using Order = typename Traits::Order;
|
||||||
Run<Order>(reference, copy_.data(), copy_.size(), shared,
|
const Traits st;
|
||||||
/*thread=*/0);
|
const size_t num_keys = copy_.size() / st.LanesPerKey();
|
||||||
|
Run<Order>(reference, reinterpret_cast<KeyType*>(copy_.data()), num_keys,
|
||||||
|
shared, /*thread=*/0);
|
||||||
|
|
||||||
for (size_t i = 0; i < copy_.size(); ++i) {
|
for (size_t i = 0; i < copy_.size(); ++i) {
|
||||||
if (copy_[i] != output[i]) {
|
if (copy_[i] != output[i]) {
|
||||||
fprintf(stderr, "Asc %d mismatch at %d: %A %A\n", Order().IsAscending(),
|
if (sizeof(KeyType) == 16) {
|
||||||
static_cast<int>(i), double(copy_[i]), double(output[i]));
|
fprintf(stderr,
|
||||||
|
"%s Asc %d mismatch at %d of %d: %" PRIu64 " %" PRIu64 "\n",
|
||||||
|
st.KeyString().c_str(), Order().IsAscending(),
|
||||||
|
static_cast<int>(i), static_cast<int>(copy_.size()),
|
||||||
|
static_cast<uint64_t>(copy_[i]),
|
||||||
|
static_cast<uint64_t>(output[i]));
|
||||||
|
} else {
|
||||||
|
fprintf(stderr, "Type %s Asc %d mismatch at %d of %d: ",
|
||||||
|
st.KeyString().c_str(), Order().IsAscending(),
|
||||||
|
static_cast<int>(i), static_cast<int>(copy_.size()));
|
||||||
|
PrintValue(copy_[i]);
|
||||||
|
PrintValue(output[i]);
|
||||||
|
fprintf(stderr, "\n");
|
||||||
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -446,7 +463,7 @@ class CompareResults {
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::vector<T> copy_;
|
std::vector<LaneType> copy_;
|
||||||
};
|
};
|
||||||
|
|
||||||
std::vector<Algo> AlgoForTest() {
|
std::vector<Algo> AlgoForTest() {
|
||||||
|
@ -467,62 +484,65 @@ std::vector<Algo> AlgoForTest() {
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class Traits, typename T>
|
template <class Traits>
|
||||||
void TestSort(size_t num) {
|
void TestSort(size_t num_lanes) {
|
||||||
// TODO(janwas): fix
|
|
||||||
if (HWY_TARGET == HWY_SSSE3) return;
|
|
||||||
// Workaround for stack overflow on clang-cl (/F 8388608 does not help).
|
// Workaround for stack overflow on clang-cl (/F 8388608 does not help).
|
||||||
#if defined(_MSC_VER) && HWY_IS_DEBUG_BUILD && (HWY_TARGET == HWY_AVX3)
|
#if defined(_MSC_VER)
|
||||||
return;
|
return;
|
||||||
#endif
|
#endif
|
||||||
|
// Only enable EMU128 on x86 - it's slow on emulators.
|
||||||
|
if (!HWY_ARCH_X86 && (HWY_TARGET == HWY_EMU128)) return;
|
||||||
|
|
||||||
|
using Order = typename Traits::Order;
|
||||||
|
using LaneType = typename Traits::LaneType;
|
||||||
|
using KeyType = typename Traits::KeyType;
|
||||||
SharedState shared;
|
SharedState shared;
|
||||||
SharedTraits<Traits> st;
|
SharedTraits<Traits> st;
|
||||||
|
|
||||||
|
// Round up to a whole number of keys.
|
||||||
|
num_lanes += (st.Is128() && (num_lanes & 1));
|
||||||
|
const size_t num_keys = num_lanes / st.LanesPerKey();
|
||||||
|
|
||||||
constexpr size_t kMaxMisalign = 16;
|
constexpr size_t kMaxMisalign = 16;
|
||||||
auto aligned = hwy::AllocateAligned<T>(kMaxMisalign + num + kMaxMisalign);
|
auto aligned =
|
||||||
|
hwy::AllocateAligned<LaneType>(kMaxMisalign + num_lanes + kMaxMisalign);
|
||||||
for (Algo algo : AlgoForTest()) {
|
for (Algo algo : AlgoForTest()) {
|
||||||
#if HAVE_IPS4O
|
|
||||||
if (st.Is128() && (algo == Algo::kIPS4O || algo == Algo::kParallelIPS4O)) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
for (Dist dist : AllDist()) {
|
for (Dist dist : AllDist()) {
|
||||||
for (size_t misalign : {size_t{0}, size_t{st.LanesPerKey()},
|
for (size_t misalign : {size_t{0}, size_t{st.LanesPerKey()},
|
||||||
size_t{3 * st.LanesPerKey()}, kMaxMisalign / 2}) {
|
size_t{3 * st.LanesPerKey()}, kMaxMisalign / 2}) {
|
||||||
T* keys = aligned.get() + misalign;
|
LaneType* lanes = aligned.get() + misalign;
|
||||||
|
|
||||||
// Set up red zones before/after the keys to sort
|
// Set up red zones before/after the keys to sort
|
||||||
for (size_t i = 0; i < misalign; ++i) {
|
for (size_t i = 0; i < misalign; ++i) {
|
||||||
aligned[i] = hwy::LowestValue<T>();
|
aligned[i] = hwy::LowestValue<LaneType>();
|
||||||
}
|
}
|
||||||
for (size_t i = 0; i < kMaxMisalign; ++i) {
|
for (size_t i = 0; i < kMaxMisalign; ++i) {
|
||||||
keys[num + i] = hwy::HighestValue<T>();
|
lanes[num_lanes + i] = hwy::HighestValue<LaneType>();
|
||||||
}
|
}
|
||||||
#if HWY_IS_MSAN
|
#if HWY_IS_MSAN
|
||||||
__msan_poison(aligned.get(), misalign * sizeof(T));
|
__msan_poison(aligned.get(), misalign * sizeof(LaneType));
|
||||||
__msan_poison(keys + num, kMaxMisalign * sizeof(T));
|
__msan_poison(lanes + num_lanes, kMaxMisalign * sizeof(LaneType));
|
||||||
#endif
|
#endif
|
||||||
InputStats<T> input_stats = GenerateInput(dist, keys, num);
|
InputStats<LaneType> input_stats =
|
||||||
|
GenerateInput(dist, lanes, num_lanes);
|
||||||
|
|
||||||
CompareResults<Traits, T> compare;
|
CompareResults<Traits> compare(lanes, num_lanes);
|
||||||
compare.SetInput(keys, num);
|
Run<Order>(algo, reinterpret_cast<KeyType*>(lanes), num_keys, shared,
|
||||||
|
/*thread=*/0);
|
||||||
Run<typename Traits::Order>(algo, keys, num, shared, /*thread=*/0);
|
HWY_ASSERT(compare.Verify(lanes));
|
||||||
HWY_ASSERT(compare.Verify(keys));
|
HWY_ASSERT(VerifySort(st, input_stats, lanes, num_lanes, "TestSort"));
|
||||||
HWY_ASSERT(VerifySort(st, input_stats, keys, num, "TestSort"));
|
|
||||||
|
|
||||||
// Check red zones
|
// Check red zones
|
||||||
#if HWY_IS_MSAN
|
#if HWY_IS_MSAN
|
||||||
__msan_unpoison(aligned.get(), misalign * sizeof(T));
|
__msan_unpoison(aligned.get(), misalign * sizeof(LaneType));
|
||||||
__msan_unpoison(keys + num, kMaxMisalign * sizeof(T));
|
__msan_unpoison(lanes + num_lanes, kMaxMisalign * sizeof(LaneType));
|
||||||
#endif
|
#endif
|
||||||
for (size_t i = 0; i < misalign; ++i) {
|
for (size_t i = 0; i < misalign; ++i) {
|
||||||
if (aligned[i] != hwy::LowestValue<T>())
|
if (aligned[i] != hwy::LowestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun left at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun left at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
for (size_t i = num; i < num + kMaxMisalign; ++i) {
|
for (size_t i = num_lanes; i < num_lanes + kMaxMisalign; ++i) {
|
||||||
if (keys[i] != hwy::HighestValue<T>())
|
if (lanes[i] != hwy::HighestValue<LaneType>())
|
||||||
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
HWY_ABORT("Overrun right at %d\n", static_cast<int>(i));
|
||||||
}
|
}
|
||||||
} // misalign
|
} // misalign
|
||||||
|
@ -531,32 +551,37 @@ void TestSort(size_t num) {
|
||||||
}
|
}
|
||||||
|
|
||||||
void TestAllSort() {
|
void TestAllSort() {
|
||||||
const size_t num = 15 * 1000;
|
for (int num : {129, 504, 20 * 1000, 34567}) {
|
||||||
|
const size_t num_lanes = AdjustedReps(static_cast<size_t>(num));
|
||||||
|
TestSort<TraitsLane<OrderAscending<int16_t> > >(num_lanes);
|
||||||
|
TestSort<TraitsLane<OrderDescending<uint16_t> > >(num_lanes);
|
||||||
|
|
||||||
TestSort<LaneTraits<OrderAscending>, int16_t>(num);
|
TestSort<TraitsLane<OrderDescending<int32_t> > >(num_lanes);
|
||||||
TestSort<LaneTraits<OrderDescending>, uint16_t>(num);
|
TestSort<TraitsLane<OrderDescending<uint32_t> > >(num_lanes);
|
||||||
|
|
||||||
TestSort<LaneTraits<OrderDescending>, int32_t>(num);
|
TestSort<TraitsLane<OrderAscending<int64_t> > >(num_lanes);
|
||||||
TestSort<LaneTraits<OrderDescending>, uint32_t>(num);
|
TestSort<TraitsLane<OrderAscending<uint64_t> > >(num_lanes);
|
||||||
|
|
||||||
TestSort<LaneTraits<OrderAscending>, int64_t>(num);
|
// WARNING: for float types, SIMD comparisons will flush denormals to
|
||||||
TestSort<LaneTraits<OrderAscending>, uint64_t>(num);
|
// zero, causing mismatches with scalar sorts. In this test, we avoid
|
||||||
|
// generating denormal inputs.
|
||||||
// WARNING: for float types, SIMD comparisons will flush denormals to zero,
|
TestSort<TraitsLane<OrderAscending<float> > >(num_lanes);
|
||||||
// causing mismatches with scalar sorts. In this test, we avoid generating
|
|
||||||
// denormal inputs.
|
|
||||||
TestSort<LaneTraits<OrderAscending>, float>(num);
|
|
||||||
#if HWY_HAVE_FLOAT64 // protects algo-inl's GenerateRandom
|
#if HWY_HAVE_FLOAT64 // protects algo-inl's GenerateRandom
|
||||||
if (Sorter::HaveFloat64()) {
|
if (Sorter::HaveFloat64()) {
|
||||||
TestSort<LaneTraits<OrderDescending>, double>(num);
|
TestSort<TraitsLane<OrderDescending<double> > >(num_lanes);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
TestSort<Traits128<OrderAscending128>, uint64_t>(num);
|
// Our HeapSort does not support 128-bit keys.
|
||||||
TestSort<Traits128<OrderAscending128>, uint64_t>(num);
|
#if VQSORT_ENABLED
|
||||||
}
|
TestSort<Traits128<OrderAscending128> >(num_lanes);
|
||||||
|
TestSort<Traits128<OrderDescending128> >(num_lanes);
|
||||||
|
|
||||||
#endif // VQSORT_TEST_SORT
|
TestSort<Traits128<OrderAscendingKV128> >(num_lanes);
|
||||||
|
TestSort<Traits128<OrderDescendingKV128> >(num_lanes);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
@ -577,10 +602,4 @@ HWY_EXPORT_AND_TEST_P(SortTest, TestAllSort);
|
||||||
} // namespace
|
} // namespace
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -21,7 +22,6 @@
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_SORTING_NETWORKS_TOGGLE
|
#define HIGHWAY_HWY_CONTRIB_SORT_SORTING_NETWORKS_TOGGLE
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/shared-inl.h" // SortConstants
|
#include "hwy/contrib/sort/shared-inl.h" // SortConstants
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
@ -30,6 +30,8 @@ namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
|
#if VQSORT_ENABLED
|
||||||
|
|
||||||
using Constants = hwy::SortConstants;
|
using Constants = hwy::SortConstants;
|
||||||
|
|
||||||
// ------------------------------ SharedTraits
|
// ------------------------------ SharedTraits
|
||||||
|
@ -589,17 +591,19 @@ HWY_INLINE void Merge16(D d, Traits st, V& v0, V& v1, V& v2, V& v3, V& v4,
|
||||||
// Reshapes `buf` into a matrix, sorts columns independently, and then merges
|
// Reshapes `buf` into a matrix, sorts columns independently, and then merges
|
||||||
// into a sorted 1D array without transposing.
|
// into a sorted 1D array without transposing.
|
||||||
//
|
//
|
||||||
// `st` is SharedTraits<LaneTraits/Traits128<Order*>>. This abstraction layer
|
// `st` is SharedTraits<Traits*<Order*>>. This abstraction layer bridges
|
||||||
// bridges differences in sort order and single-lane vs 128-bit keys.
|
// differences in sort order and single-lane vs 128-bit keys.
|
||||||
// `buf` ensures full vectors are aligned, and enables loads/stores without
|
// `buf` ensures full vectors are aligned, and enables loads/stores without
|
||||||
// bounds checks.
|
// bounds checks.
|
||||||
//
|
//
|
||||||
|
// NOINLINE because this is large and called twice from vqsort-inl.h.
|
||||||
|
//
|
||||||
// References:
|
// References:
|
||||||
// https://drops.dagstuhl.de/opus/volltexte/2021/13775/pdf/LIPIcs-SEA-2021-3.pdf
|
// https://drops.dagstuhl.de/opus/volltexte/2021/13775/pdf/LIPIcs-SEA-2021-3.pdf
|
||||||
// https://github.com/simd-sorting/fast-and-robust/blob/master/avx2_sort_demo/avx2sort.h
|
// https://github.com/simd-sorting/fast-and-robust/blob/master/avx2_sort_demo/avx2sort.h
|
||||||
// "Entwurf und Implementierung vektorisierter Sortieralgorithmen" (M. Blacher)
|
// "Entwurf und Implementierung vektorisierter Sortieralgorithmen" (M. Blacher)
|
||||||
template <class Traits, typename T>
|
template <class Traits, typename T>
|
||||||
HWY_INLINE void SortingNetwork(Traits st, T* HWY_RESTRICT buf, size_t cols) {
|
HWY_NOINLINE void SortingNetwork(Traits st, T* HWY_RESTRICT buf, size_t cols) {
|
||||||
const CappedTag<T, Constants::kMaxCols> d;
|
const CappedTag<T, Constants::kMaxCols> d;
|
||||||
using V = decltype(Zero(d));
|
using V = decltype(Zero(d));
|
||||||
|
|
||||||
|
@ -646,8 +650,8 @@ HWY_INLINE void SortingNetwork(Traits st, T* HWY_RESTRICT buf, size_t cols) {
|
||||||
Merge8(d, st, v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, va, vb, vc, vd,
|
Merge8(d, st, v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, va, vb, vc, vd,
|
||||||
ve, vf);
|
ve, vf);
|
||||||
|
|
||||||
// Avoids build timeout
|
// Avoids build timeout. Must match #if condition in kMaxCols.
|
||||||
#if !HWY_COMPILER_MSVC
|
#if !HWY_COMPILER_MSVC && !HWY_IS_DEBUG_BUILD
|
||||||
if (HWY_LIKELY(keys >= 16 && kMaxKeys >= 16)) {
|
if (HWY_LIKELY(keys >= 16 && kMaxKeys >= 16)) {
|
||||||
Merge16(d, st, v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, va, vb, vc, vd,
|
Merge16(d, st, v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, va, vb, vc, vd,
|
||||||
ve, vf);
|
ve, vf);
|
||||||
|
@ -677,6 +681,11 @@ HWY_INLINE void SortingNetwork(Traits st, T* HWY_RESTRICT buf, size_t cols) {
|
||||||
StoreU(vf, d, buf + 0xf * cols);
|
StoreU(vf, d, buf + 0xf * cols);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
template <class Base>
|
||||||
|
struct SharedTraits : public Base {};
|
||||||
|
#endif // VQSORT_ENABLED
|
||||||
|
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
} // namespace HWY_NAMESPACE
|
} // namespace HWY_NAMESPACE
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -21,36 +22,64 @@
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_TRAITS_TOGGLE
|
#define HIGHWAY_HWY_CONTRIB_SORT_TRAITS_TOGGLE
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
#include <string>
|
||||||
|
|
||||||
#include "hwy/contrib/sort/shared-inl.h" // SortConstants
|
#include "hwy/contrib/sort/shared-inl.h" // SortConstants
|
||||||
#include "hwy/contrib/sort/vqsort.h" // SortDescending
|
#include "hwy/contrib/sort/vqsort.h" // SortDescending
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
#include "hwy/print.h"
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
|
#if VQSORT_ENABLED || HWY_IDE
|
||||||
|
|
||||||
// Highway does not provide a lane type for 128-bit keys, so we use uint64_t
|
// Highway does not provide a lane type for 128-bit keys, so we use uint64_t
|
||||||
// along with an abstraction layer for single-lane vs. lane-pair, which is
|
// along with an abstraction layer for single-lane vs. lane-pair, which is
|
||||||
// independent of the order.
|
// independent of the order.
|
||||||
|
template <typename T>
|
||||||
struct KeyLane {
|
struct KeyLane {
|
||||||
|
constexpr bool Is128() const { return false; }
|
||||||
constexpr size_t LanesPerKey() const { return 1; }
|
constexpr size_t LanesPerKey() const { return 1; }
|
||||||
|
|
||||||
|
// What type bench_sort should allocate for generating inputs.
|
||||||
|
using LaneType = T;
|
||||||
|
// What type to pass to Sorter::operator().
|
||||||
|
using KeyType = T;
|
||||||
|
|
||||||
|
std::string KeyString() const {
|
||||||
|
char string100[100];
|
||||||
|
hwy::detail::TypeName(hwy::detail::MakeTypeInfo<KeyType>(), 1, string100);
|
||||||
|
return string100;
|
||||||
|
}
|
||||||
|
|
||||||
// For HeapSort
|
// For HeapSort
|
||||||
template <typename T>
|
|
||||||
HWY_INLINE void Swap(T* a, T* b) const {
|
HWY_INLINE void Swap(T* a, T* b) const {
|
||||||
const T temp = *a;
|
const T temp = *a;
|
||||||
*a = *b;
|
*a = *b;
|
||||||
*b = temp;
|
*b = temp;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class V, class M>
|
||||||
|
HWY_INLINE V CompressKeys(V keys, M mask) const {
|
||||||
|
return CompressNot(keys, mask);
|
||||||
|
}
|
||||||
|
|
||||||
// Broadcasts one key into a vector
|
// Broadcasts one key into a vector
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> SetKey(D d, const TFromD<D>* key) const {
|
HWY_INLINE Vec<D> SetKey(D d, const T* key) const {
|
||||||
return Set(d, *key);
|
return Set(d, *key);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> EqualKeys(D /*tag*/, Vec<D> a, Vec<D> b) const {
|
||||||
|
return Eq(a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_INLINE bool Equal1(const T* a, const T* b) { return *a == *b; }
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> ReverseKeys(D d, Vec<D> v) const {
|
HWY_INLINE Vec<D> ReverseKeys(D d, Vec<D> v) const {
|
||||||
return Reverse(d, v);
|
return Reverse(d, v);
|
||||||
|
@ -148,10 +177,10 @@ struct KeyLane {
|
||||||
// We avoid overloaded functions because we want all functions to be callable
|
// We avoid overloaded functions because we want all functions to be callable
|
||||||
// from a SortTraits without per-function wrappers. Specializing would work, but
|
// from a SortTraits without per-function wrappers. Specializing would work, but
|
||||||
// we are anyway going to specialize at a higher level.
|
// we are anyway going to specialize at a higher level.
|
||||||
struct OrderAscending : public KeyLane {
|
template <typename T>
|
||||||
|
struct OrderAscending : public KeyLane<T> {
|
||||||
using Order = SortAscending;
|
using Order = SortAscending;
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
||||||
return *a < *b;
|
return *a < *b;
|
||||||
}
|
}
|
||||||
|
@ -174,31 +203,31 @@ struct OrderAscending : public KeyLane {
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
||||||
TFromD<D>* HWY_RESTRICT /* buf */) const {
|
T* HWY_RESTRICT /* buf */) const {
|
||||||
return MinOfLanes(d, v);
|
return MinOfLanes(d, v);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
||||||
TFromD<D>* HWY_RESTRICT /* buf */) const {
|
T* HWY_RESTRICT /* buf */) const {
|
||||||
return MaxOfLanes(d, v);
|
return MaxOfLanes(d, v);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> FirstValue(D d) const {
|
HWY_INLINE Vec<D> FirstValue(D d) const {
|
||||||
return Set(d, hwy::LowestValue<TFromD<D>>());
|
return Set(d, hwy::LowestValue<T>());
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> LastValue(D d) const {
|
HWY_INLINE Vec<D> LastValue(D d) const {
|
||||||
return Set(d, hwy::HighestValue<TFromD<D>>());
|
return Set(d, hwy::HighestValue<T>());
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
struct OrderDescending : public KeyLane {
|
template <typename T>
|
||||||
|
struct OrderDescending : public KeyLane<T> {
|
||||||
using Order = SortDescending;
|
using Order = SortDescending;
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
||||||
return *b < *a;
|
return *b < *a;
|
||||||
}
|
}
|
||||||
|
@ -220,32 +249,30 @@ struct OrderDescending : public KeyLane {
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
||||||
TFromD<D>* HWY_RESTRICT /* buf */) const {
|
T* HWY_RESTRICT /* buf */) const {
|
||||||
return MaxOfLanes(d, v);
|
return MaxOfLanes(d, v);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
||||||
TFromD<D>* HWY_RESTRICT /* buf */) const {
|
T* HWY_RESTRICT /* buf */) const {
|
||||||
return MinOfLanes(d, v);
|
return MinOfLanes(d, v);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> FirstValue(D d) const {
|
HWY_INLINE Vec<D> FirstValue(D d) const {
|
||||||
return Set(d, hwy::HighestValue<TFromD<D>>());
|
return Set(d, hwy::HighestValue<T>());
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> LastValue(D d) const {
|
HWY_INLINE Vec<D> LastValue(D d) const {
|
||||||
return Set(d, hwy::LowestValue<TFromD<D>>());
|
return Set(d, hwy::LowestValue<T>());
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
// Shared code that depends on Order.
|
// Shared code that depends on Order.
|
||||||
template <class Base>
|
template <class Base>
|
||||||
struct LaneTraits : public Base {
|
struct TraitsLane : public Base {
|
||||||
constexpr bool Is128() const { return false; }
|
|
||||||
|
|
||||||
// For each lane i: replaces a[i] with the first and b[i] with the second
|
// For each lane i: replaces a[i] with the first and b[i] with the second
|
||||||
// according to Base.
|
// according to Base.
|
||||||
// Corresponds to a conditional swap, which is one "node" of a sorting
|
// Corresponds to a conditional swap, which is one "node" of a sorting
|
||||||
|
@ -315,6 +342,66 @@ struct LaneTraits : public Base {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
// Base class shared between OrderAscending, OrderDescending.
|
||||||
|
template <typename T>
|
||||||
|
struct KeyLane {
|
||||||
|
constexpr bool Is128() const { return false; }
|
||||||
|
constexpr size_t LanesPerKey() const { return 1; }
|
||||||
|
|
||||||
|
using LaneType = T;
|
||||||
|
using KeyType = T;
|
||||||
|
|
||||||
|
std::string KeyString() const {
|
||||||
|
char string100[100];
|
||||||
|
hwy::detail::TypeName(hwy::detail::MakeTypeInfo<KeyType>(), 1, string100);
|
||||||
|
return string100;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
struct OrderAscending : public KeyLane<T> {
|
||||||
|
using Order = SortAscending;
|
||||||
|
|
||||||
|
HWY_INLINE bool Compare1(const T* a, const T* b) { return *a < *b; }
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> Compare(D /* tag */, Vec<D> a, Vec<D> b) {
|
||||||
|
return Lt(a, b);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
struct OrderDescending : public KeyLane<T> {
|
||||||
|
using Order = SortDescending;
|
||||||
|
|
||||||
|
HWY_INLINE bool Compare1(const T* a, const T* b) { return *b < *a; }
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> Compare(D /* tag */, Vec<D> a, Vec<D> b) {
|
||||||
|
return Lt(b, a);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class Order>
|
||||||
|
struct TraitsLane : public Order {
|
||||||
|
// For HeapSort
|
||||||
|
template <typename T> // MSVC doesn't find typename Order::LaneType.
|
||||||
|
HWY_INLINE void Swap(T* a, T* b) const {
|
||||||
|
const T temp = *a;
|
||||||
|
*a = *b;
|
||||||
|
*b = temp;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> SetKey(D d, const TFromD<D>* key) const {
|
||||||
|
return Set(d, *key);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // VQSORT_ENABLED
|
||||||
|
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
} // namespace HWY_NAMESPACE
|
} // namespace HWY_NAMESPACE
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -21,6 +22,9 @@
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_TRAITS128_TOGGLE
|
#define HIGHWAY_HWY_CONTRIB_SORT_TRAITS128_TOGGLE
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#include "hwy/contrib/sort/shared-inl.h"
|
||||||
#include "hwy/contrib/sort/vqsort.h" // SortDescending
|
#include "hwy/contrib/sort/vqsort.h" // SortDescending
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
@ -29,48 +33,31 @@ namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
#if HWY_TARGET == HWY_SCALAR
|
#if VQSORT_ENABLED || HWY_IDE
|
||||||
|
|
||||||
struct OrderAscending128 {
|
|
||||||
using Order = SortAscending;
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
|
||||||
return (a[1] == b[1]) ? a[0] < b[0] : a[1] < b[1];
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
struct OrderDescending128 {
|
|
||||||
using Order = SortDescending;
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
|
||||||
return (a[1] == b[1]) ? b[0] < a[0] : b[1] < a[1];
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <class Order>
|
|
||||||
struct Traits128 : public Order {
|
|
||||||
constexpr bool Is128() const { return true; }
|
|
||||||
constexpr size_t LanesPerKey() const { return 2; }
|
|
||||||
};
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
// Highway does not provide a lane type for 128-bit keys, so we use uint64_t
|
// Highway does not provide a lane type for 128-bit keys, so we use uint64_t
|
||||||
// along with an abstraction layer for single-lane vs. lane-pair, which is
|
// along with an abstraction layer for single-lane vs. lane-pair, which is
|
||||||
// independent of the order.
|
// independent of the order.
|
||||||
struct Key128 {
|
struct KeyAny128 {
|
||||||
|
constexpr bool Is128() const { return true; }
|
||||||
constexpr size_t LanesPerKey() const { return 2; }
|
constexpr size_t LanesPerKey() const { return 2; }
|
||||||
|
|
||||||
template <typename T>
|
// What type bench_sort should allocate for generating inputs.
|
||||||
HWY_INLINE void Swap(T* a, T* b) const {
|
using LaneType = uint64_t;
|
||||||
const FixedTag<T, 2> d;
|
// KeyType and KeyString are defined by derived classes.
|
||||||
|
|
||||||
|
HWY_INLINE void Swap(LaneType* a, LaneType* b) const {
|
||||||
|
const FixedTag<LaneType, 2> d;
|
||||||
const auto temp = LoadU(d, a);
|
const auto temp = LoadU(d, a);
|
||||||
StoreU(LoadU(d, b), d, a);
|
StoreU(LoadU(d, b), d, a);
|
||||||
StoreU(temp, d, b);
|
StoreU(temp, d, b);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class V, class M>
|
||||||
|
HWY_INLINE V CompressKeys(V keys, M mask) const {
|
||||||
|
return CompressBlocksNot(keys, mask);
|
||||||
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> SetKey(D d, const TFromD<D>* key) const {
|
HWY_INLINE Vec<D> SetKey(D d, const TFromD<D>* key) const {
|
||||||
return LoadDup128(d, key);
|
return LoadDup128(d, key);
|
||||||
|
@ -135,6 +122,23 @@ struct Key128 {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Base class shared between OrderAscending128, OrderDescending128.
|
||||||
|
struct Key128 : public KeyAny128 {
|
||||||
|
// What type to pass to Sorter::operator().
|
||||||
|
using KeyType = hwy::uint128_t;
|
||||||
|
|
||||||
|
std::string KeyString() const { return "U128"; }
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> EqualKeys(D /*tag*/, Vec<D> a, Vec<D> b) const {
|
||||||
|
return Eq128(a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_INLINE bool Equal1(const LaneType* a, const LaneType* b) {
|
||||||
|
return a[0] == b[0] && a[1] == b[1];
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
// Anything order-related depends on the key traits *and* the order (see
|
// Anything order-related depends on the key traits *and* the order (see
|
||||||
// FirstOfLanes). We cannot implement just one Compare function because Lt128
|
// FirstOfLanes). We cannot implement just one Compare function because Lt128
|
||||||
// only compiles if the lane type is u64. Thus we need either overloaded
|
// only compiles if the lane type is u64. Thus we need either overloaded
|
||||||
|
@ -145,8 +149,7 @@ struct Key128 {
|
||||||
struct OrderAscending128 : public Key128 {
|
struct OrderAscending128 : public Key128 {
|
||||||
using Order = SortAscending;
|
using Order = SortAscending;
|
||||||
|
|
||||||
template <typename T>
|
HWY_INLINE bool Compare1(const LaneType* a, const LaneType* b) {
|
||||||
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
|
||||||
return (a[1] == b[1]) ? a[0] < b[0] : a[1] < b[1];
|
return (a[1] == b[1]) ? a[0] < b[0] : a[1] < b[1];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -171,30 +174,6 @@ struct OrderAscending128 : public Key128 {
|
||||||
return Max128(d, a, b);
|
return Max128(d, a, b);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
|
||||||
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
|
||||||
TFromD<D>* HWY_RESTRICT buf) const {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
Store(v, d, buf);
|
|
||||||
v = SetKey(d, buf + 0); // result must be broadcasted
|
|
||||||
for (size_t i = LanesPerKey(); i < N; i += LanesPerKey()) {
|
|
||||||
v = First(d, v, SetKey(d, buf + i));
|
|
||||||
}
|
|
||||||
return v;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D>
|
|
||||||
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
|
||||||
TFromD<D>* HWY_RESTRICT buf) const {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
Store(v, d, buf);
|
|
||||||
v = SetKey(d, buf + 0); // result must be broadcasted
|
|
||||||
for (size_t i = LanesPerKey(); i < N; i += LanesPerKey()) {
|
|
||||||
v = Last(d, v, SetKey(d, buf + i));
|
|
||||||
}
|
|
||||||
return v;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Same as for regular lanes because 128-bit lanes are u64.
|
// Same as for regular lanes because 128-bit lanes are u64.
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> FirstValue(D d) const {
|
HWY_INLINE Vec<D> FirstValue(D d) const {
|
||||||
|
@ -210,8 +189,7 @@ struct OrderAscending128 : public Key128 {
|
||||||
struct OrderDescending128 : public Key128 {
|
struct OrderDescending128 : public Key128 {
|
||||||
using Order = SortDescending;
|
using Order = SortDescending;
|
||||||
|
|
||||||
template <typename T>
|
HWY_INLINE bool Compare1(const LaneType* a, const LaneType* b) {
|
||||||
HWY_INLINE bool Compare1(const T* a, const T* b) {
|
|
||||||
return (a[1] == b[1]) ? b[0] < a[0] : b[1] < a[1];
|
return (a[1] == b[1]) ? b[0] < a[0] : b[1] < a[1];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -236,28 +214,101 @@ struct OrderDescending128 : public Key128 {
|
||||||
return Min128(d, a, b);
|
return Min128(d, a, b);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Same as for regular lanes because 128-bit lanes are u64.
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
HWY_INLINE Vec<D> FirstValue(D d) const {
|
||||||
TFromD<D>* HWY_RESTRICT buf) const {
|
return Set(d, hwy::HighestValue<TFromD<D> >());
|
||||||
const size_t N = Lanes(d);
|
|
||||||
Store(v, d, buf);
|
|
||||||
v = SetKey(d, buf + 0); // result must be broadcasted
|
|
||||||
for (size_t i = LanesPerKey(); i < N; i += LanesPerKey()) {
|
|
||||||
v = First(d, v, SetKey(d, buf + i));
|
|
||||||
}
|
|
||||||
return v;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
HWY_INLINE Vec<D> LastValue(D d) const {
|
||||||
TFromD<D>* HWY_RESTRICT buf) const {
|
return Set(d, hwy::LowestValue<TFromD<D> >());
|
||||||
const size_t N = Lanes(d);
|
|
||||||
Store(v, d, buf);
|
|
||||||
v = SetKey(d, buf + 0); // result must be broadcasted
|
|
||||||
for (size_t i = LanesPerKey(); i < N; i += LanesPerKey()) {
|
|
||||||
v = Last(d, v, SetKey(d, buf + i));
|
|
||||||
}
|
}
|
||||||
return v;
|
};
|
||||||
|
|
||||||
|
// Base class shared between OrderAscendingKV128, OrderDescendingKV128.
|
||||||
|
struct KeyValue128 : public KeyAny128 {
|
||||||
|
// What type to pass to Sorter::operator().
|
||||||
|
using KeyType = K64V64;
|
||||||
|
|
||||||
|
std::string KeyString() const { return "KV128"; }
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> EqualKeys(D /*tag*/, Vec<D> a, Vec<D> b) const {
|
||||||
|
return Eq128Upper(a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_INLINE bool Equal1(const LaneType* a, const LaneType* b) {
|
||||||
|
return a[1] == b[1];
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct OrderAscendingKV128 : public KeyValue128 {
|
||||||
|
using Order = SortAscending;
|
||||||
|
|
||||||
|
HWY_INLINE bool Compare1(const LaneType* a, const LaneType* b) {
|
||||||
|
return a[1] < b[1];
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> Compare(D d, Vec<D> a, Vec<D> b) const {
|
||||||
|
return Lt128Upper(d, a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Used by CompareTop
|
||||||
|
template <class V>
|
||||||
|
HWY_INLINE Mask<DFromV<V> > CompareLanes(V a, V b) const {
|
||||||
|
return Lt(a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> First(D d, const Vec<D> a, const Vec<D> b) const {
|
||||||
|
return Min128Upper(d, a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> Last(D d, const Vec<D> a, const Vec<D> b) const {
|
||||||
|
return Max128Upper(d, a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Same as for regular lanes because 128-bit lanes are u64.
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> FirstValue(D d) const {
|
||||||
|
return Set(d, hwy::LowestValue<TFromD<D> >());
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> LastValue(D d) const {
|
||||||
|
return Set(d, hwy::HighestValue<TFromD<D> >());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
struct OrderDescendingKV128 : public KeyValue128 {
|
||||||
|
using Order = SortDescending;
|
||||||
|
|
||||||
|
HWY_INLINE bool Compare1(const LaneType* a, const LaneType* b) {
|
||||||
|
return b[1] < a[1];
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Mask<D> Compare(D d, Vec<D> a, Vec<D> b) const {
|
||||||
|
return Lt128Upper(d, b, a);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Used by CompareTop
|
||||||
|
template <class V>
|
||||||
|
HWY_INLINE Mask<DFromV<V> > CompareLanes(V a, V b) const {
|
||||||
|
return Lt(b, a);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> First(D d, const Vec<D> a, const Vec<D> b) const {
|
||||||
|
return Max128Upper(d, a, b);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> Last(D d, const Vec<D> a, const Vec<D> b) const {
|
||||||
|
return Min128Upper(d, a, b);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Same as for regular lanes because 128-bit lanes are u64.
|
// Same as for regular lanes because 128-bit lanes are u64.
|
||||||
|
@ -275,16 +326,21 @@ struct OrderDescending128 : public Key128 {
|
||||||
// Shared code that depends on Order.
|
// Shared code that depends on Order.
|
||||||
template <class Base>
|
template <class Base>
|
||||||
class Traits128 : public Base {
|
class Traits128 : public Base {
|
||||||
#if HWY_TARGET <= HWY_AVX2
|
// Special case for >= 256 bit vectors
|
||||||
|
#if HWY_TARGET <= HWY_AVX2 || HWY_TARGET == HWY_SVE_256
|
||||||
// Returns vector with only the top u64 lane valid. Useful when the next step
|
// Returns vector with only the top u64 lane valid. Useful when the next step
|
||||||
// is to replicate the mask anyway.
|
// is to replicate the mask anyway.
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE HWY_MAYBE_UNUSED Vec<D> CompareTop(D d, Vec<D> a, Vec<D> b) const {
|
HWY_INLINE HWY_MAYBE_UNUSED Vec<D> CompareTop(D d, Vec<D> a, Vec<D> b) const {
|
||||||
const Base* base = static_cast<const Base*>(this);
|
const Base* base = static_cast<const Base*>(this);
|
||||||
const Vec<D> eqHL = VecFromMask(d, Eq(a, b));
|
const Mask<D> eqHL = Eq(a, b);
|
||||||
const Vec<D> ltHL = VecFromMask(d, base->CompareLanes(a, b));
|
const Vec<D> ltHL = VecFromMask(d, base->CompareLanes(a, b));
|
||||||
|
#if HWY_TARGET == HWY_SVE_256
|
||||||
|
return IfThenElse(eqHL, DupEven(ltHL), ltHL);
|
||||||
|
#else
|
||||||
const Vec<D> ltLX = ShiftLeftLanes<1>(ltHL);
|
const Vec<D> ltLX = ShiftLeftLanes<1>(ltHL);
|
||||||
return OrAnd(ltHL, eqHL, ltLX);
|
return OrAnd(ltHL, VecFromMask(d, eqHL), ltLX);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// We want to swap 2 u128, i.e. 4 u64 lanes, based on the 0 or FF..FF mask in
|
// We want to swap 2 u128, i.e. 4 u64 lanes, based on the 0 or FF..FF mask in
|
||||||
|
@ -292,16 +348,42 @@ class Traits128 : public Base {
|
||||||
// replicate it 4x. Only called for >= 256-bit vectors.
|
// replicate it 4x. Only called for >= 256-bit vectors.
|
||||||
template <class V>
|
template <class V>
|
||||||
HWY_INLINE V ReplicateTop4x(V v) const {
|
HWY_INLINE V ReplicateTop4x(V v) const {
|
||||||
#if HWY_TARGET <= HWY_AVX3
|
#if HWY_TARGET == HWY_SVE_256
|
||||||
|
return svdup_lane_u64(v, 3);
|
||||||
|
#elif HWY_TARGET <= HWY_AVX3
|
||||||
return V{_mm512_permutex_epi64(v.raw, _MM_SHUFFLE(3, 3, 3, 3))};
|
return V{_mm512_permutex_epi64(v.raw, _MM_SHUFFLE(3, 3, 3, 3))};
|
||||||
#else // AVX2
|
#else // AVX2
|
||||||
return V{_mm256_permute4x64_epi64(v.raw, _MM_SHUFFLE(3, 3, 3, 3))};
|
return V{_mm256_permute4x64_epi64(v.raw, _MM_SHUFFLE(3, 3, 3, 3))};
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
#endif
|
#endif // HWY_TARGET
|
||||||
|
|
||||||
public:
|
public:
|
||||||
constexpr bool Is128() const { return true; }
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> FirstOfLanes(D d, Vec<D> v,
|
||||||
|
TFromD<D>* HWY_RESTRICT buf) const {
|
||||||
|
const Base* base = static_cast<const Base*>(this);
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
Store(v, d, buf);
|
||||||
|
v = base->SetKey(d, buf + 0); // result must be broadcasted
|
||||||
|
for (size_t i = base->LanesPerKey(); i < N; i += base->LanesPerKey()) {
|
||||||
|
v = base->First(d, v, base->SetKey(d, buf + i));
|
||||||
|
}
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
HWY_INLINE Vec<D> LastOfLanes(D d, Vec<D> v,
|
||||||
|
TFromD<D>* HWY_RESTRICT buf) const {
|
||||||
|
const Base* base = static_cast<const Base*>(this);
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
Store(v, d, buf);
|
||||||
|
v = base->SetKey(d, buf + 0); // result must be broadcasted
|
||||||
|
for (size_t i = base->LanesPerKey(); i < N; i += base->LanesPerKey()) {
|
||||||
|
v = base->Last(d, v, base->SetKey(d, buf + i));
|
||||||
|
}
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE void Sort2(D d, Vec<D>& a, Vec<D>& b) const {
|
HWY_INLINE void Sort2(D d, Vec<D>& a, Vec<D>& b) const {
|
||||||
|
@ -319,7 +401,7 @@ class Traits128 : public Base {
|
||||||
const Base* base = static_cast<const Base*>(this);
|
const Base* base = static_cast<const Base*>(this);
|
||||||
Vec<D> swapped = base->ReverseKeys2(d, v);
|
Vec<D> swapped = base->ReverseKeys2(d, v);
|
||||||
|
|
||||||
#if HWY_TARGET <= HWY_AVX2
|
#if HWY_TARGET <= HWY_AVX2 || HWY_TARGET == HWY_SVE_256
|
||||||
const Vec<D> select = ReplicateTop4x(CompareTop(d, v, swapped));
|
const Vec<D> select = ReplicateTop4x(CompareTop(d, v, swapped));
|
||||||
return IfVecThenElse(select, swapped, v);
|
return IfVecThenElse(select, swapped, v);
|
||||||
#else
|
#else
|
||||||
|
@ -357,7 +439,7 @@ class Traits128 : public Base {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // HWY_TARGET != HWY_SCALAR
|
#endif // VQSORT_ENABLED
|
||||||
|
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -16,6 +17,10 @@
|
||||||
#ifndef HIGHWAY_HWY_CONTRIB_SORT_VQSORT_INL_H_
|
#ifndef HIGHWAY_HWY_CONTRIB_SORT_VQSORT_INL_H_
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_VQSORT_INL_H_
|
#define HIGHWAY_HWY_CONTRIB_SORT_VQSORT_INL_H_
|
||||||
|
|
||||||
|
#ifndef VQSORT_PRINT
|
||||||
|
#define VQSORT_PRINT 0
|
||||||
|
#endif
|
||||||
|
|
||||||
// Makes it harder for adversaries to predict our sampling locations, at the
|
// Makes it harder for adversaries to predict our sampling locations, at the
|
||||||
// cost of 1-2% increased runtime.
|
// cost of 1-2% increased runtime.
|
||||||
#ifndef VQSORT_SECURE_RNG
|
#ifndef VQSORT_SECURE_RNG
|
||||||
|
@ -26,10 +31,13 @@
|
||||||
#include "third_party/absl/random/random.h"
|
#include "third_party/absl/random/random.h"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
#include <stdio.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
#include <string.h> // memcpy
|
#include <string.h> // memcpy
|
||||||
|
|
||||||
#include "hwy/cache_control.h" // Prefetch
|
#include "hwy/cache_control.h" // Prefetch
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h" // Fill24Bytes
|
#include "hwy/contrib/sort/vqsort.h" // Fill24Bytes
|
||||||
|
|
||||||
#if HWY_IS_MSAN
|
#if HWY_IS_MSAN
|
||||||
|
@ -47,6 +55,10 @@
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_VQSORT_TOGGLE
|
#define HIGHWAY_HWY_CONTRIB_SORT_VQSORT_TOGGLE
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
#include "hwy/print-inl.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
#include "hwy/contrib/sort/shared-inl.h"
|
#include "hwy/contrib/sort/shared-inl.h"
|
||||||
#include "hwy/contrib/sort/sorting_networks-inl.h"
|
#include "hwy/contrib/sort/sorting_networks-inl.h"
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
@ -56,117 +68,66 @@ namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
#if HWY_TARGET == HWY_SCALAR
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
void Swap(T* a, T* b) {
|
|
||||||
T t = *a;
|
|
||||||
*a = *b;
|
|
||||||
*b = t;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Scalar version of HeapSort (see below)
|
|
||||||
template <class Traits, typename T>
|
|
||||||
void HeapSort(Traits st, T* HWY_RESTRICT keys, const size_t num) {
|
|
||||||
if (num < 2) return;
|
|
||||||
|
|
||||||
// Build heap.
|
|
||||||
for (size_t i = 1; i < num; i += 1) {
|
|
||||||
size_t j = i;
|
|
||||||
while (j != 0) {
|
|
||||||
const size_t idx_parent = ((j - 1) / 1 / 2);
|
|
||||||
if (!st.Compare1(keys + idx_parent, keys + j)) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
Swap(keys + j, keys + idx_parent);
|
|
||||||
j = idx_parent;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (size_t i = num - 1; i != 0; i -= 1) {
|
|
||||||
// Swap root with last
|
|
||||||
Swap(keys + 0, keys + i);
|
|
||||||
|
|
||||||
// Sift down the new root.
|
|
||||||
size_t j = 0;
|
|
||||||
while (j < i) {
|
|
||||||
const size_t left = 2 * j + 1;
|
|
||||||
const size_t right = 2 * j + 2;
|
|
||||||
if (left >= i) break;
|
|
||||||
size_t idx_larger = j;
|
|
||||||
if (st.Compare1(keys + j, keys + left)) {
|
|
||||||
idx_larger = left;
|
|
||||||
}
|
|
||||||
if (right < i && st.Compare1(keys + idx_larger, keys + right)) {
|
|
||||||
idx_larger = right;
|
|
||||||
}
|
|
||||||
if (idx_larger == j) break;
|
|
||||||
Swap(keys + j, keys + idx_larger);
|
|
||||||
j = idx_larger;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
using Constants = hwy::SortConstants;
|
using Constants = hwy::SortConstants;
|
||||||
|
|
||||||
// ------------------------------ HeapSort
|
// ------------------------------ HeapSort
|
||||||
|
|
||||||
// Heapsort: O(1) space, O(N*logN) worst-case comparisons.
|
|
||||||
// Based on LLVM sanitizer_common.h, licensed under Apache-2.0.
|
|
||||||
template <class Traits, typename T>
|
template <class Traits, typename T>
|
||||||
void HeapSort(Traits st, T* HWY_RESTRICT keys, const size_t num) {
|
void SiftDown(Traits st, T* HWY_RESTRICT lanes, const size_t num_lanes,
|
||||||
|
size_t start) {
|
||||||
constexpr size_t N1 = st.LanesPerKey();
|
constexpr size_t N1 = st.LanesPerKey();
|
||||||
const FixedTag<T, N1> d;
|
const FixedTag<T, N1> d;
|
||||||
|
|
||||||
if (num < 2 * N1) return;
|
while (start < num_lanes) {
|
||||||
|
const size_t left = 2 * start + N1;
|
||||||
// Build heap.
|
const size_t right = 2 * start + 2 * N1;
|
||||||
for (size_t i = N1; i < num; i += N1) {
|
if (left >= num_lanes) break;
|
||||||
size_t j = i;
|
size_t idx_larger = start;
|
||||||
while (j != 0) {
|
const auto key_j = st.SetKey(d, lanes + start);
|
||||||
const size_t idx_parent = ((j - N1) / N1 / 2) * N1;
|
if (AllTrue(d, st.Compare(d, key_j, st.SetKey(d, lanes + left)))) {
|
||||||
if (AllFalse(d, st.Compare(d, st.SetKey(d, keys + idx_parent),
|
|
||||||
st.SetKey(d, keys + j)))) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
st.Swap(keys + j, keys + idx_parent);
|
|
||||||
j = idx_parent;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (size_t i = num - N1; i != 0; i -= N1) {
|
|
||||||
// Swap root with last
|
|
||||||
st.Swap(keys + 0, keys + i);
|
|
||||||
|
|
||||||
// Sift down the new root.
|
|
||||||
size_t j = 0;
|
|
||||||
while (j < i) {
|
|
||||||
const size_t left = 2 * j + N1;
|
|
||||||
const size_t right = 2 * j + 2 * N1;
|
|
||||||
if (left >= i) break;
|
|
||||||
size_t idx_larger = j;
|
|
||||||
const auto key_j = st.SetKey(d, keys + j);
|
|
||||||
if (AllTrue(d, st.Compare(d, key_j, st.SetKey(d, keys + left)))) {
|
|
||||||
idx_larger = left;
|
idx_larger = left;
|
||||||
}
|
}
|
||||||
if (right < i && AllTrue(d, st.Compare(d, st.SetKey(d, keys + idx_larger),
|
if (right < num_lanes &&
|
||||||
st.SetKey(d, keys + right)))) {
|
AllTrue(d, st.Compare(d, st.SetKey(d, lanes + idx_larger),
|
||||||
|
st.SetKey(d, lanes + right)))) {
|
||||||
idx_larger = right;
|
idx_larger = right;
|
||||||
}
|
}
|
||||||
if (idx_larger == j) break;
|
if (idx_larger == start) break;
|
||||||
st.Swap(keys + j, keys + idx_larger);
|
st.Swap(lanes + start, lanes + idx_larger);
|
||||||
j = idx_larger;
|
start = idx_larger;
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Heapsort: O(1) space, O(N*logN) worst-case comparisons.
|
||||||
|
// Based on LLVM sanitizer_common.h, licensed under Apache-2.0.
|
||||||
|
template <class Traits, typename T>
|
||||||
|
void HeapSort(Traits st, T* HWY_RESTRICT lanes, const size_t num_lanes) {
|
||||||
|
constexpr size_t N1 = st.LanesPerKey();
|
||||||
|
|
||||||
|
if (num_lanes < 2 * N1) return;
|
||||||
|
|
||||||
|
// Build heap.
|
||||||
|
for (size_t i = ((num_lanes - N1) / N1 / 2) * N1; i != (~N1 + 1); i -= N1) {
|
||||||
|
SiftDown(st, lanes, num_lanes, i);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (size_t i = num_lanes - N1; i != 0; i -= N1) {
|
||||||
|
// Swap root with last
|
||||||
|
st.Swap(lanes + 0, lanes + i);
|
||||||
|
|
||||||
|
// Sift down the new root.
|
||||||
|
SiftDown(st, lanes, i, 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#if VQSORT_ENABLED || HWY_IDE
|
||||||
|
|
||||||
// ------------------------------ BaseCase
|
// ------------------------------ BaseCase
|
||||||
|
|
||||||
// Sorts `keys` within the range [0, num) via sorting network.
|
// Sorts `keys` within the range [0, num) via sorting network.
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
HWY_NOINLINE void BaseCase(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
HWY_NOINLINE void BaseCase(D d, Traits st, T* HWY_RESTRICT keys,
|
||||||
|
T* HWY_RESTRICT keys_end, size_t num,
|
||||||
T* HWY_RESTRICT buf) {
|
T* HWY_RESTRICT buf) {
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
using V = decltype(Zero(d));
|
using V = decltype(Zero(d));
|
||||||
|
@ -184,14 +145,25 @@ HWY_NOINLINE void BaseCase(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
HWY_MAX(st.LanesPerKey(), num_pow2 >> Constants::kMaxRowsLog2);
|
HWY_MAX(st.LanesPerKey(), num_pow2 >> Constants::kMaxRowsLog2);
|
||||||
HWY_DASSERT(cols <= N);
|
HWY_DASSERT(cols <= N);
|
||||||
|
|
||||||
|
// We can avoid padding and load/store directly to `keys` after checking the
|
||||||
|
// original input array has enough space. Except at the right border, it's OK
|
||||||
|
// to sort more than the current sub-array. Even if we sort across a previous
|
||||||
|
// partition point, we know that keys will not migrate across it. However, we
|
||||||
|
// must use the maximum size of the sorting network, because the StoreU of its
|
||||||
|
// last vector would otherwise write invalid data starting at kMaxRows * cols.
|
||||||
|
const size_t N_sn = Lanes(CappedTag<T, Constants::kMaxCols>());
|
||||||
|
if (HWY_LIKELY(keys + N_sn * Constants::kMaxRows <= keys_end)) {
|
||||||
|
SortingNetwork(st, keys, N_sn);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
// Copy `keys` to `buf`.
|
// Copy `keys` to `buf`.
|
||||||
size_t i;
|
size_t i;
|
||||||
for (i = 0; i + N <= num; i += N) {
|
for (i = 0; i + N <= num; i += N) {
|
||||||
Store(LoadU(d, keys + i), d, buf + i);
|
Store(LoadU(d, keys + i), d, buf + i);
|
||||||
}
|
}
|
||||||
for (; i < num; ++i) {
|
SafeCopyN(num - i, d, keys + i, buf + i);
|
||||||
buf[i] = keys[i];
|
i = num;
|
||||||
}
|
|
||||||
|
|
||||||
// Fill with padding - last in sort order, not copied to keys.
|
// Fill with padding - last in sort order, not copied to keys.
|
||||||
const V kPadding = st.LastValue(d);
|
const V kPadding = st.LastValue(d);
|
||||||
|
@ -206,9 +178,7 @@ HWY_NOINLINE void BaseCase(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
for (i = 0; i + N <= num; i += N) {
|
for (i = 0; i + N <= num; i += N) {
|
||||||
StoreU(Load(d, buf + i), d, keys + i);
|
StoreU(Load(d, buf + i), d, keys + i);
|
||||||
}
|
}
|
||||||
for (; i < num; ++i) {
|
SafeCopyN(num - i, d, buf + i, keys + i);
|
||||||
keys[i] = buf[i];
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ Partition
|
// ------------------------------ Partition
|
||||||
|
@ -268,15 +238,35 @@ HWY_NOINLINE void PartitionToMultipleOfUnroll(D d, Traits st,
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
HWY_INLINE void StoreLeftRight(D d, Traits st, const Vec<D> v,
|
HWY_INLINE void StoreLeftRight(D d, Traits st, const Vec<D> v,
|
||||||
const Vec<D> pivot, T* HWY_RESTRICT keys,
|
const Vec<D> pivot, T* HWY_RESTRICT keys,
|
||||||
size_t& writeL, size_t& writeR) {
|
size_t& writeL, size_t& remaining) {
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
const auto comp = st.Compare(d, pivot, v);
|
const auto comp = st.Compare(d, pivot, v);
|
||||||
const size_t num_left = CompressBlendedStore(v, Not(comp), d, keys + writeL);
|
|
||||||
|
remaining -= N;
|
||||||
|
if (hwy::HWY_NAMESPACE::CompressIsPartition<T>::value ||
|
||||||
|
(HWY_MAX_BYTES == 16 && st.Is128())) {
|
||||||
|
// Non-native Compress (e.g. AVX2): we are able to partition a vector using
|
||||||
|
// a single Compress+two StoreU instead of two Compress[Blended]Store. The
|
||||||
|
// latter are more expensive. Because we store entire vectors, the contents
|
||||||
|
// between the updated writeL and writeR are ignored and will be overwritten
|
||||||
|
// by subsequent calls. This works because writeL and writeR are at least
|
||||||
|
// two vectors apart.
|
||||||
|
const auto lr = st.CompressKeys(v, comp);
|
||||||
|
const size_t num_left = N - CountTrue(d, comp);
|
||||||
|
StoreU(lr, d, keys + writeL);
|
||||||
|
// Now write the right-side elements (if any), such that the previous writeR
|
||||||
|
// is one past the end of the newly written right elements, then advance.
|
||||||
|
StoreU(lr, d, keys + remaining + writeL);
|
||||||
|
writeL += num_left;
|
||||||
|
} else {
|
||||||
|
// Native Compress[Store] (e.g. AVX3), which only keep the left or right
|
||||||
|
// side, not both, hence we require two calls.
|
||||||
|
const size_t num_left = CompressStore(v, Not(comp), d, keys + writeL);
|
||||||
writeL += num_left;
|
writeL += num_left;
|
||||||
|
|
||||||
writeR -= (N - num_left);
|
(void)CompressBlendedStore(v, comp, d, keys + remaining + writeL);
|
||||||
(void)CompressBlendedStore(v, comp, d, keys + writeR);
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
|
@ -284,11 +274,11 @@ HWY_INLINE void StoreLeftRight4(D d, Traits st, const Vec<D> v0,
|
||||||
const Vec<D> v1, const Vec<D> v2,
|
const Vec<D> v1, const Vec<D> v2,
|
||||||
const Vec<D> v3, const Vec<D> pivot,
|
const Vec<D> v3, const Vec<D> pivot,
|
||||||
T* HWY_RESTRICT keys, size_t& writeL,
|
T* HWY_RESTRICT keys, size_t& writeL,
|
||||||
size_t& writeR) {
|
size_t& remaining) {
|
||||||
StoreLeftRight(d, st, v0, pivot, keys, writeL, writeR);
|
StoreLeftRight(d, st, v0, pivot, keys, writeL, remaining);
|
||||||
StoreLeftRight(d, st, v1, pivot, keys, writeL, writeR);
|
StoreLeftRight(d, st, v1, pivot, keys, writeL, remaining);
|
||||||
StoreLeftRight(d, st, v2, pivot, keys, writeL, writeR);
|
StoreLeftRight(d, st, v2, pivot, keys, writeL, remaining);
|
||||||
StoreLeftRight(d, st, v3, pivot, keys, writeL, writeR);
|
StoreLeftRight(d, st, v3, pivot, keys, writeL, remaining);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Moves "<= pivot" keys to the front, and others to the back. pivot is
|
// Moves "<= pivot" keys to the front, and others to the back. pivot is
|
||||||
|
@ -313,9 +303,39 @@ HWY_NOINLINE size_t Partition(D d, Traits st, T* HWY_RESTRICT keys, size_t left,
|
||||||
PartitionToMultipleOfUnroll(d, st, keys, left, right, pivot, buf);
|
PartitionToMultipleOfUnroll(d, st, keys, left, right, pivot, buf);
|
||||||
constexpr size_t kUnroll = Constants::kPartitionUnroll;
|
constexpr size_t kUnroll = Constants::kPartitionUnroll;
|
||||||
|
|
||||||
// Invariant: [left, writeL) and [writeR, right) are already partitioned.
|
// Partition splits the vector into 3 sections, left to right: Elements
|
||||||
|
// smaller or equal to the pivot, unpartitioned elements and elements larger
|
||||||
|
// than the pivot. To write elements unconditionally on the loop body without
|
||||||
|
// overwriting existing data, we maintain two regions of the loop where all
|
||||||
|
// elements have been copied elsewhere (e.g. vector registers.). I call these
|
||||||
|
// bufferL and bufferR, for left and right respectively.
|
||||||
|
//
|
||||||
|
// These regions are tracked by the indices (writeL, writeR, left, right) as
|
||||||
|
// presented in the diagram below.
|
||||||
|
//
|
||||||
|
// writeL writeR
|
||||||
|
// \/ \/
|
||||||
|
// | <= pivot | bufferL | unpartitioned | bufferR | > pivot |
|
||||||
|
// \/ \/
|
||||||
|
// left right
|
||||||
|
//
|
||||||
|
// In the main loop body below we choose a side, load some elements out of the
|
||||||
|
// vector and move either `left` or `right`. Next we call into StoreLeftRight
|
||||||
|
// to partition the data, and the partitioned elements will be written either
|
||||||
|
// to writeR or writeL and the corresponding index will be moved accordingly.
|
||||||
|
//
|
||||||
|
// Note that writeR is not explicitly tracked as an optimization for platforms
|
||||||
|
// with conditional operations. Instead we track writeL and the number of
|
||||||
|
// elements left to process (`remaining`). From the diagram above we can see
|
||||||
|
// that:
|
||||||
|
// writeR - writeL = remaining => writeR = remaining + writeL
|
||||||
|
//
|
||||||
|
// Tracking `remaining` is advantageous because each iteration reduces the
|
||||||
|
// number of unpartitioned elements by a fixed amount, so we can compute
|
||||||
|
// `remaining` without data dependencies.
|
||||||
|
//
|
||||||
size_t writeL = left;
|
size_t writeL = left;
|
||||||
size_t writeR = right;
|
size_t remaining = right - left;
|
||||||
|
|
||||||
const size_t num = right - left;
|
const size_t num = right - left;
|
||||||
// Cannot load if there were fewer than 2 * kUnroll * N.
|
// Cannot load if there were fewer than 2 * kUnroll * N.
|
||||||
|
@ -339,12 +359,33 @@ HWY_NOINLINE size_t Partition(D d, Traits st, T* HWY_RESTRICT keys, size_t left,
|
||||||
while (left != right) {
|
while (left != right) {
|
||||||
V v0, v1, v2, v3;
|
V v0, v1, v2, v3;
|
||||||
|
|
||||||
// Free up capacity for writing by loading from the side that has less.
|
|
||||||
// Data-dependent but branching is faster than forcing branch-free.
|
// Data-dependent but branching is faster than forcing branch-free.
|
||||||
const size_t capacityL = left - writeL;
|
const size_t capacityL = left - writeL;
|
||||||
const size_t capacityR = writeR - right;
|
HWY_DASSERT(capacityL <= num); // >= 0
|
||||||
HWY_DASSERT(capacityL <= num && capacityR <= num); // >= 0
|
// Load data from the end of the vector with less data (front or back).
|
||||||
if (capacityR < capacityL) {
|
// The next paragraphs explain how this works.
|
||||||
|
//
|
||||||
|
// let block_size = (kUnroll * N)
|
||||||
|
// On the loop prelude we load block_size elements from the front of the
|
||||||
|
// vector and an additional block_size elements from the back. On each
|
||||||
|
// iteration k elements are written to the front of the vector and
|
||||||
|
// (block_size - k) to the back.
|
||||||
|
//
|
||||||
|
// This creates a loop invariant where the capacity on the front
|
||||||
|
// (capacityL) and on the back (capacityR) always add to 2 * block_size.
|
||||||
|
// In other words:
|
||||||
|
// capacityL + capacityR = 2 * block_size
|
||||||
|
// capacityR = 2 * block_size - capacityL
|
||||||
|
//
|
||||||
|
// This means that:
|
||||||
|
// capacityL < capacityR <=>
|
||||||
|
// capacityL < 2 * block_size - capacityL <=>
|
||||||
|
// 2 * capacityL < 2 * block_size <=>
|
||||||
|
// capacityL < block_size
|
||||||
|
//
|
||||||
|
// Thus the check on the next line is equivalent to capacityL > capacityR.
|
||||||
|
//
|
||||||
|
if (kUnroll * N < capacityL) {
|
||||||
right -= kUnroll * N;
|
right -= kUnroll * N;
|
||||||
v0 = LoadU(d, keys + right + 0 * N);
|
v0 = LoadU(d, keys + right + 0 * N);
|
||||||
v1 = LoadU(d, keys + right + 1 * N);
|
v1 = LoadU(d, keys + right + 1 * N);
|
||||||
|
@ -360,16 +401,16 @@ HWY_NOINLINE size_t Partition(D d, Traits st, T* HWY_RESTRICT keys, size_t left,
|
||||||
hwy::Prefetch(keys + left + 3 * kUnroll * N);
|
hwy::Prefetch(keys + left + 3 * kUnroll * N);
|
||||||
}
|
}
|
||||||
|
|
||||||
StoreLeftRight4(d, st, v0, v1, v2, v3, pivot, keys, writeL, writeR);
|
StoreLeftRight4(d, st, v0, v1, v2, v3, pivot, keys, writeL, remaining);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Now finish writing the initial left/right to the middle.
|
// Now finish writing the initial left/right to the middle.
|
||||||
StoreLeftRight4(d, st, vL0, vL1, vL2, vL3, pivot, keys, writeL, writeR);
|
StoreLeftRight4(d, st, vL0, vL1, vL2, vL3, pivot, keys, writeL, remaining);
|
||||||
StoreLeftRight4(d, st, vR0, vR1, vR2, vR3, pivot, keys, writeL, writeR);
|
StoreLeftRight4(d, st, vR0, vR1, vR2, vR3, pivot, keys, writeL, remaining);
|
||||||
}
|
}
|
||||||
|
|
||||||
// We have partitioned [left, right) such that writeL is the boundary.
|
// We have partitioned [left, right) such that writeL is the boundary.
|
||||||
HWY_DASSERT(writeL == writeR);
|
HWY_DASSERT(remaining == 0);
|
||||||
// Make space for inserting vlast: move up to N of the first right-side keys
|
// Make space for inserting vlast: move up to N of the first right-side keys
|
||||||
// into the unused space starting at last. If we have fewer, ensure they are
|
// into the unused space starting at last. If we have fewer, ensure they are
|
||||||
// the last items in that vector by subtracting from the *load* address,
|
// the last items in that vector by subtracting from the *load* address,
|
||||||
|
@ -406,41 +447,6 @@ HWY_INLINE V MedianOf3(Traits st, V v0, V v1, V v2) {
|
||||||
return v1;
|
return v1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Replaces triplets with their median and recurses until less than 3 keys
|
|
||||||
// remain. Ignores leftover values (non-whole triplets)!
|
|
||||||
template <class D, class Traits, typename T>
|
|
||||||
Vec<D> RecursiveMedianOf3(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
|
||||||
T* HWY_RESTRICT buf) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
constexpr size_t N1 = st.LanesPerKey();
|
|
||||||
|
|
||||||
if (num < 3 * N1) return st.SetKey(d, keys);
|
|
||||||
|
|
||||||
size_t read = 0;
|
|
||||||
size_t written = 0;
|
|
||||||
|
|
||||||
// Triplets of vectors
|
|
||||||
for (; read + 3 * N <= num; read += 3 * N) {
|
|
||||||
const auto v0 = Load(d, keys + read + 0 * N);
|
|
||||||
const auto v1 = Load(d, keys + read + 1 * N);
|
|
||||||
const auto v2 = Load(d, keys + read + 2 * N);
|
|
||||||
Store(MedianOf3(st, v0, v1, v2), d, buf + written);
|
|
||||||
written += N;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Triplets of keys
|
|
||||||
for (; read + 3 * N1 <= num; read += 3 * N1) {
|
|
||||||
const auto v0 = st.SetKey(d, keys + read + 0 * N1);
|
|
||||||
const auto v1 = st.SetKey(d, keys + read + 1 * N1);
|
|
||||||
const auto v2 = st.SetKey(d, keys + read + 2 * N1);
|
|
||||||
StoreU(MedianOf3(st, v0, v1, v2), d, buf + written);
|
|
||||||
written += N1;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Tail recursion; swap buffers
|
|
||||||
return RecursiveMedianOf3(d, st, buf, written, keys);
|
|
||||||
}
|
|
||||||
|
|
||||||
#if VQSORT_SECURE_RNG
|
#if VQSORT_SECURE_RNG
|
||||||
using Generator = absl::BitGen;
|
using Generator = absl::BitGen;
|
||||||
#else
|
#else
|
||||||
|
@ -453,6 +459,11 @@ class Generator {
|
||||||
k_ = 1; // stream index: must be odd
|
k_ = 1; // stream index: must be odd
|
||||||
}
|
}
|
||||||
|
|
||||||
|
explicit Generator(uint64_t seed) {
|
||||||
|
a_ = b_ = w_ = seed;
|
||||||
|
k_ = 1;
|
||||||
|
}
|
||||||
|
|
||||||
uint64_t operator()() {
|
uint64_t operator()() {
|
||||||
const uint64_t b = b_;
|
const uint64_t b = b_;
|
||||||
w_ += k_;
|
w_ += k_;
|
||||||
|
@ -481,9 +492,183 @@ HWY_INLINE size_t RandomChunkIndex(const uint32_t num_chunks, uint32_t bits) {
|
||||||
return static_cast<size_t>(chunk_index);
|
return static_cast<size_t>(chunk_index);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class Traits, typename T>
|
||||||
|
HWY_INLINE void SortSamples(Traits st, T* HWY_RESTRICT buf) {
|
||||||
|
// buf contains 192 bytes, so 16 128-bit vectors are necessary and sufficient.
|
||||||
|
constexpr size_t kSampleLanes = 3 * 64 / sizeof(T);
|
||||||
|
const CappedTag<T, 16 / sizeof(T)> d128;
|
||||||
|
const size_t N128 = Lanes(d128);
|
||||||
|
constexpr size_t kCols = HWY_MIN(16 / sizeof(T), Constants::kMaxCols);
|
||||||
|
constexpr size_t kBytes = kCols * Constants::kMaxRows * sizeof(T);
|
||||||
|
static_assert(192 <= kBytes, "");
|
||||||
|
// Fill with padding - last in sort order.
|
||||||
|
const auto kPadding = st.LastValue(d128);
|
||||||
|
// Initialize an extra vector because SortingNetwork loads full vectors,
|
||||||
|
// which may exceed cols*kMaxRows.
|
||||||
|
for (size_t i = kSampleLanes; i <= kBytes / sizeof(T); i += N128) {
|
||||||
|
StoreU(kPadding, d128, buf + i);
|
||||||
|
}
|
||||||
|
|
||||||
|
SortingNetwork(st, buf, kCols);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class Traits, typename T>
|
||||||
|
HWY_INLINE size_t PivotRank(Traits st, T* HWY_RESTRICT buf) {
|
||||||
|
constexpr size_t kSampleLanes = 3 * 64 / sizeof(T);
|
||||||
|
constexpr size_t N1 = st.LanesPerKey();
|
||||||
|
|
||||||
|
constexpr size_t kRankMid = kSampleLanes / 2;
|
||||||
|
static_assert(kRankMid % N1 == 0, "Mid is not an aligned key");
|
||||||
|
|
||||||
|
// Find the previous value not equal to the median.
|
||||||
|
size_t rank_prev = kRankMid - N1;
|
||||||
|
for (; st.Equal1(buf + rank_prev, buf + kRankMid); rank_prev -= N1) {
|
||||||
|
// All previous samples are equal to the median.
|
||||||
|
if (rank_prev == 0) return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t rank_next = rank_prev + N1;
|
||||||
|
for (; st.Equal1(buf + rank_next, buf + kRankMid); rank_next += N1) {
|
||||||
|
// The median is also the largest sample. If it is also the largest key,
|
||||||
|
// we'd end up with an empty right partition, so choose the previous key.
|
||||||
|
if (rank_next == kSampleLanes - N1) return rank_prev;
|
||||||
|
}
|
||||||
|
|
||||||
|
// If we choose the median as pivot, the ratio of keys ending in the left
|
||||||
|
// partition will likely be rank_next/kSampleLanes (if the sample is
|
||||||
|
// representative). This is because equal-to-pivot values also land in the
|
||||||
|
// left - it's infeasible to do an in-place vectorized 3-way partition.
|
||||||
|
// Check whether prev would lead to a more balanced partition.
|
||||||
|
const size_t excess_if_median = rank_next - kRankMid;
|
||||||
|
const size_t excess_if_prev = kRankMid - rank_prev;
|
||||||
|
return excess_if_median < excess_if_prev ? kRankMid : rank_prev;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
// Compute exact min/max.
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
HWY_NOINLINE Vec<D> ChoosePivot(D d, Traits st, T* HWY_RESTRICT keys,
|
HWY_NOINLINE void ScanMinMax(D d, Traits st, const T* HWY_RESTRICT keys,
|
||||||
const size_t begin, const size_t end,
|
size_t num, T* HWY_RESTRICT buf, Vec<D>& first,
|
||||||
|
Vec<D>& last) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
|
first = st.LastValue(d);
|
||||||
|
last = st.FirstValue(d);
|
||||||
|
|
||||||
|
size_t i = 0;
|
||||||
|
for (; i + N <= num; i += N) {
|
||||||
|
const Vec<D> v = LoadU(d, keys + i);
|
||||||
|
first = st.First(d, v, first);
|
||||||
|
last = st.Last(d, v, last);
|
||||||
|
}
|
||||||
|
if (HWY_LIKELY(i != num)) {
|
||||||
|
HWY_DASSERT(num >= N); // See HandleSpecialCases
|
||||||
|
const Vec<D> v = LoadU(d, keys + num - N);
|
||||||
|
first = st.First(d, v, first);
|
||||||
|
last = st.Last(d, v, last);
|
||||||
|
}
|
||||||
|
|
||||||
|
first = st.FirstOfLanes(d, first, buf);
|
||||||
|
last = st.LastOfLanes(d, last, buf);
|
||||||
|
}
|
||||||
|
#endif // VQSORT_PRINT
|
||||||
|
|
||||||
|
template <class D, class Traits, typename T>
|
||||||
|
HWY_INLINE bool ScanEqual(D d, Traits st, const T* HWY_RESTRICT keys,
|
||||||
|
size_t num) {
|
||||||
|
using V = Vec<decltype(d)>;
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
HWY_DASSERT(num >= N); // See HandleSpecialCases
|
||||||
|
const V reference = st.SetKey(d, keys);
|
||||||
|
const V zero = Zero(d);
|
||||||
|
// Sticky bits registering any difference between `keys` and the first key.
|
||||||
|
// We use vector XOR because it may be cheaper than comparisons, especially
|
||||||
|
// for 128-bit. 2x unrolled for more ILP.
|
||||||
|
V diff0 = zero;
|
||||||
|
V diff1 = zero;
|
||||||
|
|
||||||
|
// We want to stop once a difference has been found, but without slowing down
|
||||||
|
// the loop by comparing during each iteration. The compromise is to compare
|
||||||
|
// after a 'group', which consists of kLoops times two vectors.
|
||||||
|
constexpr size_t kLoops = 4;
|
||||||
|
const size_t lanes_per_group = kLoops * 2 * N;
|
||||||
|
size_t i = 0;
|
||||||
|
for (; i + lanes_per_group <= num; i += lanes_per_group) {
|
||||||
|
for (size_t loop = 0; loop < kLoops; ++loop) {
|
||||||
|
const V v0 = LoadU(d, keys + i + loop * 2 * N);
|
||||||
|
const V v1 = LoadU(d, keys + i + loop * 2 * N + N);
|
||||||
|
// TODO(janwas): ternlog
|
||||||
|
diff0 = Or(diff0, Xor(v0, reference));
|
||||||
|
diff1 = Or(diff1, Xor(v1, reference));
|
||||||
|
}
|
||||||
|
diff0 = Or(diff0, diff1);
|
||||||
|
if (!AllTrue(d, Eq(diff0, zero))) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Whole vectors, no unrolling
|
||||||
|
for (; i + N <= num; i += N) {
|
||||||
|
const V v0 = LoadU(d, keys + i);
|
||||||
|
// TODO(janwas): ternlog
|
||||||
|
diff0 = Or(diff0, Xor(v0, reference));
|
||||||
|
if (!AllTrue(d, Eq(diff0, zero))) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// If there are remainders, re-check the last whole vector.
|
||||||
|
if (HWY_LIKELY(i != num)) {
|
||||||
|
const V v0 = LoadU(d, keys + num - N);
|
||||||
|
// TODO(janwas): ternlog
|
||||||
|
diff0 = Or(diff0, Xor(v0, reference));
|
||||||
|
if (!AllTrue(d, Eq(diff0, zero))) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns key prior to reference in sort order.
|
||||||
|
template <class D, class Traits, typename T>
|
||||||
|
HWY_INLINE Vec<D> ScanForPrev(D d, Traits st, const T* HWY_RESTRICT keys,
|
||||||
|
size_t num, Vec<D> reference,
|
||||||
|
T* HWY_RESTRICT buf) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
HWY_DASSERT(num >= N); // See HandleSpecialCases
|
||||||
|
|
||||||
|
Vec<D> prev = st.FirstValue(d);
|
||||||
|
Mask<D> any_found = st.Compare(d, prev, prev); // false
|
||||||
|
|
||||||
|
size_t i = 0;
|
||||||
|
// Whole vectors, no unrolling
|
||||||
|
for (; i + N <= num; i += N) {
|
||||||
|
const Vec<D> curr = LoadU(d, keys + i);
|
||||||
|
const auto is_before = st.Compare(d, curr, reference);
|
||||||
|
any_found = Or(any_found, is_before);
|
||||||
|
prev = IfThenElse(is_before, st.Last(d, prev, curr), prev);
|
||||||
|
}
|
||||||
|
// If there are remainders, re-check the last whole vector.
|
||||||
|
if (HWY_LIKELY(i != num)) {
|
||||||
|
const Vec<D> curr = LoadU(d, keys + num - N);
|
||||||
|
const auto is_before = st.Compare(d, curr, reference);
|
||||||
|
any_found = Or(any_found, is_before);
|
||||||
|
prev = IfThenElse(is_before, st.Last(d, prev, curr), prev);
|
||||||
|
}
|
||||||
|
|
||||||
|
const Vec<D> candidate = st.LastOfLanes(d, prev, buf);
|
||||||
|
// If we didn't find any key less than reference, we're still stuck with
|
||||||
|
// FirstValue; replace that with reference. (We cannot compare directly to
|
||||||
|
// FirstValue because that might be the desired value of prev.)
|
||||||
|
return IfThenElse(any_found, candidate, reference);
|
||||||
|
}
|
||||||
|
|
||||||
|
enum class PivotResult {
|
||||||
|
kNormal, // use partition
|
||||||
|
kAllEqual, // already done
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class D, class Traits, typename T>
|
||||||
|
HWY_INLINE void DrawSamples(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
T* HWY_RESTRICT buf, Generator& rng) {
|
T* HWY_RESTRICT buf, Generator& rng) {
|
||||||
using V = decltype(Zero(d));
|
using V = decltype(Zero(d));
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
|
@ -491,9 +676,6 @@ HWY_NOINLINE Vec<D> ChoosePivot(D d, Traits st, T* HWY_RESTRICT keys,
|
||||||
// Power of two
|
// Power of two
|
||||||
const size_t lanes_per_chunk = Constants::LanesPerChunk(sizeof(T), N);
|
const size_t lanes_per_chunk = Constants::LanesPerChunk(sizeof(T), N);
|
||||||
|
|
||||||
keys += begin;
|
|
||||||
size_t num = end - begin;
|
|
||||||
|
|
||||||
// Align start of keys to chunks. We always have at least 2 chunks because the
|
// Align start of keys to chunks. We always have at least 2 chunks because the
|
||||||
// base case would have handled anything up to 16 vectors, i.e. >= 4 chunks.
|
// base case would have handled anything up to 16 vectors, i.e. >= 4 chunks.
|
||||||
HWY_DASSERT(num >= 2 * lanes_per_chunk);
|
HWY_DASSERT(num >= 2 * lanes_per_chunk);
|
||||||
|
@ -548,49 +730,92 @@ HWY_NOINLINE Vec<D> ChoosePivot(D d, Traits st, T* HWY_RESTRICT keys,
|
||||||
const V medians2 = MedianOf3(st, v6, v7, v8);
|
const V medians2 = MedianOf3(st, v6, v7, v8);
|
||||||
Store(medians2, d, buf + i + lanes_per_chunk * 2);
|
Store(medians2, d, buf + i + lanes_per_chunk * 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
return RecursiveMedianOf3(d, st, buf, 3 * lanes_per_chunk,
|
|
||||||
buf + 3 * lanes_per_chunk);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Compute exact min/max to detect all-equal partitions. Only called after a
|
// Returns pivot, which is never the largest key (thus the right partition will
|
||||||
// degenerate Partition (none in the right partition).
|
// never be empty).
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
HWY_NOINLINE void ScanMinMax(D d, Traits st, const T* HWY_RESTRICT keys,
|
HWY_NOINLINE Vec<D> ChoosePivot(D d, Traits st, T* HWY_RESTRICT keys,
|
||||||
size_t num, T* HWY_RESTRICT buf, Vec<D>& first,
|
const size_t begin, const size_t end,
|
||||||
Vec<D>& last) {
|
T* HWY_RESTRICT buf, Generator& rng,
|
||||||
|
PivotResult& result) {
|
||||||
|
using V = decltype(Zero(d));
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
|
|
||||||
first = st.LastValue(d);
|
constexpr size_t kSampleLanes = 3 * 64 / sizeof(T);
|
||||||
last = st.FirstValue(d);
|
constexpr size_t N1 = st.LanesPerKey();
|
||||||
|
|
||||||
size_t i = 0;
|
const size_t num = end - begin;
|
||||||
for (; i + N <= num; i += N) {
|
#if VQSORT_PRINT
|
||||||
const Vec<D> v = LoadU(d, keys + i);
|
fprintf(stderr, "\nChoosePivot num %zu:\n", num);
|
||||||
first = st.First(d, v, first);
|
#endif
|
||||||
last = st.Last(d, v, last);
|
DrawSamples(d, st, keys + begin, num, buf, rng);
|
||||||
|
|
||||||
|
SortSamples(st, buf);
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
for (size_t i = 0; i < kSampleLanes; i += N) {
|
||||||
|
Print(d, "", Load(d, buf + i), 0, N);
|
||||||
}
|
}
|
||||||
if (HWY_LIKELY(i != num)) {
|
#endif
|
||||||
HWY_DASSERT(num >= N); // See HandleSpecialCases
|
|
||||||
const Vec<D> v = LoadU(d, keys + num - N);
|
// All samples are equal.
|
||||||
first = st.First(d, v, first);
|
if (st.Equal1(buf, buf + kSampleLanes - N1)) {
|
||||||
last = st.Last(d, v, last);
|
const bool all_eq = ScanEqual(d, st, keys + begin, num);
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
fprintf(stderr, "Pivot num=%zu all eq samples, keys also: %d\n", num,
|
||||||
|
all_eq);
|
||||||
|
#endif
|
||||||
|
if (all_eq) {
|
||||||
|
result = PivotResult::kAllEqual;
|
||||||
|
return Zero(d);
|
||||||
}
|
}
|
||||||
|
|
||||||
first = st.FirstOfLanes(d, first, buf);
|
// If the sample is indeed the most common key and it is the largest, then
|
||||||
last = st.LastOfLanes(d, last, buf);
|
// the right partition will be empty. Prevent this by replacing the pivot
|
||||||
|
// with the previous key in sort order. By contrast, selecting the first key
|
||||||
|
// in sort order would guarantee (minimal) progress. We instead do a full
|
||||||
|
// scan to maximize load balance in case there are numerous keys that
|
||||||
|
// precede the most common key.
|
||||||
|
result = PivotResult::kNormal;
|
||||||
|
const V reference = st.SetKey(d, buf);
|
||||||
|
const V pivot = ScanForPrev(d, st, keys + begin, num, reference, buf);
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
Print(d, "PREV pivot", pivot, 0, st.LanesPerKey());
|
||||||
|
#endif
|
||||||
|
return pivot;
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t pivot_rank = PivotRank(st, buf);
|
||||||
|
const Vec<D> pivot = st.SetKey(d, buf + pivot_rank);
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
fprintf(stderr, " Pivot rank %zu = %.0f\n", pivot_rank,
|
||||||
|
static_cast<double>(GetLane(pivot)));
|
||||||
|
#endif
|
||||||
|
result = PivotResult::kNormal;
|
||||||
|
return pivot;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
void Recurse(D d, Traits st, T* HWY_RESTRICT keys, const size_t begin,
|
void Recurse(D d, Traits st, T* HWY_RESTRICT keys, T* HWY_RESTRICT keys_end,
|
||||||
const size_t end, const Vec<D> pivot, T* HWY_RESTRICT buf,
|
const size_t begin, const size_t end, const Vec<D> pivot,
|
||||||
Generator& rng, size_t remaining_levels) {
|
T* HWY_RESTRICT buf, Generator& rng, size_t remaining_levels) {
|
||||||
HWY_DASSERT(begin + 1 < end);
|
HWY_DASSERT(begin + 1 < end);
|
||||||
const size_t num = end - begin; // >= 2
|
const size_t num = end - begin; // >= 2
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
fprintf(stderr, "- Recurse remaining %zu [%zu %zu) len %zu\n",
|
||||||
|
remaining_levels, begin, end, num);
|
||||||
|
Vec<D> first, last;
|
||||||
|
ScanMinMax(d, st, keys + begin, num, buf, first, last);
|
||||||
|
Print(d, "first", first, 0, st.LanesPerKey());
|
||||||
|
Print(d, "last", last, 0, st.LanesPerKey());
|
||||||
|
#endif
|
||||||
|
|
||||||
// Too many degenerate partitions. This is extremely unlikely to happen
|
// Too many recursions. This is unlikely to happen because we select pivots
|
||||||
// because we select pivots from large (though still O(1)) samples.
|
// from large (though still O(1)) samples.
|
||||||
if (HWY_UNLIKELY(remaining_levels == 0)) {
|
if (HWY_UNLIKELY(remaining_levels == 0)) {
|
||||||
|
#if VQSORT_PRINT
|
||||||
|
fprintf(stderr, "HeapSort reached, size=%zu\n", num);
|
||||||
|
#endif
|
||||||
HeapSort(st, keys + begin, num); // Slow but N*logN.
|
HeapSort(st, keys + begin, num); // Slow but N*logN.
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
@ -604,36 +829,32 @@ void Recurse(D d, Traits st, T* HWY_RESTRICT keys, const size_t begin,
|
||||||
const ptrdiff_t num_right =
|
const ptrdiff_t num_right =
|
||||||
static_cast<ptrdiff_t>(end) - static_cast<ptrdiff_t>(bound);
|
static_cast<ptrdiff_t>(end) - static_cast<ptrdiff_t>(bound);
|
||||||
|
|
||||||
// Check for degenerate partitions (i.e. Partition did not move any keys):
|
// ChoosePivot ensures pivot != largest key, so this should never happen.
|
||||||
if (HWY_UNLIKELY(num_right == 0)) {
|
HWY_ASSERT(num_right != 0);
|
||||||
// Because the pivot is one of the keys, it must have been equal to the
|
|
||||||
// first or last key in sort order. Scan for the actual min/max:
|
|
||||||
// passing the current pivot as the new bound is insufficient because one of
|
|
||||||
// the partitions might not actually include that key.
|
|
||||||
Vec<D> first, last;
|
|
||||||
ScanMinMax(d, st, keys + begin, num, buf, first, last);
|
|
||||||
if (AllTrue(d, Eq(first, last))) return;
|
|
||||||
|
|
||||||
// Separate recursion to make sure that we don't pick `last` as the
|
|
||||||
// pivot - that would again lead to a degenerate partition.
|
|
||||||
Recurse(d, st, keys, begin, end, first, buf, rng, remaining_levels - 1);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (HWY_UNLIKELY(num_left <= base_case_num)) {
|
if (HWY_UNLIKELY(num_left <= base_case_num)) {
|
||||||
BaseCase(d, st, keys + begin, static_cast<size_t>(num_left), buf);
|
BaseCase(d, st, keys + begin, keys_end, static_cast<size_t>(num_left), buf);
|
||||||
} else {
|
} else {
|
||||||
const Vec<D> next_pivot = ChoosePivot(d, st, keys, begin, bound, buf, rng);
|
PivotResult result;
|
||||||
Recurse(d, st, keys, begin, bound, next_pivot, buf, rng,
|
const Vec<D> next_pivot =
|
||||||
|
ChoosePivot(d, st, keys, begin, bound, buf, rng, result);
|
||||||
|
if (result != PivotResult::kAllEqual) {
|
||||||
|
Recurse(d, st, keys, keys_end, begin, bound, next_pivot, buf, rng,
|
||||||
remaining_levels - 1);
|
remaining_levels - 1);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
if (HWY_UNLIKELY(num_right <= base_case_num)) {
|
if (HWY_UNLIKELY(num_right <= base_case_num)) {
|
||||||
BaseCase(d, st, keys + bound, static_cast<size_t>(num_right), buf);
|
BaseCase(d, st, keys + bound, keys_end, static_cast<size_t>(num_right),
|
||||||
|
buf);
|
||||||
} else {
|
} else {
|
||||||
const Vec<D> next_pivot = ChoosePivot(d, st, keys, bound, end, buf, rng);
|
PivotResult result;
|
||||||
Recurse(d, st, keys, bound, end, next_pivot, buf, rng,
|
const Vec<D> next_pivot =
|
||||||
|
ChoosePivot(d, st, keys, bound, end, buf, rng, result);
|
||||||
|
if (result != PivotResult::kAllEqual) {
|
||||||
|
Recurse(d, st, keys, keys_end, bound, end, next_pivot, buf, rng,
|
||||||
remaining_levels - 1);
|
remaining_levels - 1);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns true if sorting is finished.
|
// Returns true if sorting is finished.
|
||||||
|
@ -646,10 +867,11 @@ bool HandleSpecialCases(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
// 128-bit keys require vectors with at least two u64 lanes, which is always
|
// 128-bit keys require vectors with at least two u64 lanes, which is always
|
||||||
// the case unless `d` requests partial vectors (e.g. fraction = 1/2) AND the
|
// the case unless `d` requests partial vectors (e.g. fraction = 1/2) AND the
|
||||||
// hardware vector width is less than 128bit / fraction.
|
// hardware vector width is less than 128bit / fraction.
|
||||||
const bool partial_128 = N < 2 && st.Is128();
|
const bool partial_128 = !IsFull(d) && N < 2 && st.Is128();
|
||||||
// Partition assumes its input is at least two vectors. If vectors are huge,
|
// Partition assumes its input is at least two vectors. If vectors are huge,
|
||||||
// base_case_num may actually be smaller. If so, which is only possible on
|
// base_case_num may actually be smaller. If so, which is only possible on
|
||||||
// RVV, pass a capped or partial d (LMUL < 1).
|
// RVV, pass a capped or partial d (LMUL < 1). Use HWY_MAX_BYTES instead of
|
||||||
|
// HWY_LANES to account for the largest possible LMUL.
|
||||||
constexpr bool kPotentiallyHuge =
|
constexpr bool kPotentiallyHuge =
|
||||||
HWY_MAX_BYTES / sizeof(T) > Constants::kMaxRows * Constants::kMaxCols;
|
HWY_MAX_BYTES / sizeof(T) > Constants::kMaxRows * Constants::kMaxCols;
|
||||||
const bool huge_vec = kPotentiallyHuge && (2 * N > base_case_num);
|
const bool huge_vec = kPotentiallyHuge && (2 * N > base_case_num);
|
||||||
|
@ -661,7 +883,7 @@ bool HandleSpecialCases(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
|
|
||||||
// Small arrays: use sorting network, no need for other checks.
|
// Small arrays: use sorting network, no need for other checks.
|
||||||
if (HWY_UNLIKELY(num <= base_case_num)) {
|
if (HWY_UNLIKELY(num <= base_case_num)) {
|
||||||
BaseCase(d, st, keys, num, buf);
|
BaseCase(d, st, keys, keys + num, num, buf);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -671,7 +893,7 @@ bool HandleSpecialCases(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
return false; // not finished sorting
|
return false; // not finished sorting
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif // HWY_TARGET != HWY_SCALAR
|
#endif // VQSORT_ENABLED
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
|
|
||||||
// Sorts `keys[0..num-1]` according to the order defined by `st.Compare`.
|
// Sorts `keys[0..num-1]` according to the order defined by `st.Compare`.
|
||||||
|
@ -683,17 +905,26 @@ bool HandleSpecialCases(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
// sampling only from the first 256 GiB.
|
// sampling only from the first 256 GiB.
|
||||||
//
|
//
|
||||||
// `d` is typically SortTag<T> (chooses between full and partial vectors).
|
// `d` is typically SortTag<T> (chooses between full and partial vectors).
|
||||||
// `st` is SharedTraits<{LaneTraits|Traits128}<Order*>>. This abstraction layer
|
// `st` is SharedTraits<Traits*<Order*>>. This abstraction layer bridges
|
||||||
// bridges differences in sort order and single-lane vs 128-bit keys.
|
// differences in sort order and single-lane vs 128-bit keys.
|
||||||
template <class D, class Traits, typename T>
|
template <class D, class Traits, typename T>
|
||||||
void Sort(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
void Sort(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
T* HWY_RESTRICT buf) {
|
T* HWY_RESTRICT buf) {
|
||||||
#if HWY_TARGET == HWY_SCALAR
|
#if VQSORT_PRINT
|
||||||
(void)d;
|
fprintf(stderr, "=============== Sort num %zu\n", num);
|
||||||
(void)buf;
|
#endif
|
||||||
// PERFORMANCE WARNING: vqsort is not enabled for the non-SIMD target
|
|
||||||
return detail::HeapSort(st, keys, num);
|
#if VQSORT_ENABLED || HWY_IDE
|
||||||
#else
|
#if !HWY_HAVE_SCALABLE
|
||||||
|
// On targets with fixed-size vectors, avoid _using_ the allocated memory.
|
||||||
|
// We avoid (potentially expensive for small input sizes) allocations on
|
||||||
|
// platforms where no targets are scalable. For 512-bit vectors, this fits on
|
||||||
|
// the stack (several KiB).
|
||||||
|
HWY_ALIGN T storage[SortConstants::BufNum<T>(HWY_LANES(T))] = {};
|
||||||
|
static_assert(sizeof(storage) <= 8192, "Unexpectedly large, check size");
|
||||||
|
buf = storage;
|
||||||
|
#endif // !HWY_HAVE_SCALABLE
|
||||||
|
|
||||||
if (detail::HandleSpecialCases(d, st, keys, num, buf)) return;
|
if (detail::HandleSpecialCases(d, st, keys, num, buf)) return;
|
||||||
|
|
||||||
#if HWY_MAX_BYTES > 64
|
#if HWY_MAX_BYTES > 64
|
||||||
|
@ -705,13 +936,22 @@ void Sort(D d, Traits st, T* HWY_RESTRICT keys, size_t num,
|
||||||
|
|
||||||
// Pulled out of the recursion so we can special-case degenerate partitions.
|
// Pulled out of the recursion so we can special-case degenerate partitions.
|
||||||
detail::Generator rng(keys, num);
|
detail::Generator rng(keys, num);
|
||||||
const Vec<D> pivot = detail::ChoosePivot(d, st, keys, 0, num, buf, rng);
|
detail::PivotResult result;
|
||||||
|
const Vec<D> pivot =
|
||||||
|
detail::ChoosePivot(d, st, keys, 0, num, buf, rng, result);
|
||||||
|
|
||||||
|
if (result != detail::PivotResult::kAllEqual) {
|
||||||
// Introspection: switch to worst-case N*logN heapsort after this many.
|
// Introspection: switch to worst-case N*logN heapsort after this many.
|
||||||
const size_t max_levels = 2 * hwy::CeilLog2(num) + 4;
|
const size_t max_levels = 2 * hwy::CeilLog2(num) + 4;
|
||||||
|
detail::Recurse(d, st, keys, keys + num, 0, num, pivot, buf, rng,
|
||||||
detail::Recurse(d, st, keys, 0, num, pivot, buf, rng, max_levels);
|
max_levels);
|
||||||
#endif // HWY_TARGET == HWY_SCALAR
|
}
|
||||||
|
#else
|
||||||
|
(void)d;
|
||||||
|
(void)buf;
|
||||||
|
// PERFORMANCE WARNING: vqsort is not enabled for the non-SIMD target
|
||||||
|
return detail::HeapSort(st, keys, num);
|
||||||
|
#endif // VQSORT_ENABLED
|
||||||
}
|
}
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -16,21 +17,64 @@
|
||||||
|
|
||||||
#include <string.h> // memset
|
#include <string.h> // memset
|
||||||
|
|
||||||
#include "hwy/aligned_allocator.h"
|
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/shared-inl.h"
|
#include "hwy/contrib/sort/shared-inl.h"
|
||||||
|
|
||||||
|
// Architectures for which we know HWY_HAVE_SCALABLE == 0. This opts into an
|
||||||
|
// optimization that replaces dynamic allocation with stack storage.
|
||||||
|
#ifndef VQSORT_STACK
|
||||||
|
#if HWY_ARCH_X86 || HWY_ARCH_WASM
|
||||||
|
#define VQSORT_STACK 1
|
||||||
|
#else
|
||||||
|
#define VQSORT_STACK 0
|
||||||
|
#endif
|
||||||
|
#endif // VQSORT_STACK
|
||||||
|
|
||||||
|
#if !VQSORT_STACK
|
||||||
|
#include "hwy/aligned_allocator.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Check if we have sys/random.h. First skip some systems on which the check
|
||||||
|
// itself (features.h) might be problematic.
|
||||||
|
#if defined(ANDROID) || defined(__ANDROID__) || HWY_ARCH_RVV
|
||||||
|
#define VQSORT_GETRANDOM 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if !defined(VQSORT_GETRANDOM) && HWY_OS_LINUX
|
||||||
|
#include <features.h>
|
||||||
|
|
||||||
|
// ---- which libc
|
||||||
|
#if defined(__UCLIBC__)
|
||||||
|
#define VQSORT_GETRANDOM 1 // added Mar 2015, before uclibc-ng 1.0
|
||||||
|
|
||||||
|
#elif defined(__GLIBC__) && defined(__GLIBC_PREREQ)
|
||||||
|
#if __GLIBC_PREREQ(2, 25)
|
||||||
|
#define VQSORT_GETRANDOM 1
|
||||||
|
#else
|
||||||
|
#define VQSORT_GETRANDOM 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#else
|
||||||
|
// Assume MUSL, which has getrandom since 2018. There is no macro to test, see
|
||||||
|
// https://www.openwall.com/lists/musl/2013/03/29/13.
|
||||||
|
#define VQSORT_GETRANDOM 1
|
||||||
|
|
||||||
|
#endif // ---- which libc
|
||||||
|
#endif // linux
|
||||||
|
|
||||||
|
#if !defined(VQSORT_GETRANDOM)
|
||||||
|
#define VQSORT_GETRANDOM 0
|
||||||
|
#endif
|
||||||
|
|
||||||
// Seed source for SFC generator: 1=getrandom, 2=CryptGenRandom
|
// Seed source for SFC generator: 1=getrandom, 2=CryptGenRandom
|
||||||
// (not all Android support the getrandom wrapper)
|
// (not all Android support the getrandom wrapper)
|
||||||
#ifndef VQSORT_SECURE_SEED
|
#ifndef VQSORT_SECURE_SEED
|
||||||
|
|
||||||
#if (defined(linux) || defined(__linux__)) && \
|
#if VQSORT_GETRANDOM
|
||||||
!(defined(ANDROID) || defined(__ANDROID__) || HWY_ARCH_RVV)
|
|
||||||
#define VQSORT_SECURE_SEED 1
|
#define VQSORT_SECURE_SEED 1
|
||||||
#elif defined(_WIN32) || defined(_WIN64)
|
#elif defined(_WIN32) || defined(_WIN64)
|
||||||
#define VQSORT_SECURE_SEED 2
|
#define VQSORT_SECURE_SEED 2
|
||||||
|
@ -47,7 +91,7 @@
|
||||||
#include <sys/random.h>
|
#include <sys/random.h>
|
||||||
#elif VQSORT_SECURE_SEED == 2
|
#elif VQSORT_SECURE_SEED == 2
|
||||||
#include <windows.h>
|
#include <windows.h>
|
||||||
#pragma comment(lib, "Advapi32.lib")
|
#pragma comment(lib, "advapi32.lib")
|
||||||
// Must come after windows.h.
|
// Must come after windows.h.
|
||||||
#include <wincrypt.h>
|
#include <wincrypt.h>
|
||||||
#endif // VQSORT_SECURE_SEED
|
#endif // VQSORT_SECURE_SEED
|
||||||
|
@ -72,40 +116,32 @@ namespace {
|
||||||
HWY_EXPORT(VectorSize);
|
HWY_EXPORT(VectorSize);
|
||||||
HWY_EXPORT(HaveFloat64);
|
HWY_EXPORT(HaveFloat64);
|
||||||
|
|
||||||
HWY_INLINE size_t PivotBufNum(size_t sizeof_t, size_t N) {
|
|
||||||
// 3 chunks of medians, 1 chunk of median medians plus two padding vectors.
|
|
||||||
const size_t lpc = SortConstants::LanesPerChunk(sizeof_t, N);
|
|
||||||
return (3 + 1) * lpc + 2 * N;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
Sorter::Sorter() {
|
Sorter::Sorter() {
|
||||||
|
#if VQSORT_STACK
|
||||||
|
ptr_ = nullptr; // Sort will use stack storage instead
|
||||||
|
#else
|
||||||
// Determine the largest buffer size required for any type by trying them all.
|
// Determine the largest buffer size required for any type by trying them all.
|
||||||
// (The capping of N in BaseCaseNum means that smaller N but larger sizeof_t
|
// (The capping of N in BaseCaseNum means that smaller N but larger sizeof_t
|
||||||
// may require a larger buffer.)
|
// may require a larger buffer.)
|
||||||
const size_t vector_size = HWY_DYNAMIC_DISPATCH(VectorSize)();
|
const size_t vector_size = HWY_DYNAMIC_DISPATCH(VectorSize)();
|
||||||
size_t max_bytes = 0;
|
const size_t max_bytes =
|
||||||
for (size_t sizeof_t :
|
HWY_MAX(HWY_MAX(SortConstants::BufBytes<uint16_t>(vector_size),
|
||||||
{sizeof(uint16_t), sizeof(uint32_t), sizeof(uint64_t)}) {
|
SortConstants::BufBytes<uint32_t>(vector_size)),
|
||||||
const size_t N = vector_size / sizeof_t;
|
SortConstants::BufBytes<uint64_t>(vector_size));
|
||||||
// One extra for padding plus another for full-vector loads.
|
|
||||||
const size_t base_case = SortConstants::BaseCaseNum(N) + 2 * N;
|
|
||||||
const size_t partition_num = SortConstants::PartitionBufNum(N);
|
|
||||||
const size_t buf_lanes =
|
|
||||||
HWY_MAX(base_case, HWY_MAX(partition_num, PivotBufNum(sizeof_t, N)));
|
|
||||||
max_bytes = HWY_MAX(max_bytes, buf_lanes * sizeof_t);
|
|
||||||
}
|
|
||||||
|
|
||||||
ptr_ = hwy::AllocateAlignedBytes(max_bytes, nullptr, nullptr);
|
ptr_ = hwy::AllocateAlignedBytes(max_bytes, nullptr, nullptr);
|
||||||
|
|
||||||
// Prevent msan errors by initializing.
|
// Prevent msan errors by initializing.
|
||||||
memset(ptr_, 0, max_bytes);
|
memset(ptr_, 0, max_bytes);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
void Sorter::Delete() {
|
void Sorter::Delete() {
|
||||||
|
#if !VQSORT_STACK
|
||||||
FreeAlignedBytes(ptr_, nullptr, nullptr);
|
FreeAlignedBytes(ptr_, nullptr, nullptr);
|
||||||
ptr_ = nullptr;
|
ptr_ = nullptr;
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
#if !VQSORT_SECURE_RNG
|
#if !VQSORT_SECURE_RNG
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2022 Google LLC
|
// Copyright 2022 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -13,6 +14,12 @@
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
// Interface to vectorized quicksort with dynamic dispatch.
|
// Interface to vectorized quicksort with dynamic dispatch.
|
||||||
|
// Blog post: https://tinyurl.com/vqsort-blog
|
||||||
|
// Paper with measurements: https://arxiv.org/abs/2205.05982
|
||||||
|
//
|
||||||
|
// To ensure the overhead of using wide vectors (e.g. AVX2 or AVX-512) is
|
||||||
|
// worthwhile, we recommend using this code for sorting arrays whose size is at
|
||||||
|
// least 512 KiB.
|
||||||
|
|
||||||
#ifndef HIGHWAY_HWY_CONTRIB_SORT_VQSORT_H_
|
#ifndef HIGHWAY_HWY_CONTRIB_SORT_VQSORT_H_
|
||||||
#define HIGHWAY_HWY_CONTRIB_SORT_VQSORT_H_
|
#define HIGHWAY_HWY_CONTRIB_SORT_VQSORT_H_
|
||||||
|
@ -21,15 +28,6 @@
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
|
|
||||||
// Aligned 128-bit type. Cannot use __int128 because clang doesn't yet align it:
|
|
||||||
// https://reviews.llvm.org/D86310
|
|
||||||
#pragma pack(push, 1)
|
|
||||||
struct alignas(16) uint128_t {
|
|
||||||
uint64_t lo; // little-endian layout
|
|
||||||
uint64_t hi;
|
|
||||||
};
|
|
||||||
#pragma pack(pop)
|
|
||||||
|
|
||||||
// Tag arguments that determine the sort order.
|
// Tag arguments that determine the sort order.
|
||||||
struct SortAscending {
|
struct SortAscending {
|
||||||
constexpr bool IsAscending() const { return true; }
|
constexpr bool IsAscending() const { return true; }
|
||||||
|
@ -84,6 +82,9 @@ class HWY_CONTRIB_DLLEXPORT Sorter {
|
||||||
void operator()(uint128_t* HWY_RESTRICT keys, size_t n, SortAscending) const;
|
void operator()(uint128_t* HWY_RESTRICT keys, size_t n, SortAscending) const;
|
||||||
void operator()(uint128_t* HWY_RESTRICT keys, size_t n, SortDescending) const;
|
void operator()(uint128_t* HWY_RESTRICT keys, size_t n, SortDescending) const;
|
||||||
|
|
||||||
|
void operator()(K64V64* HWY_RESTRICT keys, size_t n, SortAscending) const;
|
||||||
|
void operator()(K64V64* HWY_RESTRICT keys, size_t n, SortDescending) const;
|
||||||
|
|
||||||
// For internal use only
|
// For internal use only
|
||||||
static void Fill24Bytes(const void* seed_heap, size_t seed_num, void* bytes);
|
static void Fill24Bytes(const void* seed_heap, size_t seed_num, void* bytes);
|
||||||
static bool HaveFloat64();
|
static bool HaveFloat64();
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_128a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_128a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits128-inl.h"
|
#include "hwy/contrib/sort/traits128-inl.h"
|
||||||
|
@ -29,9 +29,16 @@ namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
void Sort128Asc(uint64_t* HWY_RESTRICT keys, size_t num,
|
void Sort128Asc(uint64_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint64_t* HWY_RESTRICT buf) {
|
uint64_t* HWY_RESTRICT buf) {
|
||||||
|
#if VQSORT_ENABLED
|
||||||
SortTag<uint64_t> d;
|
SortTag<uint64_t> d;
|
||||||
detail::SharedTraits<detail::Traits128<detail::OrderAscending128>> st;
|
detail::SharedTraits<detail::Traits128<detail::OrderAscending128>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
|
#else
|
||||||
|
(void) keys;
|
||||||
|
(void) num;
|
||||||
|
(void) buf;
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_128d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_128d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits128-inl.h"
|
#include "hwy/contrib/sort/traits128-inl.h"
|
||||||
|
@ -29,9 +29,16 @@ namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
void Sort128Desc(uint64_t* HWY_RESTRICT keys, size_t num,
|
void Sort128Desc(uint64_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint64_t* HWY_RESTRICT buf) {
|
uint64_t* HWY_RESTRICT buf) {
|
||||||
|
#if VQSORT_ENABLED
|
||||||
SortTag<uint64_t> d;
|
SortTag<uint64_t> d;
|
||||||
detail::SharedTraits<detail::Traits128<detail::OrderDescending128>> st;
|
detail::SharedTraits<detail::Traits128<detail::OrderDescending128>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
|
#else
|
||||||
|
(void) keys;
|
||||||
|
(void) num;
|
||||||
|
(void) buf;
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f32a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f32a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -29,7 +29,7 @@ namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
void SortF32Asc(float* HWY_RESTRICT keys, size_t num, float* HWY_RESTRICT buf) {
|
void SortF32Asc(float* HWY_RESTRICT keys, size_t num, float* HWY_RESTRICT buf) {
|
||||||
SortTag<float> d;
|
SortTag<float> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<float>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f32d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f32d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortF32Desc(float* HWY_RESTRICT keys, size_t num,
|
void SortF32Desc(float* HWY_RESTRICT keys, size_t num,
|
||||||
float* HWY_RESTRICT buf) {
|
float* HWY_RESTRICT buf) {
|
||||||
SortTag<float> d;
|
SortTag<float> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<float>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f64a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f64a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -31,7 +31,7 @@ void SortF64Asc(double* HWY_RESTRICT keys, size_t num,
|
||||||
double* HWY_RESTRICT buf) {
|
double* HWY_RESTRICT buf) {
|
||||||
#if HWY_HAVE_FLOAT64
|
#if HWY_HAVE_FLOAT64
|
||||||
SortTag<double> d;
|
SortTag<double> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<double>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
#else
|
#else
|
||||||
(void)keys;
|
(void)keys;
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f64d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_f64d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -31,7 +31,7 @@ void SortF64Desc(double* HWY_RESTRICT keys, size_t num,
|
||||||
double* HWY_RESTRICT buf) {
|
double* HWY_RESTRICT buf) {
|
||||||
#if HWY_HAVE_FLOAT64
|
#if HWY_HAVE_FLOAT64
|
||||||
SortTag<double> d;
|
SortTag<double> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<double>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
#else
|
#else
|
||||||
(void)keys;
|
(void)keys;
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,20 +13,16 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i16a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i16a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
#include "hwy/contrib/sort/vqsort-inl.h"
|
#include "hwy/contrib/sort/vqsort-inl.h"
|
||||||
|
|
||||||
// Workaround for build timeout
|
|
||||||
#if !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
@ -33,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortI16Asc(int16_t* HWY_RESTRICT keys, size_t num,
|
void SortI16Asc(int16_t* HWY_RESTRICT keys, size_t num,
|
||||||
int16_t* HWY_RESTRICT buf) {
|
int16_t* HWY_RESTRICT buf) {
|
||||||
SortTag<int16_t> d;
|
SortTag<int16_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<int16_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -55,5 +52,3 @@ void Sorter::operator()(int16_t* HWY_RESTRICT keys, size_t n,
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
||||||
#endif // !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,20 +13,16 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i16d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i16d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
#include "hwy/contrib/sort/vqsort-inl.h"
|
#include "hwy/contrib/sort/vqsort-inl.h"
|
||||||
|
|
||||||
// Workaround for build timeout
|
|
||||||
#if !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
@ -33,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortI16Desc(int16_t* HWY_RESTRICT keys, size_t num,
|
void SortI16Desc(int16_t* HWY_RESTRICT keys, size_t num,
|
||||||
int16_t* HWY_RESTRICT buf) {
|
int16_t* HWY_RESTRICT buf) {
|
||||||
SortTag<int16_t> d;
|
SortTag<int16_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<int16_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -55,5 +52,3 @@ void Sorter::operator()(int16_t* HWY_RESTRICT keys, size_t n,
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
||||||
#endif // !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i32a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i32a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortI32Asc(int32_t* HWY_RESTRICT keys, size_t num,
|
void SortI32Asc(int32_t* HWY_RESTRICT keys, size_t num,
|
||||||
int32_t* HWY_RESTRICT buf) {
|
int32_t* HWY_RESTRICT buf) {
|
||||||
SortTag<int32_t> d;
|
SortTag<int32_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<int32_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i32d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i32d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortI32Desc(int32_t* HWY_RESTRICT keys, size_t num,
|
void SortI32Desc(int32_t* HWY_RESTRICT keys, size_t num,
|
||||||
int32_t* HWY_RESTRICT buf) {
|
int32_t* HWY_RESTRICT buf) {
|
||||||
SortTag<int32_t> d;
|
SortTag<int32_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<int32_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i64a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i64a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortI64Asc(int64_t* HWY_RESTRICT keys, size_t num,
|
void SortI64Asc(int64_t* HWY_RESTRICT keys, size_t num,
|
||||||
int64_t* HWY_RESTRICT buf) {
|
int64_t* HWY_RESTRICT buf) {
|
||||||
SortTag<int64_t> d;
|
SortTag<int64_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<int64_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i64d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_i64d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortI64Desc(int64_t* HWY_RESTRICT keys, size_t num,
|
void SortI64Desc(int64_t* HWY_RESTRICT keys, size_t num,
|
||||||
int64_t* HWY_RESTRICT buf) {
|
int64_t* HWY_RESTRICT buf) {
|
||||||
SortTag<int64_t> d;
|
SortTag<int64_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<int64_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,65 @@
|
||||||
|
// Copyright 2021 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/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
// clang-format off
|
||||||
|
// (avoid line break, which would prevent Copybara rules from matching)
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_kv128a.cc" //NOLINT
|
||||||
|
// clang-format on
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
// After foreach_target
|
||||||
|
#include "hwy/contrib/sort/traits128-inl.h"
|
||||||
|
#include "hwy/contrib/sort/vqsort-inl.h"
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
void SortKV128Asc(uint64_t* HWY_RESTRICT keys, size_t num,
|
||||||
|
uint64_t* HWY_RESTRICT buf) {
|
||||||
|
#if VQSORT_ENABLED
|
||||||
|
SortTag<uint64_t> d;
|
||||||
|
detail::SharedTraits<detail::Traits128<detail::OrderAscendingKV128>> st;
|
||||||
|
Sort(d, st, keys, num, buf);
|
||||||
|
#else
|
||||||
|
(void) keys;
|
||||||
|
(void) num;
|
||||||
|
(void) buf;
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#if HWY_ONCE
|
||||||
|
namespace hwy {
|
||||||
|
namespace {
|
||||||
|
HWY_EXPORT(SortKV128Asc);
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
void Sorter::operator()(K64V64* HWY_RESTRICT keys, size_t n,
|
||||||
|
SortAscending) const {
|
||||||
|
HWY_DYNAMIC_DISPATCH(SortKV128Asc)
|
||||||
|
(reinterpret_cast<uint64_t*>(keys), n * 2, Get<uint64_t>());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace hwy
|
||||||
|
#endif // HWY_ONCE
|
|
@ -0,0 +1,65 @@
|
||||||
|
// Copyright 2021 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/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
// clang-format off
|
||||||
|
// (avoid line break, which would prevent Copybara rules from matching)
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_kv128d.cc" //NOLINT
|
||||||
|
// clang-format on
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
|
// After foreach_target
|
||||||
|
#include "hwy/contrib/sort/traits128-inl.h"
|
||||||
|
#include "hwy/contrib/sort/vqsort-inl.h"
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
void SortKV128Desc(uint64_t* HWY_RESTRICT keys, size_t num,
|
||||||
|
uint64_t* HWY_RESTRICT buf) {
|
||||||
|
#if VQSORT_ENABLED
|
||||||
|
SortTag<uint64_t> d;
|
||||||
|
detail::SharedTraits<detail::Traits128<detail::OrderDescendingKV128>> st;
|
||||||
|
Sort(d, st, keys, num, buf);
|
||||||
|
#else
|
||||||
|
(void) keys;
|
||||||
|
(void) num;
|
||||||
|
(void) buf;
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#if HWY_ONCE
|
||||||
|
namespace hwy {
|
||||||
|
namespace {
|
||||||
|
HWY_EXPORT(SortKV128Desc);
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
void Sorter::operator()(K64V64* HWY_RESTRICT keys, size_t n,
|
||||||
|
SortDescending) const {
|
||||||
|
HWY_DYNAMIC_DISPATCH(SortKV128Desc)
|
||||||
|
(reinterpret_cast<uint64_t*>(keys), n * 2, Get<uint64_t>());
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace hwy
|
||||||
|
#endif // HWY_ONCE
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,20 +13,16 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u16a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u16a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
#include "hwy/contrib/sort/vqsort-inl.h"
|
#include "hwy/contrib/sort/vqsort-inl.h"
|
||||||
|
|
||||||
// Workaround for build timeout
|
|
||||||
#if !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
@ -33,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortU16Asc(uint16_t* HWY_RESTRICT keys, size_t num,
|
void SortU16Asc(uint16_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint16_t* HWY_RESTRICT buf) {
|
uint16_t* HWY_RESTRICT buf) {
|
||||||
SortTag<uint16_t> d;
|
SortTag<uint16_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<uint16_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -55,5 +52,3 @@ void Sorter::operator()(uint16_t* HWY_RESTRICT keys, size_t n,
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
||||||
#endif // !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,20 +13,16 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u16d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u16d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
#include "hwy/contrib/sort/vqsort-inl.h"
|
#include "hwy/contrib/sort/vqsort-inl.h"
|
||||||
|
|
||||||
// Workaround for build timeout
|
|
||||||
#if !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
||||||
HWY_BEFORE_NAMESPACE();
|
HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
@ -33,7 +30,8 @@ namespace HWY_NAMESPACE {
|
||||||
void SortU16Desc(uint16_t* HWY_RESTRICT keys, size_t num,
|
void SortU16Desc(uint16_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint16_t* HWY_RESTRICT buf) {
|
uint16_t* HWY_RESTRICT buf) {
|
||||||
SortTag<uint16_t> d;
|
SortTag<uint16_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<uint16_t>>>
|
||||||
|
st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -55,5 +53,3 @@ void Sorter::operator()(uint16_t* HWY_RESTRICT keys, size_t n,
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
#endif // HWY_ONCE
|
#endif // HWY_ONCE
|
||||||
|
|
||||||
#endif // !HWY_COMPILER_MSVC || HWY_IS_DEBUG_BUILD
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u32a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u32a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortU32Asc(uint32_t* HWY_RESTRICT keys, size_t num,
|
void SortU32Asc(uint32_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint32_t* HWY_RESTRICT buf) {
|
uint32_t* HWY_RESTRICT buf) {
|
||||||
SortTag<uint32_t> d;
|
SortTag<uint32_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<uint32_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u32d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u32d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,8 @@ namespace HWY_NAMESPACE {
|
||||||
void SortU32Desc(uint32_t* HWY_RESTRICT keys, size_t num,
|
void SortU32Desc(uint32_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint32_t* HWY_RESTRICT buf) {
|
uint32_t* HWY_RESTRICT buf) {
|
||||||
SortTag<uint32_t> d;
|
SortTag<uint32_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<uint32_t>>>
|
||||||
|
st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u64a.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u64a.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,7 @@ namespace HWY_NAMESPACE {
|
||||||
void SortU64Asc(uint64_t* HWY_RESTRICT keys, size_t num,
|
void SortU64Asc(uint64_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint64_t* HWY_RESTRICT buf) {
|
uint64_t* HWY_RESTRICT buf) {
|
||||||
SortTag<uint64_t> d;
|
SortTag<uint64_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderAscending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderAscending<uint64_t>>> st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,12 +13,11 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#include "hwy/contrib/sort/disabled_targets.h"
|
|
||||||
#include "hwy/contrib/sort/vqsort.h"
|
#include "hwy/contrib/sort/vqsort.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u64d.cc"
|
#define HWY_TARGET_INCLUDE "hwy/contrib/sort/vqsort_u64d.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// After foreach_target
|
// After foreach_target
|
||||||
#include "hwy/contrib/sort/traits-inl.h"
|
#include "hwy/contrib/sort/traits-inl.h"
|
||||||
|
@ -30,7 +30,8 @@ namespace HWY_NAMESPACE {
|
||||||
void SortU64Desc(uint64_t* HWY_RESTRICT keys, size_t num,
|
void SortU64Desc(uint64_t* HWY_RESTRICT keys, size_t num,
|
||||||
uint64_t* HWY_RESTRICT buf) {
|
uint64_t* HWY_RESTRICT buf) {
|
||||||
SortTag<uint64_t> d;
|
SortTag<uint64_t> d;
|
||||||
detail::SharedTraits<detail::LaneTraits<detail::OrderDescending>> st;
|
detail::SharedTraits<detail::TraitsLane<detail::OrderDescending<uint64_t>>>
|
||||||
|
st;
|
||||||
Sort(d, st, keys, num, buf);
|
Sort(d, st, keys, num, buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -29,34 +30,45 @@
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Compiler
|
// Compiler
|
||||||
|
|
||||||
// clang-cl defines _MSC_VER but doesn't behave like MSVC in other aspects like
|
// Actual MSVC, not clang-cl, which defines _MSC_VER but doesn't behave like
|
||||||
// used in HWY_DIAGNOSTICS(). We include a check that we are not clang for that
|
// MSVC in other aspects (e.g. HWY_DIAGNOSTICS).
|
||||||
// purpose.
|
|
||||||
#if defined(_MSC_VER) && !defined(__clang__)
|
#if defined(_MSC_VER) && !defined(__clang__)
|
||||||
#define HWY_COMPILER_MSVC _MSC_VER
|
#define HWY_COMPILER_MSVC _MSC_VER
|
||||||
#else
|
#else
|
||||||
#define HWY_COMPILER_MSVC 0
|
#define HWY_COMPILER_MSVC 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if defined(_MSC_VER) && defined(__clang__)
|
||||||
|
#define HWY_COMPILER_CLANGCL _MSC_VER
|
||||||
|
#else
|
||||||
|
#define HWY_COMPILER_CLANGCL 0
|
||||||
|
#endif
|
||||||
|
|
||||||
#ifdef __INTEL_COMPILER
|
#ifdef __INTEL_COMPILER
|
||||||
#define HWY_COMPILER_ICC __INTEL_COMPILER
|
#define HWY_COMPILER_ICC __INTEL_COMPILER
|
||||||
#else
|
#else
|
||||||
#define HWY_COMPILER_ICC 0
|
#define HWY_COMPILER_ICC 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// HWY_COMPILER_GCC is a generic macro for all compilers implementing the GNU
|
||||||
|
// compiler extensions (eg. Clang, Intel...)
|
||||||
#ifdef __GNUC__
|
#ifdef __GNUC__
|
||||||
#define HWY_COMPILER_GCC (__GNUC__ * 100 + __GNUC_MINOR__)
|
#define HWY_COMPILER_GCC (__GNUC__ * 100 + __GNUC_MINOR__)
|
||||||
#else
|
#else
|
||||||
#define HWY_COMPILER_GCC 0
|
#define HWY_COMPILER_GCC 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Clang can masquerade as MSVC/GCC, in which case both are set.
|
// Clang or clang-cl, not GCC.
|
||||||
#ifdef __clang__
|
#ifdef __clang__
|
||||||
#ifdef __APPLE__
|
// In case of Apple LLVM (whose version number is unrelated to that of LLVM) or
|
||||||
// Apple LLVM version is unrelated to the actual Clang version, which we need
|
// an invalid version number, deduce it from the presence of warnings.
|
||||||
// for enabling workarounds. Use the presence of warning flags to deduce it.
|
|
||||||
// Adapted from https://github.com/simd-everywhere/simde/ simde-detect-clang.h.
|
// Adapted from https://github.com/simd-everywhere/simde/ simde-detect-clang.h.
|
||||||
#if __has_warning("-Wformat-insufficient-args")
|
#if defined(__APPLE__) || __clang_major__ >= 999
|
||||||
|
#if __has_warning("-Wbitwise-instead-of-logical")
|
||||||
|
#define HWY_COMPILER_CLANG 1400
|
||||||
|
#elif __has_warning("-Wreserved-identifier")
|
||||||
|
#define HWY_COMPILER_CLANG 1300
|
||||||
|
#elif __has_warning("-Wformat-insufficient-args")
|
||||||
#define HWY_COMPILER_CLANG 1200
|
#define HWY_COMPILER_CLANG 1200
|
||||||
#elif __has_warning("-Wimplicit-const-int-float-conversion")
|
#elif __has_warning("-Wimplicit-const-int-float-conversion")
|
||||||
#define HWY_COMPILER_CLANG 1100
|
#define HWY_COMPILER_CLANG 1100
|
||||||
|
@ -72,19 +84,32 @@
|
||||||
#else // Anything older than 7.0 is not recommended for Highway.
|
#else // Anything older than 7.0 is not recommended for Highway.
|
||||||
#define HWY_COMPILER_CLANG 600
|
#define HWY_COMPILER_CLANG 600
|
||||||
#endif // __has_warning chain
|
#endif // __has_warning chain
|
||||||
#else // Non-Apple: normal version
|
#else // use normal version
|
||||||
#define HWY_COMPILER_CLANG (__clang_major__ * 100 + __clang_minor__)
|
#define HWY_COMPILER_CLANG (__clang_major__ * 100 + __clang_minor__)
|
||||||
#endif
|
#endif
|
||||||
#else // Not clang
|
#else // Not clang
|
||||||
#define HWY_COMPILER_CLANG 0
|
#define HWY_COMPILER_CLANG 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if HWY_COMPILER_GCC && !HWY_COMPILER_CLANG
|
||||||
|
#define HWY_COMPILER_GCC_ACTUAL HWY_COMPILER_GCC
|
||||||
|
#else
|
||||||
|
#define HWY_COMPILER_GCC_ACTUAL 0
|
||||||
|
#endif
|
||||||
|
|
||||||
// More than one may be nonzero, but we want at least one.
|
// More than one may be nonzero, but we want at least one.
|
||||||
#if !HWY_COMPILER_MSVC && !HWY_COMPILER_ICC && !HWY_COMPILER_GCC && \
|
#if 0 == (HWY_COMPILER_MSVC + HWY_COMPILER_CLANGCL + HWY_COMPILER_ICC + \
|
||||||
!HWY_COMPILER_CLANG
|
HWY_COMPILER_GCC + HWY_COMPILER_CLANG)
|
||||||
#error "Unsupported compiler"
|
#error "Unsupported compiler"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// We should only detect one of these (only clang/clangcl overlap)
|
||||||
|
#if 1 < \
|
||||||
|
(!!HWY_COMPILER_MSVC + !!HWY_COMPILER_ICC + !!HWY_COMPILER_GCC_ACTUAL + \
|
||||||
|
!!(HWY_COMPILER_CLANGCL | HWY_COMPILER_CLANG))
|
||||||
|
#error "Detected multiple compilers"
|
||||||
|
#endif
|
||||||
|
|
||||||
#ifdef __has_builtin
|
#ifdef __has_builtin
|
||||||
#define HWY_HAS_BUILTIN(name) __has_builtin(name)
|
#define HWY_HAS_BUILTIN(name) __has_builtin(name)
|
||||||
#else
|
#else
|
||||||
|
@ -140,7 +165,7 @@
|
||||||
#define HWY_ARCH_ARM_A64 0
|
#define HWY_ARCH_ARM_A64 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(__arm__) || defined(_M_ARM)
|
#if (defined(__ARM_ARCH) && __ARM_ARCH == 7) || (defined(_M_ARM) && _M_ARM == 7)
|
||||||
#define HWY_ARCH_ARM_V7 1
|
#define HWY_ARCH_ARM_V7 1
|
||||||
#else
|
#else
|
||||||
#define HWY_ARCH_ARM_V7 0
|
#define HWY_ARCH_ARM_V7 0
|
||||||
|
@ -150,12 +175,20 @@
|
||||||
#error "Cannot have both A64 and V7"
|
#error "Cannot have both A64 and V7"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Any *supported* version of Arm, i.e. 7 or later
|
||||||
#if HWY_ARCH_ARM_A64 || HWY_ARCH_ARM_V7
|
#if HWY_ARCH_ARM_A64 || HWY_ARCH_ARM_V7
|
||||||
#define HWY_ARCH_ARM 1
|
#define HWY_ARCH_ARM 1
|
||||||
#else
|
#else
|
||||||
#define HWY_ARCH_ARM 0
|
#define HWY_ARCH_ARM 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Older than v7 (e.g. armel aka Arm v5), in which case we do not support SIMD.
|
||||||
|
#if (defined(__arm__) || defined(_M_ARM)) && !HWY_ARCH_ARM
|
||||||
|
#define HWY_ARCH_ARM_OLD 1
|
||||||
|
#else
|
||||||
|
#define HWY_ARCH_ARM_OLD 0
|
||||||
|
#endif
|
||||||
|
|
||||||
#if defined(__EMSCRIPTEN__) || defined(__wasm__) || defined(__WASM__)
|
#if defined(__EMSCRIPTEN__) || defined(__wasm__) || defined(__WASM__)
|
||||||
#define HWY_ARCH_WASM 1
|
#define HWY_ARCH_WASM 1
|
||||||
#else
|
#else
|
||||||
|
@ -170,9 +203,21 @@
|
||||||
|
|
||||||
// It is an error to detect multiple architectures at the same time, but OK to
|
// It is an error to detect multiple architectures at the same time, but OK to
|
||||||
// detect none of the above.
|
// detect none of the above.
|
||||||
#if (HWY_ARCH_X86 + HWY_ARCH_PPC + HWY_ARCH_ARM + HWY_ARCH_WASM + \
|
#if (HWY_ARCH_X86 + HWY_ARCH_PPC + HWY_ARCH_ARM + HWY_ARCH_ARM_OLD + \
|
||||||
HWY_ARCH_RVV) > 1
|
HWY_ARCH_WASM + HWY_ARCH_RVV) > 1
|
||||||
#error "Must not detect more than one architecture"
|
#error "Must not detect more than one architecture"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if defined(_WIN32) || defined(_WIN64)
|
||||||
|
#define HWY_OS_WIN 1
|
||||||
|
#else
|
||||||
|
#define HWY_OS_WIN 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if defined(linux) || defined(__linux__)
|
||||||
|
#define HWY_OS_LINUX 1
|
||||||
|
#else
|
||||||
|
#define HWY_OS_LINUX 0
|
||||||
|
#endif
|
||||||
|
|
||||||
#endif // HIGHWAY_HWY_DETECT_COMPILER_ARCH_H_
|
#endif // HIGHWAY_HWY_DETECT_COMPILER_ARCH_H_
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -37,6 +38,10 @@
|
||||||
// AVX2 target for VMs which support AVX2 but not the other instruction sets)
|
// AVX2 target for VMs which support AVX2 but not the other instruction sets)
|
||||||
// #define HWY_DISABLE_BMI2_FMA
|
// #define HWY_DISABLE_BMI2_FMA
|
||||||
|
|
||||||
|
// Uncomment to enable SSSE3/SSE4 on MSVC even if AVX is not enabled
|
||||||
|
// #define HWY_WANT_SSSE3
|
||||||
|
// #define HWY_WANT_SSE4
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Targets
|
// Targets
|
||||||
|
|
||||||
|
@ -46,59 +51,71 @@
|
||||||
// All values are unconditionally defined so we can test HWY_TARGETS without
|
// All values are unconditionally defined so we can test HWY_TARGETS without
|
||||||
// first checking the HWY_ARCH_*.
|
// first checking the HWY_ARCH_*.
|
||||||
//
|
//
|
||||||
// The C99 preprocessor evaluates #if expressions using intmax_t types, so we
|
// The C99 preprocessor evaluates #if expressions using intmax_t types. This
|
||||||
// can use 32-bit literals.
|
// holds at least 64 bits in practice (verified 2022-07-18 via Godbolt on
|
||||||
|
// 32-bit clang/GCC/MSVC compilers for x86/Arm7/AArch32/RISC-V/WASM). We now
|
||||||
// 1,2: reserved
|
// avoid overflow when computing HWY_TARGETS (subtracting one instead of
|
||||||
|
// left-shifting 2^62), but still do not use bit 63 because it is the sign bit.
|
||||||
// Currently satisfiable by Ice Lake (VNNI, VPCLMULQDQ, VBMI2, VAES). Later to
|
|
||||||
// be added: BF16 (Cooper Lake). VP2INTERSECT is only in Tiger Lake? We do not
|
|
||||||
// yet have uses for VBMI, VPOPCNTDQ, BITALG, GFNI.
|
|
||||||
#define HWY_AVX3_DL 4 // see HWY_WANT_AVX3_DL below
|
|
||||||
#define HWY_AVX3 8
|
|
||||||
#define HWY_AVX2 16
|
|
||||||
// 32: reserved for AVX
|
|
||||||
#define HWY_SSE4 64
|
|
||||||
#define HWY_SSSE3 128
|
|
||||||
// 0x100, 0x200: reserved for SSE3, SSE2
|
|
||||||
|
|
||||||
|
// --------------------------- x86: 15 targets (+ one fallback)
|
||||||
|
// Bits 0..6 reserved (7 targets)
|
||||||
|
// Currently satisfiable by Ice Lake (VNNI, VPCLMULQDQ, VPOPCNTDQ, VBMI, VBMI2,
|
||||||
|
// VAES, BITALG). Later to be added: BF16 (Cooper Lake). VP2INTERSECT is only in
|
||||||
|
// Tiger Lake? We do not yet have uses for GFNI.
|
||||||
|
#define HWY_AVX3_DL (1LL << 7) // see HWY_WANT_AVX3_DL below
|
||||||
|
#define HWY_AVX3 (1LL << 8)
|
||||||
|
#define HWY_AVX2 (1LL << 9)
|
||||||
|
// Bit 10: reserved for AVX
|
||||||
|
#define HWY_SSE4 (1LL << 11)
|
||||||
|
#define HWY_SSSE3 (1LL << 12)
|
||||||
|
// Bits 13..14 reserved for SSE3 or SSE2 (2 targets)
|
||||||
// The highest bit in the HWY_TARGETS mask that a x86 target can have. Used for
|
// The highest bit in the HWY_TARGETS mask that a x86 target can have. Used for
|
||||||
// dynamic dispatch. All x86 target bits must be lower or equal to
|
// dynamic dispatch. All x86 target bits must be lower or equal to
|
||||||
// (1 << HWY_HIGHEST_TARGET_BIT_X86) and they can only use
|
// (1 << HWY_HIGHEST_TARGET_BIT_X86) and they can only use
|
||||||
// HWY_MAX_DYNAMIC_TARGETS in total.
|
// HWY_MAX_DYNAMIC_TARGETS in total.
|
||||||
#define HWY_HIGHEST_TARGET_BIT_X86 9
|
#define HWY_HIGHEST_TARGET_BIT_X86 14
|
||||||
|
|
||||||
#define HWY_SVE2 0x400
|
// --------------------------- Arm: 15 targets (+ one fallback)
|
||||||
#define HWY_SVE 0x800
|
// Bits 15..23 reserved (9 targets)
|
||||||
// 0x1000 reserved for Helium
|
#define HWY_SVE2_128 (1LL << 24) // specialized target (e.g. Arm N2)
|
||||||
#define HWY_NEON 0x2000
|
#define HWY_SVE_256 (1LL << 25) // specialized target (e.g. Arm V1)
|
||||||
|
#define HWY_SVE2 (1LL << 26)
|
||||||
|
#define HWY_SVE (1LL << 27)
|
||||||
|
#define HWY_NEON (1LL << 28) // On A64, includes/requires AES
|
||||||
|
// Bit 29 reserved (Helium?)
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT_ARM 29
|
||||||
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT_ARM 13
|
// --------------------------- RISC-V: 9 targets (+ one fallback)
|
||||||
|
// Bits 30..36 reserved (7 targets)
|
||||||
|
#define HWY_RVV (1LL << 37)
|
||||||
|
// Bit 38 reserved
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT_RVV 38
|
||||||
|
|
||||||
// 0x4000, 0x8000 reserved
|
// --------------------------- Future expansion: 4 targets
|
||||||
#define HWY_PPC8 0x10000 // v2.07 or 3
|
// Bits 39..42 reserved
|
||||||
// 0x20000, 0x40000 reserved for prior VSX/AltiVec
|
|
||||||
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT_PPC 18
|
|
||||||
|
|
||||||
#define HWY_WASM2 0x80000 // Experimental
|
// --------------------------- IBM Power: 9 targets (+ one fallback)
|
||||||
#define HWY_WASM 0x100000
|
// Bits 43..48 reserved (6 targets)
|
||||||
|
#define HWY_PPC8 (1LL << 49) // v2.07 or 3
|
||||||
|
// Bits 50..51 reserved for prior VSX/AltiVec (2 targets)
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT_PPC 51
|
||||||
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT_WASM 20
|
// --------------------------- WebAssembly: 9 targets (+ one fallback)
|
||||||
|
// Bits 52..57 reserved (6 targets)
|
||||||
|
#define HWY_WASM_EMU256 (1LL << 58) // Experimental
|
||||||
|
#define HWY_WASM (1LL << 59)
|
||||||
|
// Bits 60 reserved
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT_WASM 60
|
||||||
|
|
||||||
// 0x200000, 0x400000, 0x800000 reserved
|
// --------------------------- Emulation: 2 targets
|
||||||
|
|
||||||
#define HWY_RVV 0x1000000
|
#define HWY_EMU128 (1LL << 61)
|
||||||
|
// We do not add/left-shift, so this will not overflow to a negative number.
|
||||||
|
#define HWY_SCALAR (1LL << 62)
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT_SCALAR 62
|
||||||
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT_RVV 24
|
// Do not use bit 63 - would be confusing to have negative numbers.
|
||||||
|
|
||||||
// 0x2000000, 0x4000000, 0x8000000, 0x10000000 reserved
|
|
||||||
|
|
||||||
#define HWY_SCALAR 0x20000000
|
|
||||||
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT_SCALAR 29
|
|
||||||
|
|
||||||
// Cannot use higher values, otherwise HWY_TARGETS computation might overflow.
|
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Set default blocklists
|
// Set default blocklists
|
||||||
|
@ -138,9 +155,9 @@
|
||||||
#define HWY_BROKEN_TARGETS (HWY_NEON)
|
#define HWY_BROKEN_TARGETS (HWY_NEON)
|
||||||
|
|
||||||
// SVE[2] require recent clang or gcc versions.
|
// SVE[2] require recent clang or gcc versions.
|
||||||
#elif (HWY_COMPILER_CLANG && HWY_COMPILER_CLANG < 1100) ||\
|
#elif (HWY_COMPILER_CLANG && HWY_COMPILER_CLANG < 1100) || \
|
||||||
(!HWY_COMPILER_CLANG && HWY_COMPILER_GCC && HWY_COMPILER_GCC < 1000)
|
(HWY_COMPILER_GCC_ACTUAL && HWY_COMPILER_GCC_ACTUAL < 1000)
|
||||||
#define HWY_BROKEN_TARGETS (HWY_SVE | HWY_SVE2)
|
#define HWY_BROKEN_TARGETS (HWY_SVE | HWY_SVE2 | HWY_SVE_256 | HWY_SVE2_128)
|
||||||
|
|
||||||
#else
|
#else
|
||||||
#define HWY_BROKEN_TARGETS 0
|
#define HWY_BROKEN_TARGETS 0
|
||||||
|
@ -152,21 +169,38 @@
|
||||||
#define HWY_ENABLED(targets) \
|
#define HWY_ENABLED(targets) \
|
||||||
((targets) & ~((HWY_DISABLED_TARGETS) | (HWY_BROKEN_TARGETS)))
|
((targets) & ~((HWY_DISABLED_TARGETS) | (HWY_BROKEN_TARGETS)))
|
||||||
|
|
||||||
|
// Opt-out for EMU128 (affected by a GCC <12 bug on ARMv7: see
|
||||||
|
// https://gcc.gnu.org/bugzilla/show_bug.cgi?id=106187). This is separate from
|
||||||
|
// HWY_BROKEN_TARGETS because it affects the fallback target, which must always
|
||||||
|
// be enabled. If 1, we instead choose HWY_SCALAR even without
|
||||||
|
// HWY_COMPILE_ONLY_SCALAR being set.
|
||||||
|
#if !defined(HWY_BROKEN_EMU128) // allow overriding
|
||||||
|
#if HWY_ARCH_ARM_V7 && HWY_COMPILER_GCC_ACTUAL && HWY_COMPILER_GCC_ACTUAL < 1140
|
||||||
|
#define HWY_BROKEN_EMU128 1
|
||||||
|
#else
|
||||||
|
#define HWY_BROKEN_EMU128 0
|
||||||
|
#endif
|
||||||
|
#endif // HWY_BROKEN_EMU128
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Detect baseline targets using predefined macros
|
// Detect baseline targets using predefined macros
|
||||||
|
|
||||||
// Baseline means the targets for which the compiler is allowed to generate
|
// Baseline means the targets for which the compiler is allowed to generate
|
||||||
// instructions, implying the target CPU would have to support them. Do not use
|
// instructions, implying the target CPU would have to support them. This does
|
||||||
// this directly because it does not take the blocklist into account. Allow the
|
// not take the blocklist into account.
|
||||||
// user to override this without any guarantee of success.
|
|
||||||
#ifndef HWY_BASELINE_TARGETS
|
#if defined(HWY_COMPILE_ONLY_SCALAR) || HWY_BROKEN_EMU128
|
||||||
|
#define HWY_BASELINE_SCALAR HWY_SCALAR
|
||||||
|
#else
|
||||||
|
#define HWY_BASELINE_SCALAR HWY_EMU128
|
||||||
|
#endif
|
||||||
|
|
||||||
// Also check HWY_ARCH to ensure that simulating unknown platforms ends up with
|
// Also check HWY_ARCH to ensure that simulating unknown platforms ends up with
|
||||||
// HWY_TARGET == HWY_SCALAR.
|
// HWY_TARGET == HWY_BASELINE_SCALAR.
|
||||||
|
|
||||||
#if HWY_ARCH_WASM && defined(__wasm_simd128__)
|
#if HWY_ARCH_WASM && defined(__wasm_simd128__)
|
||||||
#if defined(HWY_WANT_WASM2)
|
#if defined(HWY_WANT_WASM2)
|
||||||
#define HWY_BASELINE_WASM HWY_WASM2
|
#define HWY_BASELINE_WASM HWY_WASM_EMU256
|
||||||
#else
|
#else
|
||||||
#define HWY_BASELINE_WASM HWY_WASM
|
#define HWY_BASELINE_WASM HWY_WASM
|
||||||
#endif // HWY_WANT_WASM2
|
#endif // HWY_WANT_WASM2
|
||||||
|
@ -181,7 +215,6 @@
|
||||||
#define HWY_BASELINE_PPC8 0
|
#define HWY_BASELINE_PPC8 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// SVE2 compiles, but is not yet tested.
|
|
||||||
#if HWY_ARCH_ARM && defined(__ARM_FEATURE_SVE2)
|
#if HWY_ARCH_ARM && defined(__ARM_FEATURE_SVE2)
|
||||||
#define HWY_BASELINE_SVE2 HWY_SVE2
|
#define HWY_BASELINE_SVE2 HWY_SVE2
|
||||||
#else
|
#else
|
||||||
|
@ -189,6 +222,11 @@
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_ARCH_ARM && defined(__ARM_FEATURE_SVE)
|
#if HWY_ARCH_ARM && defined(__ARM_FEATURE_SVE)
|
||||||
|
// Baseline targets can be used unconditionally, which does not apply to
|
||||||
|
// HWY_SVE_256 because it requires a vector size of 256 bits. Including SVE_256
|
||||||
|
// in the baseline would also disable all 'worse' targets (including SVE and
|
||||||
|
// SVE2) in non-test builds. Therefore we instead add HWY_SVE_256 to
|
||||||
|
// HWY_ATTAINABLE_TARGETS below.
|
||||||
#define HWY_BASELINE_SVE HWY_SVE
|
#define HWY_BASELINE_SVE HWY_SVE
|
||||||
#else
|
#else
|
||||||
#define HWY_BASELINE_SVE 0
|
#define HWY_BASELINE_SVE 0
|
||||||
|
@ -201,11 +239,11 @@
|
||||||
#define HWY_BASELINE_NEON 0
|
#define HWY_BASELINE_NEON 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Special handling for MSVC because it has fewer predefined macros
|
// Special handling for MSVC because it has fewer predefined macros:
|
||||||
#if HWY_COMPILER_MSVC && !HWY_COMPILER_CLANG
|
#if HWY_COMPILER_MSVC
|
||||||
|
|
||||||
// We can only be sure SSSE3/SSE4 are enabled if AVX is
|
// 1) We can only be sure SSSE3/SSE4 are enabled if AVX is:
|
||||||
// (https://stackoverflow.com/questions/18563978/)
|
// https://stackoverflow.com/questions/18563978/.
|
||||||
#if defined(__AVX__)
|
#if defined(__AVX__)
|
||||||
#define HWY_CHECK_SSSE3 1
|
#define HWY_CHECK_SSSE3 1
|
||||||
#define HWY_CHECK_SSE4 1
|
#define HWY_CHECK_SSE4 1
|
||||||
|
@ -214,7 +252,7 @@
|
||||||
#define HWY_CHECK_SSE4 0
|
#define HWY_CHECK_SSE4 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Cannot check for PCLMUL/AES and BMI2/FMA/F16C individually; we assume
|
// 2) Cannot check for PCLMUL/AES and BMI2/FMA/F16C individually; we assume
|
||||||
// PCLMUL/AES are available if SSE4 is, and BMI2/FMA/F16C if AVX2 is.
|
// PCLMUL/AES are available if SSE4 is, and BMI2/FMA/F16C if AVX2 is.
|
||||||
#define HWY_CHECK_PCLMUL_AES 1
|
#define HWY_CHECK_PCLMUL_AES 1
|
||||||
#define HWY_CHECK_BMI2_FMA 1
|
#define HWY_CHECK_BMI2_FMA 1
|
||||||
|
@ -255,13 +293,13 @@
|
||||||
|
|
||||||
#endif // non-MSVC
|
#endif // non-MSVC
|
||||||
|
|
||||||
#if HWY_ARCH_X86 && HWY_CHECK_SSSE3
|
#if HWY_ARCH_X86 && (HWY_WANT_SSSE3 || HWY_CHECK_SSSE3)
|
||||||
#define HWY_BASELINE_SSSE3 HWY_SSSE3
|
#define HWY_BASELINE_SSSE3 HWY_SSSE3
|
||||||
#else
|
#else
|
||||||
#define HWY_BASELINE_SSSE3 0
|
#define HWY_BASELINE_SSSE3 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_ARCH_X86 && HWY_CHECK_SSE4 && HWY_CHECK_PCLMUL_AES
|
#if HWY_ARCH_X86 && (HWY_WANT_SSE4 || (HWY_CHECK_SSE4 && HWY_CHECK_PCLMUL_AES))
|
||||||
#define HWY_BASELINE_SSE4 HWY_SSE4
|
#define HWY_BASELINE_SSE4 HWY_SSE4
|
||||||
#else
|
#else
|
||||||
#define HWY_BASELINE_SSE4 0
|
#define HWY_BASELINE_SSE4 0
|
||||||
|
@ -284,7 +322,9 @@
|
||||||
|
|
||||||
// TODO(janwas): not yet known whether these will be set by MSVC
|
// TODO(janwas): not yet known whether these will be set by MSVC
|
||||||
#if HWY_BASELINE_AVX3 != 0 && defined(__AVXVNNI__) && defined(__VAES__) && \
|
#if HWY_BASELINE_AVX3 != 0 && defined(__AVXVNNI__) && defined(__VAES__) && \
|
||||||
defined(__VPCLMULQDQ__)
|
defined(__VPCLMULQDQ__) && defined(__AVX512VBMI__) && \
|
||||||
|
defined(__AVX512VBMI2__) && defined(__AVX512VPOPCNTDQ__) && \
|
||||||
|
defined(__AVX512BITALG__)
|
||||||
#define HWY_BASELINE_AVX3_DL HWY_AVX3_DL
|
#define HWY_BASELINE_AVX3_DL HWY_AVX3_DL
|
||||||
#else
|
#else
|
||||||
#define HWY_BASELINE_AVX3_DL 0
|
#define HWY_BASELINE_AVX3_DL 0
|
||||||
|
@ -296,16 +336,13 @@
|
||||||
#define HWY_BASELINE_RVV 0
|
#define HWY_BASELINE_RVV 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Allow the user to override this without any guarantee of success.
|
||||||
|
#ifndef HWY_BASELINE_TARGETS
|
||||||
#define HWY_BASELINE_TARGETS \
|
#define HWY_BASELINE_TARGETS \
|
||||||
(HWY_SCALAR | HWY_BASELINE_WASM | HWY_BASELINE_PPC8 | HWY_BASELINE_SVE2 | \
|
(HWY_BASELINE_SCALAR | HWY_BASELINE_WASM | HWY_BASELINE_PPC8 | \
|
||||||
HWY_BASELINE_SVE | HWY_BASELINE_NEON | HWY_BASELINE_SSSE3 | \
|
HWY_BASELINE_SVE2 | HWY_BASELINE_SVE | HWY_BASELINE_NEON | \
|
||||||
HWY_BASELINE_SSE4 | HWY_BASELINE_AVX2 | HWY_BASELINE_AVX3 | \
|
HWY_BASELINE_SSSE3 | HWY_BASELINE_SSE4 | HWY_BASELINE_AVX2 | \
|
||||||
HWY_BASELINE_AVX3_DL | HWY_BASELINE_RVV)
|
HWY_BASELINE_AVX3 | HWY_BASELINE_AVX3_DL | HWY_BASELINE_RVV)
|
||||||
|
|
||||||
#else
|
|
||||||
// User already defined HWY_BASELINE_TARGETS, but we still need to define
|
|
||||||
// HWY_BASELINE_AVX3 (matching user's definition) for HWY_CHECK_AVX3_DL.
|
|
||||||
#define HWY_BASELINE_AVX3_DL (HWY_BASELINE_TARGETS & HWY_AVX3_DL)
|
|
||||||
#endif // HWY_BASELINE_TARGETS
|
#endif // HWY_BASELINE_TARGETS
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
|
@ -329,33 +366,68 @@
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Choose targets for dynamic dispatch according to one of four policies
|
// Choose targets for dynamic dispatch according to one of four policies
|
||||||
|
|
||||||
#if (defined(HWY_COMPILE_ONLY_SCALAR) + defined(HWY_COMPILE_ONLY_STATIC) + \
|
#if 1 < (defined(HWY_COMPILE_ONLY_SCALAR) + defined(HWY_COMPILE_ONLY_EMU128) + \
|
||||||
defined(HWY_COMPILE_ALL_ATTAINABLE)) > 1
|
defined(HWY_COMPILE_ONLY_STATIC))
|
||||||
#error "Invalid config: can only define a single policy for targets"
|
#error "Can only define one of HWY_COMPILE_ONLY_{SCALAR|EMU128|STATIC} - bug?"
|
||||||
|
#endif
|
||||||
|
// Defining one of HWY_COMPILE_ONLY_* will trump HWY_COMPILE_ALL_ATTAINABLE.
|
||||||
|
|
||||||
|
// x86 compilers generally allow runtime dispatch. On Arm, currently only GCC
|
||||||
|
// does, and we require Linux to detect CPU capabilities.
|
||||||
|
#if HWY_ARCH_X86 || (HWY_ARCH_ARM && HWY_COMPILER_GCC_ACTUAL && HWY_OS_LINUX)
|
||||||
|
#define HWY_HAVE_RUNTIME_DISPATCH 1
|
||||||
|
#else
|
||||||
|
#define HWY_HAVE_RUNTIME_DISPATCH 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Further to checking for disabled/broken targets, we only use AVX3_DL after
|
// AVX3_DL is not widely available yet. To reduce code size and compile time,
|
||||||
// explicit opt-in (via this macro OR baseline compiler flags) to avoid
|
// only include it in the set of attainable targets (for dynamic dispatch) if
|
||||||
// generating a codepath which is only helpful if the app uses AVX3_DL features.
|
// the user opts in, OR it is in the baseline (we check whether enabled below).
|
||||||
#if defined(HWY_WANT_AVX3_DL)
|
#if defined(HWY_WANT_AVX3_DL) || (HWY_BASELINE & HWY_AVX3_DL)
|
||||||
#define HWY_CHECK_AVX3_DL HWY_AVX3_DL
|
#define HWY_ATTAINABLE_AVX3_DL HWY_AVX3_DL
|
||||||
#else
|
#else
|
||||||
#define HWY_CHECK_AVX3_DL HWY_BASELINE_AVX3_DL
|
#define HWY_ATTAINABLE_AVX3_DL 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if HWY_ARCH_ARM_A64 && \
|
||||||
|
((HWY_ENABLED_BASELINE & HWY_SVE) || HWY_HAVE_RUNTIME_DISPATCH)
|
||||||
|
#define HWY_ATTAINABLE_SVE HWY_ENABLED(HWY_SVE | HWY_SVE_256)
|
||||||
|
#else
|
||||||
|
#define HWY_ATTAINABLE_SVE 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if HWY_ARCH_ARM_A64 && \
|
||||||
|
((HWY_ENABLED_BASELINE & HWY_SVE2) || HWY_HAVE_RUNTIME_DISPATCH)
|
||||||
|
#define HWY_ATTAINABLE_SVE2 HWY_ENABLED(HWY_SVE2 | HWY_SVE2_128)
|
||||||
|
#else
|
||||||
|
#define HWY_ATTAINABLE_SVE2 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Attainable means enabled and the compiler allows intrinsics (even when not
|
// Attainable means enabled and the compiler allows intrinsics (even when not
|
||||||
// allowed to autovectorize). Used in 3 and 4.
|
// allowed to autovectorize). Used in 3 and 4.
|
||||||
#if HWY_ARCH_X86
|
#if HWY_ARCH_X86
|
||||||
#define HWY_ATTAINABLE_TARGETS \
|
#define HWY_ATTAINABLE_TARGETS \
|
||||||
HWY_ENABLED(HWY_SCALAR | HWY_SSSE3 | HWY_SSE4 | HWY_AVX2 | HWY_AVX3 | \
|
HWY_ENABLED(HWY_BASELINE_SCALAR | HWY_SSSE3 | HWY_SSE4 | HWY_AVX2 | \
|
||||||
HWY_CHECK_AVX3_DL)
|
HWY_AVX3 | HWY_ATTAINABLE_AVX3_DL)
|
||||||
|
#elif HWY_ARCH_ARM && HWY_HAVE_RUNTIME_DISPATCH
|
||||||
|
#define HWY_ATTAINABLE_TARGETS \
|
||||||
|
HWY_ENABLED(HWY_BASELINE_SCALAR | HWY_NEON | HWY_ATTAINABLE_SVE | \
|
||||||
|
HWY_ATTAINABLE_SVE2)
|
||||||
#else
|
#else
|
||||||
#define HWY_ATTAINABLE_TARGETS HWY_ENABLED_BASELINE
|
#define HWY_ATTAINABLE_TARGETS \
|
||||||
|
(HWY_ENABLED_BASELINE | HWY_ATTAINABLE_SVE | HWY_ATTAINABLE_SVE2)
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// 1) For older compilers: disable all SIMD (could also set HWY_DISABLED_TARGETS
|
// 1) For older compilers: avoid SIMD intrinsics, but still support all ops.
|
||||||
// to ~HWY_SCALAR, but this is more explicit).
|
#if defined(HWY_COMPILE_ONLY_EMU128) && !HWY_BROKEN_EMU128
|
||||||
#if defined(HWY_COMPILE_ONLY_SCALAR)
|
#undef HWY_STATIC_TARGET
|
||||||
|
#define HWY_STATIC_TARGET HWY_EMU128 // override baseline
|
||||||
|
#define HWY_TARGETS HWY_EMU128
|
||||||
|
|
||||||
|
// 1b) HWY_SCALAR is less capable than HWY_EMU128 (which supports all ops), but
|
||||||
|
// we currently still support it for backwards compatibility.
|
||||||
|
#elif defined(HWY_COMPILE_ONLY_SCALAR) || \
|
||||||
|
(defined(HWY_COMPILE_ONLY_EMU128) && HWY_BROKEN_EMU128)
|
||||||
#undef HWY_STATIC_TARGET
|
#undef HWY_STATIC_TARGET
|
||||||
#define HWY_STATIC_TARGET HWY_SCALAR // override baseline
|
#define HWY_STATIC_TARGET HWY_SCALAR // override baseline
|
||||||
#define HWY_TARGETS HWY_SCALAR
|
#define HWY_TARGETS HWY_SCALAR
|
||||||
|
@ -369,9 +441,12 @@
|
||||||
#define HWY_TARGETS HWY_ATTAINABLE_TARGETS
|
#define HWY_TARGETS HWY_ATTAINABLE_TARGETS
|
||||||
|
|
||||||
// 4) Default: attainable WITHOUT non-best baseline. This reduces code size by
|
// 4) Default: attainable WITHOUT non-best baseline. This reduces code size by
|
||||||
// excluding superseded targets, in particular scalar.
|
// excluding superseded targets, in particular scalar. Note: HWY_STATIC_TARGET
|
||||||
|
// may be 2^62 (HWY_SCALAR), so we must not left-shift/add it. Subtracting one
|
||||||
|
// sets all lower bits (better targets), then we also include the static target.
|
||||||
#else
|
#else
|
||||||
#define HWY_TARGETS (HWY_ATTAINABLE_TARGETS & (2 * HWY_STATIC_TARGET - 1))
|
#define HWY_TARGETS \
|
||||||
|
(HWY_ATTAINABLE_TARGETS & ((HWY_STATIC_TARGET - 1LL) | HWY_STATIC_TARGET))
|
||||||
|
|
||||||
#endif // target policy
|
#endif // target policy
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -12,10 +13,6 @@
|
||||||
// See the License for the specific language governing permissions and
|
// See the License for the specific language governing permissions and
|
||||||
// limitations under the License.
|
// limitations under the License.
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
|
||||||
#define HWY_TARGET_INCLUDE "hwy/examples/benchmark.cc"
|
|
||||||
#include "hwy/foreach_target.h"
|
|
||||||
|
|
||||||
#include <inttypes.h>
|
#include <inttypes.h>
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
@ -24,8 +21,12 @@
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <numeric> // iota
|
#include <numeric> // iota
|
||||||
|
|
||||||
#include "hwy/aligned_allocator.h"
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/examples/benchmark.cc"
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// Must come after foreach_target.h to avoid redefinition errors.
|
// Must come after foreach_target.h to avoid redefinition errors.
|
||||||
|
#include "hwy/aligned_allocator.h"
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
#include "hwy/nanobenchmark.h"
|
#include "hwy/nanobenchmark.h"
|
||||||
|
|
||||||
|
@ -81,7 +82,8 @@ void RunBenchmark(const char* caption) {
|
||||||
benchmark.Verify(num_items);
|
benchmark.Verify(num_items);
|
||||||
|
|
||||||
for (size_t i = 0; i < num_results; ++i) {
|
for (size_t i = 0; i < num_results; ++i) {
|
||||||
const double cycles_per_item = results[i].ticks / double(results[i].input);
|
const double cycles_per_item =
|
||||||
|
results[i].ticks / static_cast<double>(results[i].input);
|
||||||
const double mad = results[i].variability * cycles_per_item;
|
const double mad = results[i].variability * cycles_per_item;
|
||||||
printf("%6" PRIu64 ": %6.3f (+/- %5.3f)\n",
|
printf("%6" PRIu64 ": %6.3f (+/- %5.3f)\n",
|
||||||
static_cast<uint64_t>(results[i].input), cycles_per_item, mad);
|
static_cast<uint64_t>(results[i].input), cycles_per_item, mad);
|
||||||
|
@ -233,7 +235,7 @@ namespace hwy {
|
||||||
HWY_EXPORT(RunBenchmarks);
|
HWY_EXPORT(RunBenchmarks);
|
||||||
|
|
||||||
void Run() {
|
void Run() {
|
||||||
for (uint32_t target : SupportedAndGeneratedTargets()) {
|
for (int64_t target : SupportedAndGeneratedTargets()) {
|
||||||
SetSupportedTargetsForTest(target);
|
SetSupportedTargetsForTest(target);
|
||||||
HWY_DYNAMIC_DISPATCH(RunBenchmarks)();
|
HWY_DYNAMIC_DISPATCH(RunBenchmarks)();
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -22,7 +23,7 @@
|
||||||
// __FILE__ is not reliable) so that foreach_target.h can re-include it.
|
// __FILE__ is not reliable) so that foreach_target.h can re-include it.
|
||||||
#define HWY_TARGET_INCLUDE "hwy/examples/skeleton.cc"
|
#define HWY_TARGET_INCLUDE "hwy/examples/skeleton.cc"
|
||||||
// Generates code for each enabled target by re-including this source file.
|
// Generates code for each enabled target by re-including this source file.
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// Must come after foreach_target.h to avoid redefinition errors.
|
// Must come after foreach_target.h to avoid redefinition errors.
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
@ -35,28 +36,21 @@ namespace skeleton {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
// Highway ops reside here; ADL does not find templates nor builtins.
|
// Highway ops reside here; ADL does not find templates nor builtins.
|
||||||
using namespace hwy::HWY_NAMESPACE;
|
namespace hn = hwy::HWY_NAMESPACE;
|
||||||
|
|
||||||
// For reasons unknown, optimized msan builds encounter long build times here;
|
|
||||||
// work around it until a cause is found.
|
|
||||||
#if HWY_COMPILER_CLANG && defined(MEMORY_SANITIZER) && defined(__OPTIMIZE__)
|
|
||||||
#define ATTR_MSAN __attribute__((optnone))
|
|
||||||
#else
|
|
||||||
#define ATTR_MSAN
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Computes log2 by converting to a vector of floats. Compiled once per target.
|
// Computes log2 by converting to a vector of floats. Compiled once per target.
|
||||||
template <class DF>
|
template <class DF>
|
||||||
ATTR_MSAN void OneFloorLog2(const DF df, const uint8_t* HWY_RESTRICT values,
|
HWY_ATTR_NO_MSAN void OneFloorLog2(const DF df,
|
||||||
|
const uint8_t* HWY_RESTRICT values,
|
||||||
uint8_t* HWY_RESTRICT log2) {
|
uint8_t* HWY_RESTRICT log2) {
|
||||||
// Type tags for converting to other element types (Rebind = same count).
|
// Type tags for converting to other element types (Rebind = same count).
|
||||||
const RebindToSigned<DF> d32;
|
const hn::RebindToSigned<DF> d32;
|
||||||
const Rebind<uint8_t, DF> d8;
|
const hn::Rebind<uint8_t, DF> d8;
|
||||||
|
|
||||||
const auto u8 = Load(d8, values);
|
const auto u8 = hn::Load(d8, values);
|
||||||
const auto bits = BitCast(d32, ConvertTo(df, PromoteTo(d32, u8)));
|
const auto bits = hn::BitCast(d32, hn::ConvertTo(df, hn::PromoteTo(d32, u8)));
|
||||||
const auto exponent = Sub(ShiftRight<23>(bits), Set(d32, 127));
|
const auto exponent = hn::Sub(hn::ShiftRight<23>(bits), hn::Set(d32, 127));
|
||||||
Store(DemoteTo(d8, exponent), d8, log2);
|
hn::Store(hn::DemoteTo(d8, exponent), d8, log2);
|
||||||
}
|
}
|
||||||
|
|
||||||
void CodepathDemo() {
|
void CodepathDemo() {
|
||||||
|
@ -74,14 +68,14 @@ void FloorLog2(const uint8_t* HWY_RESTRICT values, size_t count,
|
||||||
uint8_t* HWY_RESTRICT log2) {
|
uint8_t* HWY_RESTRICT log2) {
|
||||||
CodepathDemo();
|
CodepathDemo();
|
||||||
|
|
||||||
const ScalableTag<float> df;
|
const hn::ScalableTag<float> df;
|
||||||
const size_t N = Lanes(df);
|
const size_t N = hn::Lanes(df);
|
||||||
size_t i = 0;
|
size_t i = 0;
|
||||||
for (; i + N <= count; i += N) {
|
for (; i + N <= count; i += N) {
|
||||||
OneFloorLog2(df, values + i, log2 + i);
|
OneFloorLog2(df, values + i, log2 + i);
|
||||||
}
|
}
|
||||||
for (; i < count; ++i) {
|
for (; i < count; ++i) {
|
||||||
CappedTag<float, 1> d1;
|
hn::CappedTag<float, 1> d1;
|
||||||
OneFloorLog2(d1, values + i, log2 + i);
|
OneFloorLog2(d1, values + i, log2 + i);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -105,7 +99,8 @@ HWY_EXPORT(FloorLog2);
|
||||||
// This function is optional and only needed in the case of exposing it in the
|
// This function is optional and only needed in the case of exposing it in the
|
||||||
// header file. Otherwise using HWY_DYNAMIC_DISPATCH(FloorLog2) in this module
|
// header file. Otherwise using HWY_DYNAMIC_DISPATCH(FloorLog2) in this module
|
||||||
// is equivalent to inlining this function.
|
// is equivalent to inlining this function.
|
||||||
void CallFloorLog2(const uint8_t* HWY_RESTRICT in, const size_t count,
|
HWY_DLLEXPORT void CallFloorLog2(const uint8_t* HWY_RESTRICT in,
|
||||||
|
const size_t count,
|
||||||
uint8_t* HWY_RESTRICT out) {
|
uint8_t* HWY_RESTRICT out) {
|
||||||
// This must reside outside of HWY_NAMESPACE because it references (calls the
|
// This must reside outside of HWY_NAMESPACE because it references (calls the
|
||||||
// appropriate one from) the per-target implementations there.
|
// appropriate one from) the per-target implementations there.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -27,8 +28,8 @@
|
||||||
namespace skeleton {
|
namespace skeleton {
|
||||||
|
|
||||||
// Computes base-2 logarithm by converting to float. Supports dynamic dispatch.
|
// Computes base-2 logarithm by converting to float. Supports dynamic dispatch.
|
||||||
void CallFloorLog2(const uint8_t* HWY_RESTRICT in, const size_t count,
|
HWY_DLLEXPORT void CallFloorLog2(const uint8_t* HWY_RESTRICT in,
|
||||||
uint8_t* HWY_RESTRICT out);
|
const size_t count, uint8_t* HWY_RESTRICT out);
|
||||||
|
|
||||||
} // namespace skeleton
|
} // namespace skeleton
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -20,7 +21,7 @@
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "examples/skeleton_test.cc"
|
#define HWY_TARGET_INCLUDE "examples/skeleton_test.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
|
||||||
// Must come after foreach_target.h to avoid redefinition errors.
|
// Must come after foreach_target.h to avoid redefinition errors.
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
|
@ -34,13 +35,13 @@ HWY_BEFORE_NAMESPACE();
|
||||||
namespace skeleton {
|
namespace skeleton {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
using namespace hwy::HWY_NAMESPACE;
|
namespace hn = hwy::HWY_NAMESPACE;
|
||||||
|
|
||||||
// Calls function defined in skeleton.cc.
|
// Calls function defined in skeleton.cc.
|
||||||
struct TestFloorLog2 {
|
struct TestFloorLog2 {
|
||||||
template <class T, class DF>
|
template <class T, class DF>
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, DF df) {
|
HWY_NOINLINE void operator()(T /*unused*/, DF df) {
|
||||||
const size_t count = 5 * Lanes(df);
|
const size_t count = 5 * hn::Lanes(df);
|
||||||
auto in = hwy::AllocateAligned<uint8_t>(count);
|
auto in = hwy::AllocateAligned<uint8_t>(count);
|
||||||
auto expected = hwy::AllocateAligned<uint8_t>(count);
|
auto expected = hwy::AllocateAligned<uint8_t>(count);
|
||||||
|
|
||||||
|
@ -70,7 +71,7 @@ struct TestSumMulAdd {
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
hwy::RandomState rng;
|
hwy::RandomState rng;
|
||||||
const size_t count = 4096;
|
const size_t count = 4096;
|
||||||
EXPECT_TRUE(count % Lanes(d) == 0);
|
EXPECT_EQ(0, count % hn::Lanes(d));
|
||||||
auto mul = hwy::AllocateAligned<T>(count);
|
auto mul = hwy::AllocateAligned<T>(count);
|
||||||
auto x = hwy::AllocateAligned<T>(count);
|
auto x = hwy::AllocateAligned<T>(count);
|
||||||
auto add = hwy::AllocateAligned<T>(count);
|
auto add = hwy::AllocateAligned<T>(count);
|
||||||
|
@ -106,10 +107,4 @@ HWY_EXPORT_AND_TEST_P(SkeletonTest, TestAllFloorLog2);
|
||||||
HWY_EXPORT_AND_TEST_P(SkeletonTest, TestAllSumMulAdd);
|
HWY_EXPORT_AND_TEST_P(SkeletonTest, TestAllSumMulAdd);
|
||||||
} // namespace skeleton
|
} // namespace skeleton
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char **argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -52,6 +53,17 @@
|
||||||
#error ">1 target enabled => define HWY_TARGET_INCLUDE before foreach_target.h"
|
#error ">1 target enabled => define HWY_TARGET_INCLUDE before foreach_target.h"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if (HWY_TARGETS & HWY_EMU128) && (HWY_STATIC_TARGET != HWY_EMU128)
|
||||||
|
#undef HWY_TARGET
|
||||||
|
#define HWY_TARGET HWY_EMU128
|
||||||
|
#include HWY_TARGET_INCLUDE
|
||||||
|
#ifdef HWY_TARGET_TOGGLE
|
||||||
|
#undef HWY_TARGET_TOGGLE
|
||||||
|
#else
|
||||||
|
#define HWY_TARGET_TOGGLE
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
#if (HWY_TARGETS & HWY_SCALAR) && (HWY_STATIC_TARGET != HWY_SCALAR)
|
#if (HWY_TARGETS & HWY_SCALAR) && (HWY_STATIC_TARGET != HWY_SCALAR)
|
||||||
#undef HWY_TARGET
|
#undef HWY_TARGET
|
||||||
#define HWY_TARGET HWY_SCALAR
|
#define HWY_TARGET HWY_SCALAR
|
||||||
|
@ -74,6 +86,17 @@
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if (HWY_TARGETS & HWY_RVV) && (HWY_STATIC_TARGET != HWY_RVV)
|
||||||
|
#undef HWY_TARGET
|
||||||
|
#define HWY_TARGET HWY_RVV
|
||||||
|
#include HWY_TARGET_INCLUDE
|
||||||
|
#ifdef HWY_TARGET_TOGGLE
|
||||||
|
#undef HWY_TARGET_TOGGLE
|
||||||
|
#else
|
||||||
|
#define HWY_TARGET_TOGGLE
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
#if (HWY_TARGETS & HWY_SVE) && (HWY_STATIC_TARGET != HWY_SVE)
|
#if (HWY_TARGETS & HWY_SVE) && (HWY_STATIC_TARGET != HWY_SVE)
|
||||||
#undef HWY_TARGET
|
#undef HWY_TARGET
|
||||||
#define HWY_TARGET HWY_SVE
|
#define HWY_TARGET HWY_SVE
|
||||||
|
@ -96,6 +119,28 @@
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if (HWY_TARGETS & HWY_SVE_256) && (HWY_STATIC_TARGET != HWY_SVE_256)
|
||||||
|
#undef HWY_TARGET
|
||||||
|
#define HWY_TARGET HWY_SVE_256
|
||||||
|
#include HWY_TARGET_INCLUDE
|
||||||
|
#ifdef HWY_TARGET_TOGGLE
|
||||||
|
#undef HWY_TARGET_TOGGLE
|
||||||
|
#else
|
||||||
|
#define HWY_TARGET_TOGGLE
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if (HWY_TARGETS & HWY_SVE2_128) && (HWY_STATIC_TARGET != HWY_SVE2_128)
|
||||||
|
#undef HWY_TARGET
|
||||||
|
#define HWY_TARGET HWY_SVE2_128
|
||||||
|
#include HWY_TARGET_INCLUDE
|
||||||
|
#ifdef HWY_TARGET_TOGGLE
|
||||||
|
#undef HWY_TARGET_TOGGLE
|
||||||
|
#else
|
||||||
|
#define HWY_TARGET_TOGGLE
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
#if (HWY_TARGETS & HWY_SSSE3) && (HWY_STATIC_TARGET != HWY_SSSE3)
|
#if (HWY_TARGETS & HWY_SSSE3) && (HWY_STATIC_TARGET != HWY_SSSE3)
|
||||||
#undef HWY_TARGET
|
#undef HWY_TARGET
|
||||||
#define HWY_TARGET HWY_SSSE3
|
#define HWY_TARGET HWY_SSSE3
|
||||||
|
@ -151,9 +196,9 @@
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if (HWY_TARGETS & HWY_WASM2) && (HWY_STATIC_TARGET != HWY_WASM2)
|
#if (HWY_TARGETS & HWY_WASM_EMU256) && (HWY_STATIC_TARGET != HWY_WASM_EMU256)
|
||||||
#undef HWY_TARGET
|
#undef HWY_TARGET
|
||||||
#define HWY_TARGET HWY_WASM2
|
#define HWY_TARGET HWY_WASM_EMU256
|
||||||
#include HWY_TARGET_INCLUDE
|
#include HWY_TARGET_INCLUDE
|
||||||
#ifdef HWY_TARGET_TOGGLE
|
#ifdef HWY_TARGET_TOGGLE
|
||||||
#undef HWY_TARGET_TOGGLE
|
#undef HWY_TARGET_TOGGLE
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -26,8 +27,8 @@
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
|
|
||||||
// API version (https://semver.org/); keep in sync with CMakeLists.txt.
|
// API version (https://semver.org/); keep in sync with CMakeLists.txt.
|
||||||
#define HWY_MAJOR 0
|
#define HWY_MAJOR 1
|
||||||
#define HWY_MINOR 16
|
#define HWY_MINOR 0
|
||||||
#define HWY_PATCH 0
|
#define HWY_PATCH 0
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
|
@ -39,7 +40,7 @@ namespace hwy {
|
||||||
// registers in the group, and is ignored on targets that do not support groups.
|
// registers in the group, and is ignored on targets that do not support groups.
|
||||||
#define HWY_FULL1(T) hwy::HWY_NAMESPACE::ScalableTag<T>
|
#define HWY_FULL1(T) hwy::HWY_NAMESPACE::ScalableTag<T>
|
||||||
#define HWY_FULL2(T, LMUL) \
|
#define HWY_FULL2(T, LMUL) \
|
||||||
hwy::HWY_NAMESPACE::ScalableTag<T, CeilLog2(HWY_MAX(0, LMUL))>
|
hwy::HWY_NAMESPACE::ScalableTag<T, hwy::CeilLog2(HWY_MAX(0, LMUL))>
|
||||||
#define HWY_3TH_ARG(arg1, arg2, arg3, ...) arg3
|
#define HWY_3TH_ARG(arg1, arg2, arg3, ...) arg3
|
||||||
// Workaround for MSVC grouping __VA_ARGS__ into a single argument
|
// Workaround for MSVC grouping __VA_ARGS__ into a single argument
|
||||||
#define HWY_FULL_RECOMPOSER(args_with_paren) HWY_3TH_ARG args_with_paren
|
#define HWY_FULL_RECOMPOSER(args_with_paren) HWY_3TH_ARG args_with_paren
|
||||||
|
@ -49,8 +50,7 @@ namespace hwy {
|
||||||
#define HWY_FULL(...) HWY_CHOOSE_FULL(__VA_ARGS__())(__VA_ARGS__)
|
#define HWY_FULL(...) HWY_CHOOSE_FULL(__VA_ARGS__())(__VA_ARGS__)
|
||||||
|
|
||||||
// Vector of up to MAX_N lanes. It's better to use full vectors where possible.
|
// Vector of up to MAX_N lanes. It's better to use full vectors where possible.
|
||||||
#define HWY_CAPPED(T, MAX_N) \
|
#define HWY_CAPPED(T, MAX_N) hwy::HWY_NAMESPACE::CappedTag<T, MAX_N>
|
||||||
hwy::HWY_NAMESPACE::CappedTag<T, HWY_MIN(MAX_N, HWY_LANES(T))>
|
|
||||||
|
|
||||||
//------------------------------------------------------------------------------
|
//------------------------------------------------------------------------------
|
||||||
// Export user functions for static/dynamic dispatch
|
// Export user functions for static/dynamic dispatch
|
||||||
|
@ -68,10 +68,12 @@ namespace hwy {
|
||||||
// defined), and can be used to deduce the return type of Choose*.
|
// defined), and can be used to deduce the return type of Choose*.
|
||||||
#if HWY_STATIC_TARGET == HWY_SCALAR
|
#if HWY_STATIC_TARGET == HWY_SCALAR
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SCALAR::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SCALAR::FUNC_NAME
|
||||||
|
#elif HWY_STATIC_TARGET == HWY_EMU128
|
||||||
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_EMU128::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_RVV
|
#elif HWY_STATIC_TARGET == HWY_RVV
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_RVV::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_RVV::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_WASM2
|
#elif HWY_STATIC_TARGET == HWY_WASM_EMU256
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_WASM2::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_WASM_EMU256::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_WASM
|
#elif HWY_STATIC_TARGET == HWY_WASM
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_WASM::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_WASM::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_NEON
|
#elif HWY_STATIC_TARGET == HWY_NEON
|
||||||
|
@ -80,6 +82,10 @@ namespace hwy {
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SVE::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SVE::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_SVE2
|
#elif HWY_STATIC_TARGET == HWY_SVE2
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SVE2::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SVE2::FUNC_NAME
|
||||||
|
#elif HWY_STATIC_TARGET == HWY_SVE_256
|
||||||
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SVE_256::FUNC_NAME
|
||||||
|
#elif HWY_STATIC_TARGET == HWY_SVE2_128
|
||||||
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_SVE2_128::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_PPC8
|
#elif HWY_STATIC_TARGET == HWY_PPC8
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_PPC8::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_PPC8::FUNC_NAME
|
||||||
#elif HWY_STATIC_TARGET == HWY_SSSE3
|
#elif HWY_STATIC_TARGET == HWY_SSSE3
|
||||||
|
@ -94,51 +100,22 @@ namespace hwy {
|
||||||
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_AVX3_DL::FUNC_NAME
|
#define HWY_STATIC_DISPATCH(FUNC_NAME) N_AVX3_DL::FUNC_NAME
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Dynamic dispatch declarations.
|
|
||||||
|
|
||||||
template <typename RetType, typename... Args>
|
|
||||||
struct FunctionCache {
|
|
||||||
public:
|
|
||||||
typedef RetType(FunctionType)(Args...);
|
|
||||||
|
|
||||||
// A template function that when instantiated has the same signature as the
|
|
||||||
// function being called. This function initializes the global cache of the
|
|
||||||
// current supported targets mask used for dynamic dispatch and calls the
|
|
||||||
// appropriate function. Since this mask used for dynamic dispatch is a
|
|
||||||
// global cache, all the highway exported functions, even those exposed by
|
|
||||||
// different modules, will be initialized after this function runs for any one
|
|
||||||
// of those exported functions.
|
|
||||||
template <FunctionType* const table[]>
|
|
||||||
static RetType ChooseAndCall(Args... args) {
|
|
||||||
// If we are running here it means we need to update the chosen target.
|
|
||||||
ChosenTarget& chosen_target = GetChosenTarget();
|
|
||||||
chosen_target.Update();
|
|
||||||
return (table[chosen_target.GetIndex()])(args...);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Factory function only used to infer the template parameters RetType and Args
|
|
||||||
// from a function passed to the factory.
|
|
||||||
template <typename RetType, typename... Args>
|
|
||||||
FunctionCache<RetType, Args...> FunctionCacheFactory(RetType (*)(Args...)) {
|
|
||||||
return FunctionCache<RetType, Args...>();
|
|
||||||
}
|
|
||||||
|
|
||||||
// HWY_CHOOSE_*(FUNC_NAME) expands to the function pointer for that target or
|
// HWY_CHOOSE_*(FUNC_NAME) expands to the function pointer for that target or
|
||||||
// nullptr is that target was not compiled.
|
// nullptr is that target was not compiled.
|
||||||
#if HWY_TARGETS & HWY_SCALAR
|
#if HWY_TARGETS & HWY_EMU128
|
||||||
#define HWY_CHOOSE_SCALAR(FUNC_NAME) &N_SCALAR::FUNC_NAME
|
#define HWY_CHOOSE_FALLBACK(FUNC_NAME) &N_EMU128::FUNC_NAME
|
||||||
|
#elif HWY_TARGETS & HWY_SCALAR
|
||||||
|
#define HWY_CHOOSE_FALLBACK(FUNC_NAME) &N_SCALAR::FUNC_NAME
|
||||||
#else
|
#else
|
||||||
// When scalar is not present and we try to use scalar because other targets
|
// When HWY_SCALAR/HWY_EMU128 are not present and other targets were disabled at
|
||||||
// were disabled at runtime we fall back to the baseline with
|
// runtime, fall back to the baseline with HWY_STATIC_DISPATCH().
|
||||||
// HWY_STATIC_DISPATCH()
|
#define HWY_CHOOSE_FALLBACK(FUNC_NAME) &HWY_STATIC_DISPATCH(FUNC_NAME)
|
||||||
#define HWY_CHOOSE_SCALAR(FUNC_NAME) &HWY_STATIC_DISPATCH(FUNC_NAME)
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_TARGETS & HWY_WASM2
|
#if HWY_TARGETS & HWY_WASM_EMU256
|
||||||
#define HWY_CHOOSE_WASM2(FUNC_NAME) &N_WASM2::FUNC_NAME
|
#define HWY_CHOOSE_WASM_EMU256(FUNC_NAME) &N_WASM_EMU256::FUNC_NAME
|
||||||
#else
|
#else
|
||||||
#define HWY_CHOOSE_WASM2(FUNC_NAME) nullptr
|
#define HWY_CHOOSE_WASM_EMU256(FUNC_NAME) nullptr
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_TARGETS & HWY_WASM
|
#if HWY_TARGETS & HWY_WASM
|
||||||
|
@ -171,6 +148,18 @@ FunctionCache<RetType, Args...> FunctionCacheFactory(RetType (*)(Args...)) {
|
||||||
#define HWY_CHOOSE_SVE2(FUNC_NAME) nullptr
|
#define HWY_CHOOSE_SVE2(FUNC_NAME) nullptr
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if HWY_TARGETS & HWY_SVE_256
|
||||||
|
#define HWY_CHOOSE_SVE_256(FUNC_NAME) &N_SVE_256::FUNC_NAME
|
||||||
|
#else
|
||||||
|
#define HWY_CHOOSE_SVE_256(FUNC_NAME) nullptr
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if HWY_TARGETS & HWY_SVE2_128
|
||||||
|
#define HWY_CHOOSE_SVE2_128(FUNC_NAME) &N_SVE2_128::FUNC_NAME
|
||||||
|
#else
|
||||||
|
#define HWY_CHOOSE_SVE2_128(FUNC_NAME) nullptr
|
||||||
|
#endif
|
||||||
|
|
||||||
#if HWY_TARGETS & HWY_PPC8
|
#if HWY_TARGETS & HWY_PPC8
|
||||||
#define HWY_CHOOSE_PCC8(FUNC_NAME) &N_PPC8::FUNC_NAME
|
#define HWY_CHOOSE_PCC8(FUNC_NAME) &N_PPC8::FUNC_NAME
|
||||||
#else
|
#else
|
||||||
|
@ -207,6 +196,53 @@ FunctionCache<RetType, Args...> FunctionCacheFactory(RetType (*)(Args...)) {
|
||||||
#define HWY_CHOOSE_AVX3_DL(FUNC_NAME) nullptr
|
#define HWY_CHOOSE_AVX3_DL(FUNC_NAME) nullptr
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// MSVC 2017 workaround: the non-type template parameter to ChooseAndCall
|
||||||
|
// apparently cannot be an array. Use a function pointer instead, which has the
|
||||||
|
// disadvantage that we call the static (not best) target on the first call to
|
||||||
|
// any HWY_DYNAMIC_DISPATCH.
|
||||||
|
#if HWY_COMPILER_MSVC && HWY_COMPILER_MSVC < 1915
|
||||||
|
#define HWY_DISPATCH_WORKAROUND 1
|
||||||
|
#else
|
||||||
|
#define HWY_DISPATCH_WORKAROUND 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Provides a static member function which is what is called during the first
|
||||||
|
// HWY_DYNAMIC_DISPATCH, where GetIndex is still zero, and instantiations of
|
||||||
|
// this function are the first entry in the tables created by HWY_EXPORT.
|
||||||
|
template <typename RetType, typename... Args>
|
||||||
|
struct FunctionCache {
|
||||||
|
public:
|
||||||
|
typedef RetType(FunctionType)(Args...);
|
||||||
|
|
||||||
|
#if HWY_DISPATCH_WORKAROUND
|
||||||
|
template <FunctionType* const func>
|
||||||
|
static RetType ChooseAndCall(Args... args) {
|
||||||
|
ChosenTarget& chosen_target = GetChosenTarget();
|
||||||
|
chosen_target.Update(SupportedTargets());
|
||||||
|
return (*func)(args...);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
// A template function that when instantiated has the same signature as the
|
||||||
|
// function being called. This function initializes the bit array of targets
|
||||||
|
// supported by the current CPU and then calls the appropriate entry within
|
||||||
|
// the HWY_EXPORT table. Subsequent calls via HWY_DYNAMIC_DISPATCH to any
|
||||||
|
// exported functions, even those defined by different translation units,
|
||||||
|
// will dispatch directly to the best available target.
|
||||||
|
template <FunctionType* const table[]>
|
||||||
|
static RetType ChooseAndCall(Args... args) {
|
||||||
|
ChosenTarget& chosen_target = GetChosenTarget();
|
||||||
|
chosen_target.Update(SupportedTargets());
|
||||||
|
return (table[chosen_target.GetIndex()])(args...);
|
||||||
|
}
|
||||||
|
#endif // HWY_DISPATCH_WORKAROUND
|
||||||
|
};
|
||||||
|
|
||||||
|
// Used to deduce the template parameters RetType and Args from a function.
|
||||||
|
template <typename RetType, typename... Args>
|
||||||
|
FunctionCache<RetType, Args...> DeduceFunctionCache(RetType (*)(Args...)) {
|
||||||
|
return FunctionCache<RetType, Args...>();
|
||||||
|
}
|
||||||
|
|
||||||
#define HWY_DISPATCH_TABLE(FUNC_NAME) \
|
#define HWY_DISPATCH_TABLE(FUNC_NAME) \
|
||||||
HWY_CONCAT(FUNC_NAME, HighwayDispatchTable)
|
HWY_CONCAT(FUNC_NAME, HighwayDispatchTable)
|
||||||
|
|
||||||
|
@ -215,7 +251,7 @@ FunctionCache<RetType, Args...> FunctionCacheFactory(RetType (*)(Args...)) {
|
||||||
// static array must be defined at the same namespace level as the function
|
// static array must be defined at the same namespace level as the function
|
||||||
// it is exporting.
|
// it is exporting.
|
||||||
// After being exported, it can be called from other parts of the same source
|
// After being exported, it can be called from other parts of the same source
|
||||||
// file using HWY_DYNAMIC_DISTPATCH(), in particular from a function wrapper
|
// file using HWY_DYNAMIC_DISPATCH(), in particular from a function wrapper
|
||||||
// like in the following example:
|
// like in the following example:
|
||||||
//
|
//
|
||||||
// #include "hwy/highway.h"
|
// #include "hwy/highway.h"
|
||||||
|
@ -246,25 +282,43 @@ FunctionCache<RetType, Args...> FunctionCacheFactory(RetType (*)(Args...)) {
|
||||||
// same compile error conditions as with the dynamic dispatch case when multiple
|
// same compile error conditions as with the dynamic dispatch case when multiple
|
||||||
// targets are being compiled.
|
// targets are being compiled.
|
||||||
#define HWY_EXPORT(FUNC_NAME) \
|
#define HWY_EXPORT(FUNC_NAME) \
|
||||||
HWY_MAYBE_UNUSED static decltype(&HWY_STATIC_DISPATCH(FUNC_NAME)) \
|
HWY_MAYBE_UNUSED static decltype(&HWY_STATIC_DISPATCH(FUNC_NAME)) const \
|
||||||
const HWY_DISPATCH_TABLE(FUNC_NAME)[1] = { \
|
HWY_DISPATCH_TABLE(FUNC_NAME)[1] = {&HWY_STATIC_DISPATCH(FUNC_NAME)}
|
||||||
&HWY_STATIC_DISPATCH(FUNC_NAME)}
|
|
||||||
#define HWY_DYNAMIC_DISPATCH(FUNC_NAME) HWY_STATIC_DISPATCH(FUNC_NAME)
|
#define HWY_DYNAMIC_DISPATCH(FUNC_NAME) HWY_STATIC_DISPATCH(FUNC_NAME)
|
||||||
|
|
||||||
#else
|
#else
|
||||||
|
|
||||||
// Dynamic dispatch case with one entry per dynamic target plus the scalar
|
// Simplified version for MSVC 2017: function pointer instead of table.
|
||||||
// mode and the initialization wrapper.
|
#if HWY_DISPATCH_WORKAROUND
|
||||||
|
|
||||||
#define HWY_EXPORT(FUNC_NAME) \
|
#define HWY_EXPORT(FUNC_NAME) \
|
||||||
static decltype(&HWY_STATIC_DISPATCH(FUNC_NAME)) \
|
static decltype(&HWY_STATIC_DISPATCH(FUNC_NAME)) const HWY_DISPATCH_TABLE( \
|
||||||
const HWY_DISPATCH_TABLE(FUNC_NAME)[HWY_MAX_DYNAMIC_TARGETS + 2] = { \
|
FUNC_NAME)[HWY_MAX_DYNAMIC_TARGETS + 2] = { \
|
||||||
|
/* The first entry in the table initializes the global cache and \
|
||||||
|
* calls the function from HWY_STATIC_TARGET. */ \
|
||||||
|
&decltype(hwy::DeduceFunctionCache(&HWY_STATIC_DISPATCH( \
|
||||||
|
FUNC_NAME)))::ChooseAndCall<&HWY_STATIC_DISPATCH(FUNC_NAME)>, \
|
||||||
|
HWY_CHOOSE_TARGET_LIST(FUNC_NAME), \
|
||||||
|
HWY_CHOOSE_FALLBACK(FUNC_NAME), \
|
||||||
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
// Dynamic dispatch case with one entry per dynamic target plus the fallback
|
||||||
|
// target and the initialization wrapper.
|
||||||
|
#define HWY_EXPORT(FUNC_NAME) \
|
||||||
|
static decltype(&HWY_STATIC_DISPATCH(FUNC_NAME)) const HWY_DISPATCH_TABLE( \
|
||||||
|
FUNC_NAME)[HWY_MAX_DYNAMIC_TARGETS + 2] = { \
|
||||||
/* The first entry in the table initializes the global cache and \
|
/* The first entry in the table initializes the global cache and \
|
||||||
* calls the appropriate function. */ \
|
* calls the appropriate function. */ \
|
||||||
&decltype(hwy::FunctionCacheFactory(&HWY_STATIC_DISPATCH( \
|
&decltype(hwy::DeduceFunctionCache(&HWY_STATIC_DISPATCH( \
|
||||||
FUNC_NAME)))::ChooseAndCall<HWY_DISPATCH_TABLE(FUNC_NAME)>, \
|
FUNC_NAME)))::ChooseAndCall<HWY_DISPATCH_TABLE(FUNC_NAME)>, \
|
||||||
HWY_CHOOSE_TARGET_LIST(FUNC_NAME), \
|
HWY_CHOOSE_TARGET_LIST(FUNC_NAME), \
|
||||||
HWY_CHOOSE_SCALAR(FUNC_NAME), \
|
HWY_CHOOSE_FALLBACK(FUNC_NAME), \
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#endif // HWY_DISPATCH_WORKAROUND
|
||||||
|
|
||||||
#define HWY_DYNAMIC_DISPATCH(FUNC_NAME) \
|
#define HWY_DYNAMIC_DISPATCH(FUNC_NAME) \
|
||||||
(*(HWY_DISPATCH_TABLE(FUNC_NAME)[hwy::GetChosenTarget().GetIndex()]))
|
(*(HWY_DISPATCH_TABLE(FUNC_NAME)[hwy::GetChosenTarget().GetIndex()]))
|
||||||
|
|
||||||
|
@ -302,14 +356,17 @@ FunctionCache<RetType, Args...> FunctionCacheFactory(RetType (*)(Args...)) {
|
||||||
#error "PPC is not yet supported"
|
#error "PPC is not yet supported"
|
||||||
#elif HWY_TARGET == HWY_NEON
|
#elif HWY_TARGET == HWY_NEON
|
||||||
#include "hwy/ops/arm_neon-inl.h"
|
#include "hwy/ops/arm_neon-inl.h"
|
||||||
#elif HWY_TARGET == HWY_SVE || HWY_TARGET == HWY_SVE2
|
#elif HWY_TARGET == HWY_SVE || HWY_TARGET == HWY_SVE2 || \
|
||||||
|
HWY_TARGET == HWY_SVE_256 || HWY_TARGET == HWY_SVE2_128
|
||||||
#include "hwy/ops/arm_sve-inl.h"
|
#include "hwy/ops/arm_sve-inl.h"
|
||||||
#elif HWY_TARGET == HWY_WASM2
|
#elif HWY_TARGET == HWY_WASM_EMU256
|
||||||
#include "hwy/ops/wasm_256-inl.h"
|
#include "hwy/ops/wasm_256-inl.h"
|
||||||
#elif HWY_TARGET == HWY_WASM
|
#elif HWY_TARGET == HWY_WASM
|
||||||
#include "hwy/ops/wasm_128-inl.h"
|
#include "hwy/ops/wasm_128-inl.h"
|
||||||
#elif HWY_TARGET == HWY_RVV
|
#elif HWY_TARGET == HWY_RVV
|
||||||
#include "hwy/ops/rvv-inl.h"
|
#include "hwy/ops/rvv-inl.h"
|
||||||
|
#elif HWY_TARGET == HWY_EMU128
|
||||||
|
#include "hwy/ops/emu128-inl.h"
|
||||||
#elif HWY_TARGET == HWY_SCALAR
|
#elif HWY_TARGET == HWY_SCALAR
|
||||||
#include "hwy/ops/scalar-inl.h"
|
#include "hwy/ops/scalar-inl.h"
|
||||||
#else
|
#else
|
||||||
|
|
|
@ -9,19 +9,11 @@
|
||||||
#ifndef HWY_DLLEXPORT_H
|
#ifndef HWY_DLLEXPORT_H
|
||||||
#define HWY_DLLEXPORT_H
|
#define HWY_DLLEXPORT_H
|
||||||
|
|
||||||
// Bazel build are always static:
|
#if !defined(HWY_SHARED_DEFINE)
|
||||||
#if !defined(HWY_SHARED_DEFINE) && !defined(HWY_STATIC_DEFINE)
|
|
||||||
#define HWY_STATIC_DEFINE
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifdef HWY_STATIC_DEFINE
|
|
||||||
#define HWY_DLLEXPORT
|
#define HWY_DLLEXPORT
|
||||||
#define HWY_NO_EXPORT
|
|
||||||
#define HWY_CONTRIB_DLLEXPORT
|
#define HWY_CONTRIB_DLLEXPORT
|
||||||
#define HWY_CONTRIB_NO_EXPORT
|
|
||||||
#define HWY_TEST_DLLEXPORT
|
#define HWY_TEST_DLLEXPORT
|
||||||
#define HWY_TEST_NO_EXPORT
|
#else // !HWY_SHARED_DEFINE
|
||||||
#else
|
|
||||||
|
|
||||||
#ifndef HWY_DLLEXPORT
|
#ifndef HWY_DLLEXPORT
|
||||||
#if defined(hwy_EXPORTS)
|
#if defined(hwy_EXPORTS)
|
||||||
|
@ -31,23 +23,15 @@
|
||||||
#else
|
#else
|
||||||
#define HWY_DLLEXPORT __attribute__((visibility("default")))
|
#define HWY_DLLEXPORT __attribute__((visibility("default")))
|
||||||
#endif
|
#endif
|
||||||
#else
|
#else // defined(hwy_EXPORTS)
|
||||||
/* We are using this library */
|
/* We are using this library */
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
#define HWY_DLLEXPORT __declspec(dllimport)
|
#define HWY_DLLEXPORT __declspec(dllimport)
|
||||||
#else
|
#else
|
||||||
#define HWY_DLLEXPORT __attribute__((visibility("default")))
|
#define HWY_DLLEXPORT __attribute__((visibility("default")))
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif // defined(hwy_EXPORTS)
|
||||||
#endif
|
#endif // HWY_DLLEXPORT
|
||||||
|
|
||||||
#ifndef HWY_NO_EXPORT
|
|
||||||
#ifdef _WIN32
|
|
||||||
#define HWY_NO_EXPORT
|
|
||||||
#else
|
|
||||||
#define HWY_NO_EXPORT __attribute__((visibility("hidden")))
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef HWY_CONTRIB_DLLEXPORT
|
#ifndef HWY_CONTRIB_DLLEXPORT
|
||||||
#if defined(hwy_contrib_EXPORTS)
|
#if defined(hwy_contrib_EXPORTS)
|
||||||
|
@ -57,23 +41,15 @@
|
||||||
#else
|
#else
|
||||||
#define HWY_CONTRIB_DLLEXPORT __attribute__((visibility("default")))
|
#define HWY_CONTRIB_DLLEXPORT __attribute__((visibility("default")))
|
||||||
#endif
|
#endif
|
||||||
#else
|
#else // defined(hwy_contrib_EXPORTS)
|
||||||
/* We are using this library */
|
/* We are using this library */
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
#define HWY_CONTRIB_DLLEXPORT __declspec(dllimport)
|
#define HWY_CONTRIB_DLLEXPORT __declspec(dllimport)
|
||||||
#else
|
#else
|
||||||
#define HWY_CONTRIB_DLLEXPORT __attribute__((visibility("default")))
|
#define HWY_CONTRIB_DLLEXPORT __attribute__((visibility("default")))
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif // defined(hwy_contrib_EXPORTS)
|
||||||
#endif
|
#endif // HWY_CONTRIB_DLLEXPORT
|
||||||
|
|
||||||
#ifndef HWY_CONTRIB_NO_EXPORT
|
|
||||||
#ifdef _WIN32
|
|
||||||
#define HWY_CONTRIB_NO_EXPORT
|
|
||||||
#else
|
|
||||||
#define HWY_CONTRIB_NO_EXPORT __attribute__((visibility("hidden")))
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef HWY_TEST_DLLEXPORT
|
#ifndef HWY_TEST_DLLEXPORT
|
||||||
#if defined(hwy_test_EXPORTS)
|
#if defined(hwy_test_EXPORTS)
|
||||||
|
@ -83,24 +59,16 @@
|
||||||
#else
|
#else
|
||||||
#define HWY_TEST_DLLEXPORT __attribute__((visibility("default")))
|
#define HWY_TEST_DLLEXPORT __attribute__((visibility("default")))
|
||||||
#endif
|
#endif
|
||||||
#else
|
#else // defined(hwy_test_EXPORTS)
|
||||||
/* We are using this library */
|
/* We are using this library */
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
#define HWY_TEST_DLLEXPORT __declspec(dllimport)
|
#define HWY_TEST_DLLEXPORT __declspec(dllimport)
|
||||||
#else
|
#else
|
||||||
#define HWY_TEST_DLLEXPORT __attribute__((visibility("default")))
|
#define HWY_TEST_DLLEXPORT __attribute__((visibility("default")))
|
||||||
#endif
|
#endif
|
||||||
#endif
|
#endif // defined(hwy_test_EXPORTS)
|
||||||
#endif
|
#endif // HWY_TEST_DLLEXPORT
|
||||||
|
|
||||||
#ifndef HWY_TEST_NO_EXPORT
|
#endif // !HWY_SHARED_DEFINE
|
||||||
#ifdef _WIN32
|
|
||||||
#define HWY_TEST_NO_EXPORT
|
|
||||||
#else
|
|
||||||
#define HWY_TEST_NO_EXPORT __attribute__((visibility("hidden")))
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif /* HWY_DLLEXPORT_H */
|
#endif /* HWY_DLLEXPORT_H */
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -14,12 +15,15 @@
|
||||||
|
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
#include <bitset>
|
#include <bitset>
|
||||||
|
|
||||||
|
#include "hwy/base.h"
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "highway_test.cc"
|
#define HWY_TARGET_INCLUDE "highway_test.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
#include "hwy/nanobenchmark.h" // Unpredictable1
|
#include "hwy/nanobenchmark.h" // Unpredictable1
|
||||||
#include "hwy/tests/test_util-inl.h"
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
@ -28,6 +32,38 @@ HWY_BEFORE_NAMESPACE();
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace HWY_NAMESPACE {
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
template <size_t kLimit, typename T>
|
||||||
|
HWY_NOINLINE void TestCappedLimit(T /* tag */) {
|
||||||
|
CappedTag<T, kLimit> d;
|
||||||
|
// Ensure two ops compile
|
||||||
|
HWY_ASSERT_VEC_EQ(d, Zero(d), Set(d, T{0}));
|
||||||
|
|
||||||
|
// Ensure we do not write more than kLimit lanes
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
if (kLimit < N) {
|
||||||
|
auto lanes = AllocateAligned<T>(N);
|
||||||
|
std::fill(lanes.get(), lanes.get() + N, T{0});
|
||||||
|
Store(Set(d, T{1}), d, lanes.get());
|
||||||
|
for (size_t i = kLimit; i < N; ++i) {
|
||||||
|
HWY_ASSERT_EQ(lanes[i], T{0});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Adapter for ForAllTypes - we are constructing our own Simd<> and thus do not
|
||||||
|
// use ForPartialVectors etc.
|
||||||
|
struct TestCapped {
|
||||||
|
template <typename T>
|
||||||
|
void operator()(T t) const {
|
||||||
|
TestCappedLimit<1>(t);
|
||||||
|
TestCappedLimit<3>(t);
|
||||||
|
TestCappedLimit<5>(t);
|
||||||
|
TestCappedLimit<1ull << 15>(t);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
HWY_NOINLINE void TestAllCapped() { ForAllTypes(TestCapped()); }
|
||||||
|
|
||||||
// For testing that ForPartialVectors reaches every possible size:
|
// For testing that ForPartialVectors reaches every possible size:
|
||||||
using NumLanesSet = std::bitset<HWY_MAX_BYTES + 1>;
|
using NumLanesSet = std::bitset<HWY_MAX_BYTES + 1>;
|
||||||
|
|
||||||
|
@ -48,7 +84,7 @@ struct TestMaxLanes {
|
||||||
template <class T, class D>
|
template <class T, class D>
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
const size_t kMax = MaxLanes(d);
|
const size_t kMax = MaxLanes(d); // for RVV, includes LMUL
|
||||||
HWY_ASSERT(N <= kMax);
|
HWY_ASSERT(N <= kMax);
|
||||||
HWY_ASSERT(kMax <= (HWY_MAX_BYTES / sizeof(T)));
|
HWY_ASSERT(kMax <= (HWY_MAX_BYTES / sizeof(T)));
|
||||||
|
|
||||||
|
@ -175,26 +211,19 @@ HWY_NOINLINE void TestAllSignBit() {
|
||||||
ForFloatTypes(ForPartialVectors<TestSignBitFloat>());
|
ForFloatTypes(ForPartialVectors<TestSignBitFloat>());
|
||||||
}
|
}
|
||||||
|
|
||||||
// std::isnan returns false for 0x7F..FF in clang AVX3 builds, so DIY.
|
// inline to work around incorrect SVE codegen (only first 128 bits used).
|
||||||
template <typename TF>
|
|
||||||
bool IsNaN(TF f) {
|
|
||||||
MakeUnsigned<TF> bits;
|
|
||||||
memcpy(&bits, &f, sizeof(TF));
|
|
||||||
bits += bits;
|
|
||||||
bits >>= 1; // clear sign bit
|
|
||||||
// NaN if all exponent bits are set and the mantissa is not zero.
|
|
||||||
return bits > ExponentMask<decltype(bits)>();
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class D, class V>
|
template <class D, class V>
|
||||||
HWY_NOINLINE void AssertNaN(D d, VecArg<V> v, const char* file, int line) {
|
HWY_INLINE void AssertNaN(D d, VecArg<V> v, const char* file, int line) {
|
||||||
using T = TFromD<D>;
|
using T = TFromD<D>;
|
||||||
const T lane = GetLane(v);
|
const size_t N = Lanes(d);
|
||||||
if (!IsNaN(lane)) {
|
if (!AllTrue(d, IsNaN(v))) {
|
||||||
const std::string type_name = TypeName(T(), Lanes(d));
|
Print(d, "not all NaN", v, 0, N);
|
||||||
|
Print(d, "mask", VecFromMask(d, IsNaN(v)), 0, N);
|
||||||
|
const std::string type_name = TypeName(T(), N);
|
||||||
// RVV lacks PRIu64 and MSYS still has problems with %zu, so print bytes to
|
// RVV lacks PRIu64 and MSYS still has problems with %zu, so print bytes to
|
||||||
// avoid truncating doubles.
|
// avoid truncating doubles.
|
||||||
uint8_t bytes[HWY_MAX(sizeof(T), 8)] = {0};
|
uint8_t bytes[HWY_MAX(sizeof(T), 8)] = {0};
|
||||||
|
const T lane = GetLane(v);
|
||||||
memcpy(bytes, &lane, sizeof(T));
|
memcpy(bytes, &lane, sizeof(T));
|
||||||
Abort(file, line,
|
Abort(file, line,
|
||||||
"Expected %s NaN, got %E (bytes %02x %02x %02x %02x %02x %02x %02x "
|
"Expected %s NaN, got %E (bytes %02x %02x %02x %02x %02x %02x %02x "
|
||||||
|
@ -264,14 +293,14 @@ struct TestNaN {
|
||||||
|
|
||||||
// Reduction
|
// Reduction
|
||||||
HWY_ASSERT_NAN(d, SumOfLanes(d, nan));
|
HWY_ASSERT_NAN(d, SumOfLanes(d, nan));
|
||||||
// TODO(janwas): re-enable after QEMU is fixed
|
// TODO(janwas): re-enable after QEMU/Spike are fixed
|
||||||
#if HWY_TARGET != HWY_RVV
|
#if HWY_TARGET != HWY_RVV
|
||||||
HWY_ASSERT_NAN(d, MinOfLanes(d, nan));
|
HWY_ASSERT_NAN(d, MinOfLanes(d, nan));
|
||||||
HWY_ASSERT_NAN(d, MaxOfLanes(d, nan));
|
HWY_ASSERT_NAN(d, MaxOfLanes(d, nan));
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Min
|
// Min
|
||||||
#if HWY_ARCH_X86 && HWY_TARGET != HWY_SCALAR
|
#if HWY_ARCH_X86 && (HWY_TARGET != HWY_SCALAR && HWY_TARGET != HWY_EMU128)
|
||||||
// x86 SIMD returns the second operand if any input is NaN.
|
// x86 SIMD returns the second operand if any input is NaN.
|
||||||
HWY_ASSERT_VEC_EQ(d, v1, Min(nan, v1));
|
HWY_ASSERT_VEC_EQ(d, v1, Min(nan, v1));
|
||||||
HWY_ASSERT_VEC_EQ(d, v1, Max(nan, v1));
|
HWY_ASSERT_VEC_EQ(d, v1, Max(nan, v1));
|
||||||
|
@ -317,6 +346,74 @@ HWY_NOINLINE void TestAllNaN() {
|
||||||
ForPartialVectors<TestF32NaN>()(float());
|
ForPartialVectors<TestF32NaN>()(float());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct TestIsNaN {
|
||||||
|
template <class T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
|
const auto v1 = Set(d, T(Unpredictable1()));
|
||||||
|
const auto inf = IfThenElse(Eq(v1, Set(d, T(1))), Inf(d), v1);
|
||||||
|
const auto nan = IfThenElse(Eq(v1, Set(d, T(1))), NaN(d), v1);
|
||||||
|
const auto neg = Set(d, T{-1});
|
||||||
|
HWY_ASSERT_NAN(d, nan);
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsNaN(inf));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsNaN(CopySign(inf, neg)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsNaN(nan));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsNaN(CopySign(nan, neg)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsNaN(v1));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsNaN(Zero(d)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsNaN(Set(d, hwy::LowestValue<T>())));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsNaN(Set(d, hwy::HighestValue<T>())));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
HWY_NOINLINE void TestAllIsNaN() {
|
||||||
|
ForFloatTypes(ForPartialVectors<TestIsNaN>());
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TestIsInf {
|
||||||
|
template <class T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
|
const auto v1 = Set(d, T(Unpredictable1()));
|
||||||
|
const auto inf = IfThenElse(Eq(v1, Set(d, T(1))), Inf(d), v1);
|
||||||
|
const auto nan = IfThenElse(Eq(v1, Set(d, T(1))), NaN(d), v1);
|
||||||
|
const auto neg = Set(d, T{-1});
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsInf(inf));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsInf(CopySign(inf, neg)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsInf(nan));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsInf(CopySign(nan, neg)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsInf(v1));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsInf(Zero(d)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsInf(Set(d, hwy::LowestValue<T>())));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsInf(Set(d, hwy::HighestValue<T>())));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
HWY_NOINLINE void TestAllIsInf() {
|
||||||
|
ForFloatTypes(ForPartialVectors<TestIsInf>());
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TestIsFinite {
|
||||||
|
template <class T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
|
const auto v1 = Set(d, T(Unpredictable1()));
|
||||||
|
const auto inf = IfThenElse(Eq(v1, Set(d, T(1))), Inf(d), v1);
|
||||||
|
const auto nan = IfThenElse(Eq(v1, Set(d, T(1))), NaN(d), v1);
|
||||||
|
const auto neg = Set(d, T{-1});
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsFinite(inf));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsFinite(CopySign(inf, neg)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsFinite(nan));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskFalse(d), IsFinite(CopySign(nan, neg)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsFinite(v1));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsFinite(Zero(d)));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d), IsFinite(Set(d, hwy::LowestValue<T>())));
|
||||||
|
HWY_ASSERT_MASK_EQ(d, MaskTrue(d),
|
||||||
|
IsFinite(Set(d, hwy::HighestValue<T>())));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
HWY_NOINLINE void TestAllIsFinite() {
|
||||||
|
ForFloatTypes(ForPartialVectors<TestIsFinite>());
|
||||||
|
}
|
||||||
|
|
||||||
struct TestCopyAndAssign {
|
struct TestCopyAndAssign {
|
||||||
template <class T, class D>
|
template <class T, class D>
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
|
@ -371,21 +468,19 @@ HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
HWY_BEFORE_TEST(HighwayTest);
|
HWY_BEFORE_TEST(HighwayTest);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllCapped);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllMaxLanes);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllMaxLanes);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllSet);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllSet);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllOverflow);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllOverflow);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllClamp);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllClamp);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllSignBit);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllSignBit);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllNaN);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllNaN);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllIsNaN);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllIsInf);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllIsFinite);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllCopyAndAssign);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllCopyAndAssign);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllGetLane);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllGetLane);
|
||||||
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllDFromV);
|
HWY_EXPORT_AND_TEST_P(HighwayTest, TestAllDFromV);
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -24,6 +25,7 @@
|
||||||
#include <algorithm> // sort
|
#include <algorithm> // sort
|
||||||
#include <array>
|
#include <array>
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
#include <chrono> //NOLINT
|
||||||
#include <limits>
|
#include <limits>
|
||||||
#include <numeric> // iota
|
#include <numeric> // iota
|
||||||
#include <random>
|
#include <random>
|
||||||
|
@ -122,6 +124,9 @@ inline Ticks Start() {
|
||||||
Ticks t;
|
Ticks t;
|
||||||
#if HWY_ARCH_PPC && defined(__GLIBC__)
|
#if HWY_ARCH_PPC && defined(__GLIBC__)
|
||||||
asm volatile("mfspr %0, %1" : "=r"(t) : "i"(268));
|
asm volatile("mfspr %0, %1" : "=r"(t) : "i"(268));
|
||||||
|
#elif HWY_ARCH_ARM_A64 && !HWY_COMPILER_MSVC
|
||||||
|
// pmccntr_el0 is privileged but cntvct_el0 is accessible in Linux and QEMU.
|
||||||
|
asm volatile("mrs %0, cntvct_el0" : "=r"(t));
|
||||||
#elif HWY_ARCH_X86 && HWY_COMPILER_MSVC
|
#elif HWY_ARCH_X86 && HWY_COMPILER_MSVC
|
||||||
_ReadWriteBarrier();
|
_ReadWriteBarrier();
|
||||||
_mm_lfence();
|
_mm_lfence();
|
||||||
|
@ -160,10 +165,14 @@ inline Ticks Start() {
|
||||||
return t;
|
return t;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// WARNING: on x86, caller must check HasRDTSCP before using this!
|
||||||
inline Ticks Stop() {
|
inline Ticks Stop() {
|
||||||
uint64_t t;
|
uint64_t t;
|
||||||
#if HWY_ARCH_PPC && defined(__GLIBC__)
|
#if HWY_ARCH_PPC && defined(__GLIBC__)
|
||||||
asm volatile("mfspr %0, %1" : "=r"(t) : "i"(268));
|
asm volatile("mfspr %0, %1" : "=r"(t) : "i"(268));
|
||||||
|
#elif HWY_ARCH_ARM_A64 && !HWY_COMPILER_MSVC
|
||||||
|
// pmccntr_el0 is privileged but cntvct_el0 is accessible in Linux and QEMU.
|
||||||
|
asm volatile("mrs %0, cntvct_el0" : "=r"(t));
|
||||||
#elif HWY_ARCH_X86 && HWY_COMPILER_MSVC
|
#elif HWY_ARCH_X86 && HWY_COMPILER_MSVC
|
||||||
_ReadWriteBarrier();
|
_ReadWriteBarrier();
|
||||||
unsigned aux;
|
unsigned aux;
|
||||||
|
@ -302,7 +311,8 @@ T MedianAbsoluteDeviation(const T* values, const size_t num_values,
|
||||||
std::vector<T> abs_deviations;
|
std::vector<T> abs_deviations;
|
||||||
abs_deviations.reserve(num_values);
|
abs_deviations.reserve(num_values);
|
||||||
for (size_t i = 0; i < num_values; ++i) {
|
for (size_t i = 0; i < num_values; ++i) {
|
||||||
const int64_t abs = std::abs(int64_t(values[i]) - int64_t(median));
|
const int64_t abs = std::abs(static_cast<int64_t>(values[i]) -
|
||||||
|
static_cast<int64_t>(median));
|
||||||
abs_deviations.push_back(static_cast<T>(abs));
|
abs_deviations.push_back(static_cast<T>(abs));
|
||||||
}
|
}
|
||||||
return Median(abs_deviations.data(), num_values);
|
return Median(abs_deviations.data(), num_values);
|
||||||
|
@ -331,6 +341,38 @@ inline void PreventElision(T&& output) {
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Measures the actual current frequency of Ticks. We cannot rely on the nominal
|
||||||
|
// frequency encoded in x86 BrandString because it is misleading on M1 Rosetta,
|
||||||
|
// and not reported by AMD. CPUID 0x15 is also not yet widely supported. Also
|
||||||
|
// used on RISC-V and ARM64.
|
||||||
|
HWY_MAYBE_UNUSED double MeasureNominalClockRate() {
|
||||||
|
double max_ticks_per_sec = 0.0;
|
||||||
|
// Arbitrary, enough to ignore 2 outliers without excessive init time.
|
||||||
|
for (int rep = 0; rep < 3; ++rep) {
|
||||||
|
auto time0 = std::chrono::steady_clock::now();
|
||||||
|
using Time = decltype(time0);
|
||||||
|
const timer::Ticks ticks0 = timer::Start();
|
||||||
|
const Time time_min = time0 + std::chrono::milliseconds(10);
|
||||||
|
|
||||||
|
Time time1;
|
||||||
|
timer::Ticks ticks1;
|
||||||
|
for (;;) {
|
||||||
|
time1 = std::chrono::steady_clock::now();
|
||||||
|
// Ideally this would be Stop, but that requires RDTSCP on x86. To avoid
|
||||||
|
// another codepath, just use Start instead. now() presumably has its own
|
||||||
|
// fence-like behavior.
|
||||||
|
ticks1 = timer::Start(); // Do not use Stop, see comment above
|
||||||
|
if (time1 >= time_min) break;
|
||||||
|
}
|
||||||
|
|
||||||
|
const double dticks = static_cast<double>(ticks1 - ticks0);
|
||||||
|
std::chrono::duration<double, std::ratio<1>> dtime = time1 - time0;
|
||||||
|
const double ticks_per_sec = dticks / dtime.count();
|
||||||
|
max_ticks_per_sec = std::max(max_ticks_per_sec, ticks_per_sec);
|
||||||
|
}
|
||||||
|
return max_ticks_per_sec;
|
||||||
|
}
|
||||||
|
|
||||||
#if HWY_ARCH_X86
|
#if HWY_ARCH_X86
|
||||||
|
|
||||||
void Cpuid(const uint32_t level, const uint32_t count,
|
void Cpuid(const uint32_t level, const uint32_t count,
|
||||||
|
@ -378,50 +420,27 @@ std::string BrandString() {
|
||||||
return brand_string;
|
return brand_string;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns the frequency quoted inside the brand string. This does not
|
|
||||||
// account for throttling nor Turbo Boost.
|
|
||||||
double NominalClockRate() {
|
|
||||||
const std::string& brand_string = BrandString();
|
|
||||||
// Brand strings include the maximum configured frequency. These prefixes are
|
|
||||||
// defined by Intel CPUID documentation.
|
|
||||||
const char* prefixes[3] = {"MHz", "GHz", "THz"};
|
|
||||||
const double multipliers[3] = {1E6, 1E9, 1E12};
|
|
||||||
for (size_t i = 0; i < 3; ++i) {
|
|
||||||
const size_t pos_prefix = brand_string.find(prefixes[i]);
|
|
||||||
if (pos_prefix != std::string::npos) {
|
|
||||||
const size_t pos_space = brand_string.rfind(' ', pos_prefix - 1);
|
|
||||||
if (pos_space != std::string::npos) {
|
|
||||||
const std::string digits =
|
|
||||||
brand_string.substr(pos_space + 1, pos_prefix - pos_space - 1);
|
|
||||||
return std::stod(digits) * multipliers[i];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0.0;
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // HWY_ARCH_X86
|
#endif // HWY_ARCH_X86
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
HWY_DLLEXPORT double InvariantTicksPerSecond() {
|
HWY_DLLEXPORT double InvariantTicksPerSecond() {
|
||||||
#if HWY_ARCH_PPC && defined(__GLIBC__)
|
#if HWY_ARCH_PPC && defined(__GLIBC__)
|
||||||
return double(__ppc_get_timebase_freq());
|
return static_cast<double>(__ppc_get_timebase_freq());
|
||||||
#elif HWY_ARCH_X86
|
#elif HWY_ARCH_X86 || HWY_ARCH_RVV || (HWY_ARCH_ARM_A64 && !HWY_COMPILER_MSVC)
|
||||||
// We assume the TSC is invariant; it is on all recent Intel/AMD CPUs.
|
// We assume the x86 TSC is invariant; it is on all recent Intel/AMD CPUs.
|
||||||
return NominalClockRate();
|
static const double freq = MeasureNominalClockRate();
|
||||||
|
return freq;
|
||||||
#elif defined(_WIN32) || defined(_WIN64)
|
#elif defined(_WIN32) || defined(_WIN64)
|
||||||
LARGE_INTEGER freq;
|
LARGE_INTEGER freq;
|
||||||
(void)QueryPerformanceFrequency(&freq);
|
(void)QueryPerformanceFrequency(&freq);
|
||||||
return double(freq.QuadPart);
|
return static_cast<double>(freq.QuadPart);
|
||||||
#elif defined(__APPLE__)
|
#elif defined(__APPLE__)
|
||||||
// https://developer.apple.com/library/mac/qa/qa1398/_index.html
|
// https://developer.apple.com/library/mac/qa/qa1398/_index.html
|
||||||
mach_timebase_info_data_t timebase;
|
mach_timebase_info_data_t timebase;
|
||||||
(void)mach_timebase_info(&timebase);
|
(void)mach_timebase_info(&timebase);
|
||||||
return double(timebase.denom) / timebase.numer * 1E9;
|
return static_cast<double>(timebase.denom) / timebase.numer * 1E9;
|
||||||
#else
|
#else
|
||||||
// TODO(janwas): ARM? Unclear how to reliably query cntvct_el0 frequency.
|
|
||||||
return 1E9; // Haiku and clock_gettime return nanoseconds.
|
return 1E9; // Haiku and clock_gettime return nanoseconds.
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
@ -432,15 +451,29 @@ HWY_DLLEXPORT double Now() {
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_DLLEXPORT uint64_t TimerResolution() {
|
HWY_DLLEXPORT uint64_t TimerResolution() {
|
||||||
|
#if HWY_ARCH_X86
|
||||||
|
bool can_use_stop = platform::HasRDTSCP();
|
||||||
|
#else
|
||||||
|
constexpr bool can_use_stop = true;
|
||||||
|
#endif
|
||||||
|
|
||||||
// Nested loop avoids exceeding stack/L1 capacity.
|
// Nested loop avoids exceeding stack/L1 capacity.
|
||||||
timer::Ticks repetitions[Params::kTimerSamples];
|
timer::Ticks repetitions[Params::kTimerSamples];
|
||||||
for (size_t rep = 0; rep < Params::kTimerSamples; ++rep) {
|
for (size_t rep = 0; rep < Params::kTimerSamples; ++rep) {
|
||||||
timer::Ticks samples[Params::kTimerSamples];
|
timer::Ticks samples[Params::kTimerSamples];
|
||||||
|
if (can_use_stop) {
|
||||||
for (size_t i = 0; i < Params::kTimerSamples; ++i) {
|
for (size_t i = 0; i < Params::kTimerSamples; ++i) {
|
||||||
const timer::Ticks t0 = timer::Start();
|
const timer::Ticks t0 = timer::Start();
|
||||||
const timer::Ticks t1 = timer::Stop();
|
const timer::Ticks t1 = timer::Stop(); // we checked HasRDTSCP above
|
||||||
samples[i] = t1 - t0;
|
samples[i] = t1 - t0;
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
for (size_t i = 0; i < Params::kTimerSamples; ++i) {
|
||||||
|
const timer::Ticks t0 = timer::Start();
|
||||||
|
const timer::Ticks t1 = timer::Start(); // do not use Stop, see above
|
||||||
|
samples[i] = t1 - t0;
|
||||||
|
}
|
||||||
|
}
|
||||||
repetitions[rep] = robust_statistics::Mode(samples);
|
repetitions[rep] = robust_statistics::Mode(samples);
|
||||||
}
|
}
|
||||||
return robust_statistics::Mode(repetitions);
|
return robust_statistics::Mode(repetitions);
|
||||||
|
@ -459,7 +492,7 @@ timer::Ticks SampleUntilStable(const double max_rel_mad, double* rel_mad,
|
||||||
// Choose initial samples_per_eval based on a single estimated duration.
|
// Choose initial samples_per_eval based on a single estimated duration.
|
||||||
timer::Ticks t0 = timer::Start();
|
timer::Ticks t0 = timer::Start();
|
||||||
lambda();
|
lambda();
|
||||||
timer::Ticks t1 = timer::Stop();
|
timer::Ticks t1 = timer::Stop(); // Caller checks HasRDTSCP
|
||||||
timer::Ticks est = t1 - t0;
|
timer::Ticks est = t1 - t0;
|
||||||
static const double ticks_per_second = platform::InvariantTicksPerSecond();
|
static const double ticks_per_second = platform::InvariantTicksPerSecond();
|
||||||
const size_t ticks_per_eval =
|
const size_t ticks_per_eval =
|
||||||
|
@ -483,7 +516,7 @@ timer::Ticks SampleUntilStable(const double max_rel_mad, double* rel_mad,
|
||||||
for (size_t i = 0; i < samples_per_eval; ++i) {
|
for (size_t i = 0; i < samples_per_eval; ++i) {
|
||||||
t0 = timer::Start();
|
t0 = timer::Start();
|
||||||
lambda();
|
lambda();
|
||||||
t1 = timer::Stop();
|
t1 = timer::Stop(); // Caller checks HasRDTSCP
|
||||||
samples.push_back(t1 - t0);
|
samples.push_back(t1 - t0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -57,7 +58,7 @@ std::mt19937 rng;
|
||||||
// A function whose runtime depends on rng.
|
// A function whose runtime depends on rng.
|
||||||
FuncOutput Random(const void* /*arg*/, FuncInput in) {
|
FuncOutput Random(const void* /*arg*/, FuncInput in) {
|
||||||
const size_t r = rng() & 0xF;
|
const size_t r = rng() & 0xF;
|
||||||
uint32_t ret = in;
|
FuncOutput ret = static_cast<FuncOutput>(in);
|
||||||
for (size_t i = 0; i < r; ++i) {
|
for (size_t i = 0; i < r; ++i) {
|
||||||
ret /= ((rng() & 1) + 2);
|
ret /= ((rng() & 1) + 2);
|
||||||
}
|
}
|
||||||
|
@ -88,9 +89,3 @@ TEST(NanobenchmarkTest, RunAll) {
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -58,9 +59,8 @@ HWY_API V CombineShiftRightLanes(D d, const V hi, const V lo) {
|
||||||
// Returns lanes with the most significant bit set and all other bits zero.
|
// Returns lanes with the most significant bit set and all other bits zero.
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_API Vec<D> SignBit(D d) {
|
HWY_API Vec<D> SignBit(D d) {
|
||||||
using Unsigned = MakeUnsigned<TFromD<D>>;
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
const Unsigned bit = Unsigned(1) << (sizeof(Unsigned) * 8 - 1);
|
return BitCast(d, Set(du, SignMask<TFromD<D>>()));
|
||||||
return BitCast(d, Set(Rebind<Unsigned, D>(), bit));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Returns quiet NaN.
|
// Returns quiet NaN.
|
||||||
|
@ -72,6 +72,906 @@ HWY_API Vec<D> NaN(D d) {
|
||||||
return BitCast(d, Set(di, LimitsMax<TFromD<decltype(di)>>()));
|
return BitCast(d, Set(di, LimitsMax<TFromD<decltype(di)>>()));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Returns positive infinity.
|
||||||
|
template <class D>
|
||||||
|
HWY_API Vec<D> Inf(D d) {
|
||||||
|
const RebindToUnsigned<D> du;
|
||||||
|
using T = TFromD<D>;
|
||||||
|
using TU = TFromD<decltype(du)>;
|
||||||
|
const TU max_x2 = static_cast<TU>(MaxExponentTimes2<T>());
|
||||||
|
return BitCast(d, Set(du, max_x2 >> 1));
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ SafeFillN
|
||||||
|
|
||||||
|
template <class D, typename T = TFromD<D>>
|
||||||
|
HWY_API void SafeFillN(const size_t num, const T value, D d,
|
||||||
|
T* HWY_RESTRICT to) {
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
(void)d;
|
||||||
|
for (size_t i = 0; i < num; ++i) {
|
||||||
|
to[i] = value;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
BlendedStore(Set(d, value), FirstN(d, num), d, to);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ SafeCopyN
|
||||||
|
|
||||||
|
template <class D, typename T = TFromD<D>>
|
||||||
|
HWY_API void SafeCopyN(const size_t num, D d, const T* HWY_RESTRICT from,
|
||||||
|
T* HWY_RESTRICT to) {
|
||||||
|
#if HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
(void)d;
|
||||||
|
for (size_t i = 0; i < num; ++i) {
|
||||||
|
to[i] = from[i];
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
const Mask<D> mask = FirstN(d, num);
|
||||||
|
BlendedStore(MaskedLoad(mask, d, from), mask, d, to);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// "Include guard": skip if native instructions are available. The generic
|
||||||
|
// implementation is currently shared between x86_* and wasm_*, and is too large
|
||||||
|
// to duplicate.
|
||||||
|
|
||||||
|
#if (defined(HWY_NATIVE_LOAD_STORE_INTERLEAVED) == defined(HWY_TARGET_TOGGLE))
|
||||||
|
#ifdef HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
#undef HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
#else
|
||||||
|
#define HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// ------------------------------ LoadInterleaved2
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V>
|
||||||
|
HWY_API void LoadInterleaved2(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1) {
|
||||||
|
const V A = LoadU(d, unaligned + 0 * N); // v1[1] v0[1] v1[0] v0[0]
|
||||||
|
const V B = LoadU(d, unaligned + 1 * N);
|
||||||
|
v0 = ConcatEven(d, B, A);
|
||||||
|
v1 = ConcatOdd(d, B, A);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, class V>
|
||||||
|
HWY_API void LoadInterleaved2(Simd<T, 1, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1) {
|
||||||
|
v0 = LoadU(d, unaligned + 0);
|
||||||
|
v1 = LoadU(d, unaligned + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ LoadInterleaved3 (CombineShiftRightBytes)
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// Default for <= 128-bit vectors; x86_256 and x86_512 have their own overload.
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LE128(T, N)>
|
||||||
|
HWY_API void LoadTransposedBlocks3(Simd<T, N, 0> d,
|
||||||
|
const T* HWY_RESTRICT unaligned, V& A, V& B,
|
||||||
|
V& C) {
|
||||||
|
A = LoadU(d, unaligned + 0 * N);
|
||||||
|
B = LoadU(d, unaligned + 1 * N);
|
||||||
|
C = LoadU(d, unaligned + 2 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 16)>
|
||||||
|
HWY_API void LoadInterleaved3(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2) {
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
// Compact notation so these fit on one line: 12 := v1[2].
|
||||||
|
V A; // 05 24 14 04 23 13 03 22 12 02 21 11 01 20 10 00
|
||||||
|
V B; // 1a 0a 29 19 09 28 18 08 27 17 07 26 16 06 25 15
|
||||||
|
V C; // 2f 1f 0f 2e 1e 0e 2d 1d 0d 2c 1c 0c 2b 1b 0b 2a
|
||||||
|
detail::LoadTransposedBlocks3(d, unaligned, A, B, C);
|
||||||
|
// Compress all lanes belonging to v0 into consecutive lanes.
|
||||||
|
constexpr uint8_t Z = 0x80;
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v0A[16] = {0, 3, 6, 9, 12, 15, Z, Z,
|
||||||
|
Z, Z, Z, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v0B[16] = {Z, Z, Z, Z, Z, Z, 2, 5,
|
||||||
|
8, 11, 14, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v0C[16] = {Z, Z, Z, Z, Z, Z, Z, Z,
|
||||||
|
Z, Z, Z, 1, 4, 7, 10, 13};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v1A[16] = {1, 4, 7, 10, 13, Z, Z, Z,
|
||||||
|
Z, Z, Z, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v1B[16] = {Z, Z, Z, Z, Z, 0, 3, 6,
|
||||||
|
9, 12, 15, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v1C[16] = {Z, Z, Z, Z, Z, Z, Z, Z,
|
||||||
|
Z, Z, Z, 2, 5, 8, 11, 14};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v2A[16] = {2, 5, 8, 11, 14, Z, Z, Z,
|
||||||
|
Z, Z, Z, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v2B[16] = {Z, Z, Z, Z, Z, 1, 4, 7,
|
||||||
|
10, 13, Z, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v2C[16] = {Z, Z, Z, Z, Z, Z, Z, Z,
|
||||||
|
Z, Z, 0, 3, 6, 9, 12, 15};
|
||||||
|
const V v0L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v0A)));
|
||||||
|
const V v0M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v0B)));
|
||||||
|
const V v0U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v0C)));
|
||||||
|
const V v1L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v1A)));
|
||||||
|
const V v1M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v1B)));
|
||||||
|
const V v1U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v1C)));
|
||||||
|
const V v2L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v2A)));
|
||||||
|
const V v2M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v2B)));
|
||||||
|
const V v2U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v2C)));
|
||||||
|
v0 = Or3(v0L, v0M, v0U);
|
||||||
|
v1 = Or3(v1L, v1M, v1U);
|
||||||
|
v2 = Or3(v2L, v2M, v2U);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 8-bit lanes x8
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 1),
|
||||||
|
HWY_IF_LANES_PER_BLOCK(T, N, 8)>
|
||||||
|
HWY_API void LoadInterleaved3(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2) {
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
V A; // v1[2] v0[2] v2[1] v1[1] v0[1] v2[0] v1[0] v0[0]
|
||||||
|
V B; // v0[5] v2[4] v1[4] v0[4] v2[3] v1[3] v0[3] v2[2]
|
||||||
|
V C; // v2[7] v1[7] v0[7] v2[6] v1[6] v0[6] v2[5] v1[5]
|
||||||
|
detail::LoadTransposedBlocks3(d, unaligned, A, B, C);
|
||||||
|
// Compress all lanes belonging to v0 into consecutive lanes.
|
||||||
|
constexpr uint8_t Z = 0x80;
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v0A[16] = {0, 3, 6, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v0B[16] = {Z, Z, Z, 1, 4, 7, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v0C[16] = {Z, Z, Z, Z, Z, Z, 2, 5};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v1A[16] = {1, 4, 7, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v1B[16] = {Z, Z, Z, 2, 5, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v1C[16] = {Z, Z, Z, Z, Z, 0, 3, 6};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v2A[16] = {2, 5, Z, Z, Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v2B[16] = {Z, Z, 0, 3, 6, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint8_t kIdx_v2C[16] = {Z, Z, Z, Z, Z, 1, 4, 7};
|
||||||
|
const V v0L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v0A)));
|
||||||
|
const V v0M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v0B)));
|
||||||
|
const V v0U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v0C)));
|
||||||
|
const V v1L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v1A)));
|
||||||
|
const V v1M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v1B)));
|
||||||
|
const V v1U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v1C)));
|
||||||
|
const V v2L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v2A)));
|
||||||
|
const V v2M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v2B)));
|
||||||
|
const V v2U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v2C)));
|
||||||
|
v0 = Or3(v0L, v0M, v0U);
|
||||||
|
v1 = Or3(v1L, v1M, v1U);
|
||||||
|
v2 = Or3(v2L, v2M, v2U);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 16-bit lanes x8
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 2),
|
||||||
|
HWY_IF_LANES_PER_BLOCK(T, N, 8)>
|
||||||
|
HWY_API void LoadInterleaved3(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2) {
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
V A; // v1[2] v0[2] v2[1] v1[1] v0[1] v2[0] v1[0] v0[0]
|
||||||
|
V B; // v0[5] v2[4] v1[4] v0[4] v2[3] v1[3] v0[3] v2[2]
|
||||||
|
V C; // v2[7] v1[7] v0[7] v2[6] v1[6] v0[6] v2[5] v1[5]
|
||||||
|
detail::LoadTransposedBlocks3(d, unaligned, A, B, C);
|
||||||
|
// Compress all lanes belonging to v0 into consecutive lanes. Same as above,
|
||||||
|
// but each element of the array contains two byte indices for a lane.
|
||||||
|
constexpr uint16_t Z = 0x8080;
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v0A[8] = {0x0100, 0x0706, 0x0D0C, Z,
|
||||||
|
Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v0B[8] = {Z, Z, Z, 0x0302,
|
||||||
|
0x0908, 0x0F0E, Z, Z};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v0C[8] = {Z, Z, Z, Z,
|
||||||
|
Z, Z, 0x0504, 0x0B0A};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v1A[8] = {0x0302, 0x0908, 0x0F0E, Z,
|
||||||
|
Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v1B[8] = {Z, Z, Z, 0x0504,
|
||||||
|
0x0B0A, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v1C[8] = {Z, Z, Z, Z,
|
||||||
|
Z, 0x0100, 0x0706, 0x0D0C};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v2A[8] = {0x0504, 0x0B0A, Z, Z,
|
||||||
|
Z, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v2B[8] = {Z, Z, 0x0100, 0x0706,
|
||||||
|
0x0D0C, Z, Z, Z};
|
||||||
|
alignas(16) constexpr uint16_t kIdx_v2C[8] = {Z, Z, Z, Z,
|
||||||
|
Z, 0x0302, 0x0908, 0x0F0E};
|
||||||
|
const V v0L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v0A)));
|
||||||
|
const V v0M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v0B)));
|
||||||
|
const V v0U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v0C)));
|
||||||
|
const V v1L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v1A)));
|
||||||
|
const V v1M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v1B)));
|
||||||
|
const V v1U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v1C)));
|
||||||
|
const V v2L = BitCast(d, TableLookupBytesOr0(A, LoadDup128(du, kIdx_v2A)));
|
||||||
|
const V v2M = BitCast(d, TableLookupBytesOr0(B, LoadDup128(du, kIdx_v2B)));
|
||||||
|
const V v2U = BitCast(d, TableLookupBytesOr0(C, LoadDup128(du, kIdx_v2C)));
|
||||||
|
v0 = Or3(v0L, v0M, v0U);
|
||||||
|
v1 = Or3(v1L, v1M, v1U);
|
||||||
|
v2 = Or3(v2L, v2M, v2U);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 4)>
|
||||||
|
HWY_API void LoadInterleaved3(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2) {
|
||||||
|
V A; // v0[1] v2[0] v1[0] v0[0]
|
||||||
|
V B; // v1[2] v0[2] v2[1] v1[1]
|
||||||
|
V C; // v2[3] v1[3] v0[3] v2[2]
|
||||||
|
detail::LoadTransposedBlocks3(d, unaligned, A, B, C);
|
||||||
|
|
||||||
|
const V vxx_02_03_xx = OddEven(C, B);
|
||||||
|
v0 = detail::Shuffle1230(A, vxx_02_03_xx);
|
||||||
|
|
||||||
|
// Shuffle2301 takes the upper/lower halves of the output from one input, so
|
||||||
|
// we cannot just combine 13 and 10 with 12 and 11 (similar to v0/v2). Use
|
||||||
|
// OddEven because it may have higher throughput than Shuffle.
|
||||||
|
const V vxx_xx_10_11 = OddEven(A, B);
|
||||||
|
const V v12_13_xx_xx = OddEven(B, C);
|
||||||
|
v1 = detail::Shuffle2301(vxx_xx_10_11, v12_13_xx_xx);
|
||||||
|
|
||||||
|
const V vxx_20_21_xx = OddEven(B, A);
|
||||||
|
v2 = detail::Shuffle3012(vxx_20_21_xx, C);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 2)>
|
||||||
|
HWY_API void LoadInterleaved3(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2) {
|
||||||
|
V A; // v1[0] v0[0]
|
||||||
|
V B; // v0[1] v2[0]
|
||||||
|
V C; // v2[1] v1[1]
|
||||||
|
detail::LoadTransposedBlocks3(d, unaligned, A, B, C);
|
||||||
|
v0 = OddEven(B, A);
|
||||||
|
v1 = CombineShiftRightBytes<sizeof(T)>(d, C, A);
|
||||||
|
v2 = OddEven(C, B);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, class V>
|
||||||
|
HWY_API void LoadInterleaved3(Simd<T, 1, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2) {
|
||||||
|
v0 = LoadU(d, unaligned + 0);
|
||||||
|
v1 = LoadU(d, unaligned + 1);
|
||||||
|
v2 = LoadU(d, unaligned + 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ LoadInterleaved4
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// Default for <= 128-bit vectors; x86_256 and x86_512 have their own overload.
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LE128(T, N)>
|
||||||
|
HWY_API void LoadTransposedBlocks4(Simd<T, N, 0> d,
|
||||||
|
const T* HWY_RESTRICT unaligned, V& A, V& B,
|
||||||
|
V& C, V& D) {
|
||||||
|
A = LoadU(d, unaligned + 0 * N);
|
||||||
|
B = LoadU(d, unaligned + 1 * N);
|
||||||
|
C = LoadU(d, unaligned + 2 * N);
|
||||||
|
D = LoadU(d, unaligned + 3 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 16)>
|
||||||
|
HWY_API void LoadInterleaved4(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2, V& v3) {
|
||||||
|
const Repartition<uint64_t, decltype(d)> d64;
|
||||||
|
using V64 = VFromD<decltype(d64)>;
|
||||||
|
// 16 lanes per block; the lowest four blocks are at the bottom of A,B,C,D.
|
||||||
|
// Here int[i] means the four interleaved values of the i-th 4-tuple and
|
||||||
|
// int[3..0] indicates four consecutive 4-tuples (0 = least-significant).
|
||||||
|
V A; // int[13..10] int[3..0]
|
||||||
|
V B; // int[17..14] int[7..4]
|
||||||
|
V C; // int[1b..18] int[b..8]
|
||||||
|
V D; // int[1f..1c] int[f..c]
|
||||||
|
detail::LoadTransposedBlocks4(d, unaligned, A, B, C, D);
|
||||||
|
|
||||||
|
// For brevity, the comments only list the lower block (upper = lower + 0x10)
|
||||||
|
const V v5140 = InterleaveLower(d, A, B); // int[5,1,4,0]
|
||||||
|
const V vd9c8 = InterleaveLower(d, C, D); // int[d,9,c,8]
|
||||||
|
const V v7362 = InterleaveUpper(d, A, B); // int[7,3,6,2]
|
||||||
|
const V vfbea = InterleaveUpper(d, C, D); // int[f,b,e,a]
|
||||||
|
|
||||||
|
const V v6420 = InterleaveLower(d, v5140, v7362); // int[6,4,2,0]
|
||||||
|
const V veca8 = InterleaveLower(d, vd9c8, vfbea); // int[e,c,a,8]
|
||||||
|
const V v7531 = InterleaveUpper(d, v5140, v7362); // int[7,5,3,1]
|
||||||
|
const V vfdb9 = InterleaveUpper(d, vd9c8, vfbea); // int[f,d,b,9]
|
||||||
|
|
||||||
|
const V64 v10L = BitCast(d64, InterleaveLower(d, v6420, v7531)); // v10[7..0]
|
||||||
|
const V64 v10U = BitCast(d64, InterleaveLower(d, veca8, vfdb9)); // v10[f..8]
|
||||||
|
const V64 v32L = BitCast(d64, InterleaveUpper(d, v6420, v7531)); // v32[7..0]
|
||||||
|
const V64 v32U = BitCast(d64, InterleaveUpper(d, veca8, vfdb9)); // v32[f..8]
|
||||||
|
|
||||||
|
v0 = BitCast(d, InterleaveLower(d64, v10L, v10U));
|
||||||
|
v1 = BitCast(d, InterleaveUpper(d64, v10L, v10U));
|
||||||
|
v2 = BitCast(d, InterleaveLower(d64, v32L, v32U));
|
||||||
|
v3 = BitCast(d, InterleaveUpper(d64, v32L, v32U));
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 8)>
|
||||||
|
HWY_API void LoadInterleaved4(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2, V& v3) {
|
||||||
|
// In the last step, we interleave by half of the block size, which is usually
|
||||||
|
// 8 bytes but half that for 8-bit x8 vectors.
|
||||||
|
using TW = hwy::UnsignedFromSize<sizeof(T) * N == 8 ? 4 : 8>;
|
||||||
|
const Repartition<TW, decltype(d)> dw;
|
||||||
|
using VW = VFromD<decltype(dw)>;
|
||||||
|
|
||||||
|
// (Comments are for 256-bit vectors.)
|
||||||
|
// 8 lanes per block; the lowest four blocks are at the bottom of A,B,C,D.
|
||||||
|
V A; // v3210[9]v3210[8] v3210[1]v3210[0]
|
||||||
|
V B; // v3210[b]v3210[a] v3210[3]v3210[2]
|
||||||
|
V C; // v3210[d]v3210[c] v3210[5]v3210[4]
|
||||||
|
V D; // v3210[f]v3210[e] v3210[7]v3210[6]
|
||||||
|
detail::LoadTransposedBlocks4(d, unaligned, A, B, C, D);
|
||||||
|
|
||||||
|
const V va820 = InterleaveLower(d, A, B); // v3210[a,8] v3210[2,0]
|
||||||
|
const V vec64 = InterleaveLower(d, C, D); // v3210[e,c] v3210[6,4]
|
||||||
|
const V vb931 = InterleaveUpper(d, A, B); // v3210[b,9] v3210[3,1]
|
||||||
|
const V vfd75 = InterleaveUpper(d, C, D); // v3210[f,d] v3210[7,5]
|
||||||
|
|
||||||
|
const VW v10_b830 = // v10[b..8] v10[3..0]
|
||||||
|
BitCast(dw, InterleaveLower(d, va820, vb931));
|
||||||
|
const VW v10_fc74 = // v10[f..c] v10[7..4]
|
||||||
|
BitCast(dw, InterleaveLower(d, vec64, vfd75));
|
||||||
|
const VW v32_b830 = // v32[b..8] v32[3..0]
|
||||||
|
BitCast(dw, InterleaveUpper(d, va820, vb931));
|
||||||
|
const VW v32_fc74 = // v32[f..c] v32[7..4]
|
||||||
|
BitCast(dw, InterleaveUpper(d, vec64, vfd75));
|
||||||
|
|
||||||
|
v0 = BitCast(d, InterleaveLower(dw, v10_b830, v10_fc74));
|
||||||
|
v1 = BitCast(d, InterleaveUpper(dw, v10_b830, v10_fc74));
|
||||||
|
v2 = BitCast(d, InterleaveLower(dw, v32_b830, v32_fc74));
|
||||||
|
v3 = BitCast(d, InterleaveUpper(dw, v32_b830, v32_fc74));
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 4)>
|
||||||
|
HWY_API void LoadInterleaved4(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2, V& v3) {
|
||||||
|
V A; // v3210[4] v3210[0]
|
||||||
|
V B; // v3210[5] v3210[1]
|
||||||
|
V C; // v3210[6] v3210[2]
|
||||||
|
V D; // v3210[7] v3210[3]
|
||||||
|
detail::LoadTransposedBlocks4(d, unaligned, A, B, C, D);
|
||||||
|
const V v10_ev = InterleaveLower(d, A, C); // v1[6,4] v0[6,4] v1[2,0] v0[2,0]
|
||||||
|
const V v10_od = InterleaveLower(d, B, D); // v1[7,5] v0[7,5] v1[3,1] v0[3,1]
|
||||||
|
const V v32_ev = InterleaveUpper(d, A, C); // v3[6,4] v2[6,4] v3[2,0] v2[2,0]
|
||||||
|
const V v32_od = InterleaveUpper(d, B, D); // v3[7,5] v2[7,5] v3[3,1] v2[3,1]
|
||||||
|
|
||||||
|
v0 = InterleaveLower(d, v10_ev, v10_od);
|
||||||
|
v1 = InterleaveUpper(d, v10_ev, v10_od);
|
||||||
|
v2 = InterleaveLower(d, v32_ev, v32_od);
|
||||||
|
v3 = InterleaveUpper(d, v32_ev, v32_od);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANES_PER_BLOCK(T, N, 2)>
|
||||||
|
HWY_API void LoadInterleaved4(Simd<T, N, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2, V& v3) {
|
||||||
|
V A, B, C, D;
|
||||||
|
detail::LoadTransposedBlocks4(d, unaligned, A, B, C, D);
|
||||||
|
v0 = InterleaveLower(d, A, C);
|
||||||
|
v1 = InterleaveUpper(d, A, C);
|
||||||
|
v2 = InterleaveLower(d, B, D);
|
||||||
|
v3 = InterleaveUpper(d, B, D);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Any T x1
|
||||||
|
template <typename T, class V>
|
||||||
|
HWY_API void LoadInterleaved4(Simd<T, 1, 0> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
V& v0, V& v1, V& v2, V& v3) {
|
||||||
|
v0 = LoadU(d, unaligned + 0);
|
||||||
|
v1 = LoadU(d, unaligned + 1);
|
||||||
|
v2 = LoadU(d, unaligned + 2);
|
||||||
|
v3 = LoadU(d, unaligned + 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ StoreInterleaved2
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// Default for <= 128-bit vectors; x86_256 and x86_512 have their own overload.
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LE128(T, N)>
|
||||||
|
HWY_API void StoreTransposedBlocks2(const V A, const V B, Simd<T, N, 0> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
StoreU(A, d, unaligned + 0 * N);
|
||||||
|
StoreU(B, d, unaligned + 1 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
|
// >= 128 bit vector
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved2(const V v0, const V v1, Simd<T, N, 0> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
const auto v10L = InterleaveLower(d, v0, v1); // .. v1[0] v0[0]
|
||||||
|
const auto v10U = InterleaveUpper(d, v0, v1); // .. v1[N/2] v0[N/2]
|
||||||
|
detail::StoreTransposedBlocks2(v10L, v10U, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64 bits
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void StoreInterleaved2(const Vec64<T> part0, const Vec64<T> part1,
|
||||||
|
Full64<T> /*tag*/, T* HWY_RESTRICT unaligned) {
|
||||||
|
// Use full vectors to reduce the number of stores.
|
||||||
|
const Full128<T> d_full;
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const auto v10 = InterleaveLower(d_full, v0, v1);
|
||||||
|
StoreU(v10, d_full, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// <= 32 bits
|
||||||
|
template <typename T, size_t N, HWY_IF_LE32(T, N)>
|
||||||
|
HWY_API void StoreInterleaved2(const Vec128<T, N> part0,
|
||||||
|
const Vec128<T, N> part1, Simd<T, N, 0> /*tag*/,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
// Use full vectors to reduce the number of stores.
|
||||||
|
const Full128<T> d_full;
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const auto v10 = InterleaveLower(d_full, v0, v1);
|
||||||
|
alignas(16) T buf[16 / sizeof(T)];
|
||||||
|
StoreU(v10, d_full, buf);
|
||||||
|
CopyBytes<2 * N * sizeof(T)>(buf, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ StoreInterleaved3 (CombineShiftRightBytes,
|
||||||
|
// TableLookupBytes)
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// Default for <= 128-bit vectors; x86_256 and x86_512 have their own overload.
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LE128(T, N)>
|
||||||
|
HWY_API void StoreTransposedBlocks3(const V A, const V B, const V C,
|
||||||
|
Simd<T, N, 0> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
StoreU(A, d, unaligned + 0 * N);
|
||||||
|
StoreU(B, d, unaligned + 1 * N);
|
||||||
|
StoreU(C, d, unaligned + 2 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
|
// >= 128-bit vector, 8-bit lanes
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 1),
|
||||||
|
HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved3(const V v0, const V v1, const V v2,
|
||||||
|
Simd<T, N, 0> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
const auto k5 = Set(du, 5);
|
||||||
|
const auto k6 = Set(du, 6);
|
||||||
|
|
||||||
|
// Interleave (v0,v1,v2) to (MSB on left, lane 0 on right):
|
||||||
|
// v0[5], v2[4],v1[4],v0[4] .. v2[0],v1[0],v0[0]. We're expanding v0 lanes
|
||||||
|
// to their place, with 0x80 so lanes to be filled from other vectors are 0
|
||||||
|
// to enable blending by ORing together.
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v0[16] = {
|
||||||
|
0, 0x80, 0x80, 1, 0x80, 0x80, 2, 0x80, 0x80, //
|
||||||
|
3, 0x80, 0x80, 4, 0x80, 0x80, 5};
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v1[16] = {
|
||||||
|
0x80, 0, 0x80, 0x80, 1, 0x80, //
|
||||||
|
0x80, 2, 0x80, 0x80, 3, 0x80, 0x80, 4, 0x80, 0x80};
|
||||||
|
// The interleaved vectors will be named A, B, C; temporaries with suffix
|
||||||
|
// 0..2 indicate which input vector's lanes they hold.
|
||||||
|
const auto shuf_A0 = LoadDup128(du, tbl_v0);
|
||||||
|
const auto shuf_A1 = LoadDup128(du, tbl_v1); // cannot reuse shuf_A0 (has 5)
|
||||||
|
const auto shuf_A2 = CombineShiftRightBytes<15>(du, shuf_A1, shuf_A1);
|
||||||
|
const auto A0 = TableLookupBytesOr0(v0, shuf_A0); // 5..4..3..2..1..0
|
||||||
|
const auto A1 = TableLookupBytesOr0(v1, shuf_A1); // ..4..3..2..1..0.
|
||||||
|
const auto A2 = TableLookupBytesOr0(v2, shuf_A2); // .4..3..2..1..0..
|
||||||
|
const V A = BitCast(d, A0 | A1 | A2);
|
||||||
|
|
||||||
|
// B: v1[10],v0[10], v2[9],v1[9],v0[9] .. , v2[6],v1[6],v0[6], v2[5],v1[5]
|
||||||
|
const auto shuf_B0 = shuf_A2 + k6; // .A..9..8..7..6..
|
||||||
|
const auto shuf_B1 = shuf_A0 + k5; // A..9..8..7..6..5
|
||||||
|
const auto shuf_B2 = shuf_A1 + k5; // ..9..8..7..6..5.
|
||||||
|
const auto B0 = TableLookupBytesOr0(v0, shuf_B0);
|
||||||
|
const auto B1 = TableLookupBytesOr0(v1, shuf_B1);
|
||||||
|
const auto B2 = TableLookupBytesOr0(v2, shuf_B2);
|
||||||
|
const V B = BitCast(d, B0 | B1 | B2);
|
||||||
|
|
||||||
|
// C: v2[15],v1[15],v0[15], v2[11],v1[11],v0[11], v2[10]
|
||||||
|
const auto shuf_C0 = shuf_B2 + k6; // ..F..E..D..C..B.
|
||||||
|
const auto shuf_C1 = shuf_B0 + k5; // .F..E..D..C..B..
|
||||||
|
const auto shuf_C2 = shuf_B1 + k5; // F..E..D..C..B..A
|
||||||
|
const auto C0 = TableLookupBytesOr0(v0, shuf_C0);
|
||||||
|
const auto C1 = TableLookupBytesOr0(v1, shuf_C1);
|
||||||
|
const auto C2 = TableLookupBytesOr0(v2, shuf_C2);
|
||||||
|
const V C = BitCast(d, C0 | C1 | C2);
|
||||||
|
|
||||||
|
detail::StoreTransposedBlocks3(A, B, C, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// >= 128-bit vector, 16-bit lanes
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 2),
|
||||||
|
HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved3(const V v0, const V v1, const V v2,
|
||||||
|
Simd<T, N, 0> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
const Repartition<uint8_t, decltype(d)> du8;
|
||||||
|
const auto k2 = Set(du8, 2 * sizeof(T));
|
||||||
|
const auto k3 = Set(du8, 3 * sizeof(T));
|
||||||
|
|
||||||
|
// Interleave (v0,v1,v2) to (MSB on left, lane 0 on right):
|
||||||
|
// v1[2],v0[2], v2[1],v1[1],v0[1], v2[0],v1[0],v0[0]. 0x80 so lanes to be
|
||||||
|
// filled from other vectors are 0 for blending. Note that these are byte
|
||||||
|
// indices for 16-bit lanes.
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v1[16] = {
|
||||||
|
0x80, 0x80, 0, 1, 0x80, 0x80, 0x80, 0x80,
|
||||||
|
2, 3, 0x80, 0x80, 0x80, 0x80, 4, 5};
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v2[16] = {
|
||||||
|
0x80, 0x80, 0x80, 0x80, 0, 1, 0x80, 0x80,
|
||||||
|
0x80, 0x80, 2, 3, 0x80, 0x80, 0x80, 0x80};
|
||||||
|
|
||||||
|
// The interleaved vectors will be named A, B, C; temporaries with suffix
|
||||||
|
// 0..2 indicate which input vector's lanes they hold.
|
||||||
|
const auto shuf_A1 = LoadDup128(du8, tbl_v1); // 2..1..0.
|
||||||
|
// .2..1..0
|
||||||
|
const auto shuf_A0 = CombineShiftRightBytes<2>(du8, shuf_A1, shuf_A1);
|
||||||
|
const auto shuf_A2 = LoadDup128(du8, tbl_v2); // ..1..0..
|
||||||
|
|
||||||
|
const auto A0 = TableLookupBytesOr0(v0, shuf_A0);
|
||||||
|
const auto A1 = TableLookupBytesOr0(v1, shuf_A1);
|
||||||
|
const auto A2 = TableLookupBytesOr0(v2, shuf_A2);
|
||||||
|
const V A = BitCast(d, A0 | A1 | A2);
|
||||||
|
|
||||||
|
// B: v0[5] v2[4],v1[4],v0[4], v2[3],v1[3],v0[3], v2[2]
|
||||||
|
const auto shuf_B0 = shuf_A1 + k3; // 5..4..3.
|
||||||
|
const auto shuf_B1 = shuf_A2 + k3; // ..4..3..
|
||||||
|
const auto shuf_B2 = shuf_A0 + k2; // .4..3..2
|
||||||
|
const auto B0 = TableLookupBytesOr0(v0, shuf_B0);
|
||||||
|
const auto B1 = TableLookupBytesOr0(v1, shuf_B1);
|
||||||
|
const auto B2 = TableLookupBytesOr0(v2, shuf_B2);
|
||||||
|
const V B = BitCast(d, B0 | B1 | B2);
|
||||||
|
|
||||||
|
// C: v2[7],v1[7],v0[7], v2[6],v1[6],v0[6], v2[5],v1[5]
|
||||||
|
const auto shuf_C0 = shuf_B1 + k3; // ..7..6..
|
||||||
|
const auto shuf_C1 = shuf_B2 + k3; // .7..6..5
|
||||||
|
const auto shuf_C2 = shuf_B0 + k2; // 7..6..5.
|
||||||
|
const auto C0 = TableLookupBytesOr0(v0, shuf_C0);
|
||||||
|
const auto C1 = TableLookupBytesOr0(v1, shuf_C1);
|
||||||
|
const auto C2 = TableLookupBytesOr0(v2, shuf_C2);
|
||||||
|
const V C = BitCast(d, C0 | C1 | C2);
|
||||||
|
|
||||||
|
detail::StoreTransposedBlocks3(A, B, C, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// >= 128-bit vector, 32-bit lanes
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 4),
|
||||||
|
HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved3(const V v0, const V v1, const V v2,
|
||||||
|
Simd<T, N, 0> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
const RepartitionToWide<decltype(d)> dw;
|
||||||
|
|
||||||
|
const V v10_v00 = InterleaveLower(d, v0, v1);
|
||||||
|
const V v01_v20 = OddEven(v0, v2);
|
||||||
|
// A: v0[1], v2[0],v1[0],v0[0] (<- lane 0)
|
||||||
|
const V A = BitCast(
|
||||||
|
d, InterleaveLower(dw, BitCast(dw, v10_v00), BitCast(dw, v01_v20)));
|
||||||
|
|
||||||
|
const V v1_321 = ShiftRightLanes<1>(d, v1);
|
||||||
|
const V v0_32 = ShiftRightLanes<2>(d, v0);
|
||||||
|
const V v21_v11 = OddEven(v2, v1_321);
|
||||||
|
const V v12_v02 = OddEven(v1_321, v0_32);
|
||||||
|
// B: v1[2],v0[2], v2[1],v1[1]
|
||||||
|
const V B = BitCast(
|
||||||
|
d, InterleaveLower(dw, BitCast(dw, v21_v11), BitCast(dw, v12_v02)));
|
||||||
|
|
||||||
|
// Notation refers to the upper 2 lanes of the vector for InterleaveUpper.
|
||||||
|
const V v23_v13 = OddEven(v2, v1_321);
|
||||||
|
const V v03_v22 = OddEven(v0, v2);
|
||||||
|
// C: v2[3],v1[3],v0[3], v2[2]
|
||||||
|
const V C = BitCast(
|
||||||
|
d, InterleaveUpper(dw, BitCast(dw, v03_v22), BitCast(dw, v23_v13)));
|
||||||
|
|
||||||
|
detail::StoreTransposedBlocks3(A, B, C, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// >= 128-bit vector, 64-bit lanes
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 8),
|
||||||
|
HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved3(const V v0, const V v1, const V v2,
|
||||||
|
Simd<T, N, 0> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
const V A = InterleaveLower(d, v0, v1);
|
||||||
|
const V B = OddEven(v0, v2);
|
||||||
|
const V C = InterleaveUpper(d, v1, v2);
|
||||||
|
detail::StoreTransposedBlocks3(A, B, C, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64-bit vector, 8-bit lanes
|
||||||
|
template <typename T, HWY_IF_LANE_SIZE(T, 1)>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec64<T> part0, const Vec64<T> part1,
|
||||||
|
const Vec64<T> part2, Full64<T> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
constexpr size_t N = 16 / sizeof(T);
|
||||||
|
// Use full vectors for the shuffles and first result.
|
||||||
|
const Full128<uint8_t> du;
|
||||||
|
const Full128<T> d_full;
|
||||||
|
const auto k5 = Set(du, 5);
|
||||||
|
const auto k6 = Set(du, 6);
|
||||||
|
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
|
||||||
|
// Interleave (v0,v1,v2) to (MSB on left, lane 0 on right):
|
||||||
|
// v1[2],v0[2], v2[1],v1[1],v0[1], v2[0],v1[0],v0[0]. 0x80 so lanes to be
|
||||||
|
// filled from other vectors are 0 for blending.
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v0[16] = {
|
||||||
|
0, 0x80, 0x80, 1, 0x80, 0x80, 2, 0x80, 0x80, //
|
||||||
|
3, 0x80, 0x80, 4, 0x80, 0x80, 5};
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v1[16] = {
|
||||||
|
0x80, 0, 0x80, 0x80, 1, 0x80, //
|
||||||
|
0x80, 2, 0x80, 0x80, 3, 0x80, 0x80, 4, 0x80, 0x80};
|
||||||
|
// The interleaved vectors will be named A, B, C; temporaries with suffix
|
||||||
|
// 0..2 indicate which input vector's lanes they hold.
|
||||||
|
const auto shuf_A0 = Load(du, tbl_v0);
|
||||||
|
const auto shuf_A1 = Load(du, tbl_v1); // cannot reuse shuf_A0 (5 in MSB)
|
||||||
|
const auto shuf_A2 = CombineShiftRightBytes<15>(du, shuf_A1, shuf_A1);
|
||||||
|
const auto A0 = TableLookupBytesOr0(v0, shuf_A0); // 5..4..3..2..1..0
|
||||||
|
const auto A1 = TableLookupBytesOr0(v1, shuf_A1); // ..4..3..2..1..0.
|
||||||
|
const auto A2 = TableLookupBytesOr0(v2, shuf_A2); // .4..3..2..1..0..
|
||||||
|
const auto A = BitCast(d_full, A0 | A1 | A2);
|
||||||
|
StoreU(A, d_full, unaligned + 0 * N);
|
||||||
|
|
||||||
|
// Second (HALF) vector: v2[7],v1[7],v0[7], v2[6],v1[6],v0[6], v2[5],v1[5]
|
||||||
|
const auto shuf_B0 = shuf_A2 + k6; // ..7..6..
|
||||||
|
const auto shuf_B1 = shuf_A0 + k5; // .7..6..5
|
||||||
|
const auto shuf_B2 = shuf_A1 + k5; // 7..6..5.
|
||||||
|
const auto B0 = TableLookupBytesOr0(v0, shuf_B0);
|
||||||
|
const auto B1 = TableLookupBytesOr0(v1, shuf_B1);
|
||||||
|
const auto B2 = TableLookupBytesOr0(v2, shuf_B2);
|
||||||
|
const Vec64<T> B{(B0 | B1 | B2).raw};
|
||||||
|
StoreU(B, d, unaligned + 1 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64-bit vector, 16-bit lanes
|
||||||
|
template <typename T, HWY_IF_LANE_SIZE(T, 2)>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec64<T> part0, const Vec64<T> part1,
|
||||||
|
const Vec64<T> part2, Full64<T> dh,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
const Full128<T> d;
|
||||||
|
const Full128<uint8_t> du8;
|
||||||
|
constexpr size_t N = 16 / sizeof(T);
|
||||||
|
const auto k2 = Set(du8, 2 * sizeof(T));
|
||||||
|
const auto k3 = Set(du8, 3 * sizeof(T));
|
||||||
|
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
|
||||||
|
// Interleave part (v0,v1,v2) to full (MSB on left, lane 0 on right):
|
||||||
|
// v1[2],v0[2], v2[1],v1[1],v0[1], v2[0],v1[0],v0[0]. We're expanding v0 lanes
|
||||||
|
// to their place, with 0x80 so lanes to be filled from other vectors are 0
|
||||||
|
// to enable blending by ORing together.
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v1[16] = {
|
||||||
|
0x80, 0x80, 0, 1, 0x80, 0x80, 0x80, 0x80,
|
||||||
|
2, 3, 0x80, 0x80, 0x80, 0x80, 4, 5};
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v2[16] = {
|
||||||
|
0x80, 0x80, 0x80, 0x80, 0, 1, 0x80, 0x80,
|
||||||
|
0x80, 0x80, 2, 3, 0x80, 0x80, 0x80, 0x80};
|
||||||
|
|
||||||
|
// The interleaved vectors will be named A, B; temporaries with suffix
|
||||||
|
// 0..2 indicate which input vector's lanes they hold.
|
||||||
|
const auto shuf_A1 = Load(du8, tbl_v1); // 2..1..0.
|
||||||
|
// .2..1..0
|
||||||
|
const auto shuf_A0 = CombineShiftRightBytes<2>(du8, shuf_A1, shuf_A1);
|
||||||
|
const auto shuf_A2 = Load(du8, tbl_v2); // ..1..0..
|
||||||
|
|
||||||
|
const auto A0 = TableLookupBytesOr0(v0, shuf_A0);
|
||||||
|
const auto A1 = TableLookupBytesOr0(v1, shuf_A1);
|
||||||
|
const auto A2 = TableLookupBytesOr0(v2, shuf_A2);
|
||||||
|
const Vec128<T> A = BitCast(d, A0 | A1 | A2);
|
||||||
|
StoreU(A, d, unaligned + 0 * N);
|
||||||
|
|
||||||
|
// Second (HALF) vector: v2[3],v1[3],v0[3], v2[2]
|
||||||
|
const auto shuf_B0 = shuf_A1 + k3; // ..3.
|
||||||
|
const auto shuf_B1 = shuf_A2 + k3; // .3..
|
||||||
|
const auto shuf_B2 = shuf_A0 + k2; // 3..2
|
||||||
|
const auto B0 = TableLookupBytesOr0(v0, shuf_B0);
|
||||||
|
const auto B1 = TableLookupBytesOr0(v1, shuf_B1);
|
||||||
|
const auto B2 = TableLookupBytesOr0(v2, shuf_B2);
|
||||||
|
const Vec128<T> B = BitCast(d, B0 | B1 | B2);
|
||||||
|
StoreU(Vec64<T>{B.raw}, dh, unaligned + 1 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64-bit vector, 32-bit lanes
|
||||||
|
template <typename T, HWY_IF_LANE_SIZE(T, 4)>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec64<T> v0, const Vec64<T> v1,
|
||||||
|
const Vec64<T> v2, Full64<T> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
// (same code as 128-bit vector, 64-bit lanes)
|
||||||
|
constexpr size_t N = 2;
|
||||||
|
const Vec64<T> v10_v00 = InterleaveLower(d, v0, v1);
|
||||||
|
const Vec64<T> v01_v20 = OddEven(v0, v2);
|
||||||
|
const Vec64<T> v21_v11 = InterleaveUpper(d, v1, v2);
|
||||||
|
StoreU(v10_v00, d, unaligned + 0 * N);
|
||||||
|
StoreU(v01_v20, d, unaligned + 1 * N);
|
||||||
|
StoreU(v21_v11, d, unaligned + 2 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64-bit lanes are handled by the N=1 case below.
|
||||||
|
|
||||||
|
// <= 32-bit vector, 8-bit lanes
|
||||||
|
template <typename T, size_t N, HWY_IF_LANE_SIZE(T, 1), HWY_IF_LE32(T, N)>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec128<T, N> part0,
|
||||||
|
const Vec128<T, N> part1,
|
||||||
|
const Vec128<T, N> part2, Simd<T, N, 0> /*tag*/,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
// Use full vectors for the shuffles and result.
|
||||||
|
const Full128<uint8_t> du;
|
||||||
|
const Full128<T> d_full;
|
||||||
|
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
|
||||||
|
// Interleave (v0,v1,v2). We're expanding v0 lanes to their place, with 0x80
|
||||||
|
// so lanes to be filled from other vectors are 0 to enable blending by ORing
|
||||||
|
// together.
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v0[16] = {
|
||||||
|
0, 0x80, 0x80, 1, 0x80, 0x80, 2, 0x80,
|
||||||
|
0x80, 3, 0x80, 0x80, 0x80, 0x80, 0x80, 0x80};
|
||||||
|
// The interleaved vector will be named A; temporaries with suffix
|
||||||
|
// 0..2 indicate which input vector's lanes they hold.
|
||||||
|
const auto shuf_A0 = Load(du, tbl_v0);
|
||||||
|
const auto shuf_A1 = CombineShiftRightBytes<15>(du, shuf_A0, shuf_A0);
|
||||||
|
const auto shuf_A2 = CombineShiftRightBytes<14>(du, shuf_A0, shuf_A0);
|
||||||
|
const auto A0 = TableLookupBytesOr0(v0, shuf_A0); // ......3..2..1..0
|
||||||
|
const auto A1 = TableLookupBytesOr0(v1, shuf_A1); // .....3..2..1..0.
|
||||||
|
const auto A2 = TableLookupBytesOr0(v2, shuf_A2); // ....3..2..1..0..
|
||||||
|
const Vec128<T> A = BitCast(d_full, A0 | A1 | A2);
|
||||||
|
alignas(16) T buf[16 / sizeof(T)];
|
||||||
|
StoreU(A, d_full, buf);
|
||||||
|
CopyBytes<N * 3 * sizeof(T)>(buf, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 32-bit vector, 16-bit lanes
|
||||||
|
template <typename T, HWY_IF_LANE_SIZE(T, 2)>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec128<T, 2> part0,
|
||||||
|
const Vec128<T, 2> part1,
|
||||||
|
const Vec128<T, 2> part2, Simd<T, 2, 0> /*tag*/,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
constexpr size_t N = 4 / sizeof(T);
|
||||||
|
// Use full vectors for the shuffles and result.
|
||||||
|
const Full128<uint8_t> du8;
|
||||||
|
const Full128<T> d_full;
|
||||||
|
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
|
||||||
|
// Interleave (v0,v1,v2). We're expanding v0 lanes to their place, with 0x80
|
||||||
|
// so lanes to be filled from other vectors are 0 to enable blending by ORing
|
||||||
|
// together.
|
||||||
|
alignas(16) static constexpr uint8_t tbl_v2[16] = {
|
||||||
|
0x80, 0x80, 0x80, 0x80, 0, 1, 0x80, 0x80,
|
||||||
|
0x80, 0x80, 2, 3, 0x80, 0x80, 0x80, 0x80};
|
||||||
|
// The interleaved vector will be named A; temporaries with suffix
|
||||||
|
// 0..2 indicate which input vector's lanes they hold.
|
||||||
|
const auto shuf_A2 = // ..1..0..
|
||||||
|
Load(du8, tbl_v2);
|
||||||
|
const auto shuf_A1 = // ...1..0.
|
||||||
|
CombineShiftRightBytes<2>(du8, shuf_A2, shuf_A2);
|
||||||
|
const auto shuf_A0 = // ....1..0
|
||||||
|
CombineShiftRightBytes<4>(du8, shuf_A2, shuf_A2);
|
||||||
|
const auto A0 = TableLookupBytesOr0(v0, shuf_A0); // ..1..0
|
||||||
|
const auto A1 = TableLookupBytesOr0(v1, shuf_A1); // .1..0.
|
||||||
|
const auto A2 = TableLookupBytesOr0(v2, shuf_A2); // 1..0..
|
||||||
|
const auto A = BitCast(d_full, A0 | A1 | A2);
|
||||||
|
alignas(16) T buf[16 / sizeof(T)];
|
||||||
|
StoreU(A, d_full, buf);
|
||||||
|
CopyBytes<N * 3 * sizeof(T)>(buf, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Single-element vector, any lane size: just store directly
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec128<T, 1> v0, const Vec128<T, 1> v1,
|
||||||
|
const Vec128<T, 1> v2, Simd<T, 1, 0> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
StoreU(v0, d, unaligned + 0);
|
||||||
|
StoreU(v1, d, unaligned + 1);
|
||||||
|
StoreU(v2, d, unaligned + 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ StoreInterleaved4
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// Default for <= 128-bit vectors; x86_256 and x86_512 have their own overload.
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LE128(T, N)>
|
||||||
|
HWY_API void StoreTransposedBlocks4(const V A, const V B, const V C, const V D,
|
||||||
|
Simd<T, N, 0> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
StoreU(A, d, unaligned + 0 * N);
|
||||||
|
StoreU(B, d, unaligned + 1 * N);
|
||||||
|
StoreU(C, d, unaligned + 2 * N);
|
||||||
|
StoreU(D, d, unaligned + 3 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
|
// >= 128-bit vector, 8..32-bit lanes
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_NOT_LANE_SIZE(T, 8),
|
||||||
|
HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved4(const V v0, const V v1, const V v2, const V v3,
|
||||||
|
Simd<T, N, 0> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
const RepartitionToWide<decltype(d)> dw;
|
||||||
|
const auto v10L = ZipLower(dw, v0, v1); // .. v1[0] v0[0]
|
||||||
|
const auto v32L = ZipLower(dw, v2, v3);
|
||||||
|
const auto v10U = ZipUpper(dw, v0, v1);
|
||||||
|
const auto v32U = ZipUpper(dw, v2, v3);
|
||||||
|
// The interleaved vectors are A, B, C, D.
|
||||||
|
const auto A = BitCast(d, InterleaveLower(dw, v10L, v32L)); // 3210
|
||||||
|
const auto B = BitCast(d, InterleaveUpper(dw, v10L, v32L));
|
||||||
|
const auto C = BitCast(d, InterleaveLower(dw, v10U, v32U));
|
||||||
|
const auto D = BitCast(d, InterleaveUpper(dw, v10U, v32U));
|
||||||
|
detail::StoreTransposedBlocks4(A, B, C, D, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// >= 128-bit vector, 64-bit lanes
|
||||||
|
template <typename T, size_t N, class V, HWY_IF_LANE_SIZE(T, 8),
|
||||||
|
HWY_IF_GE128(T, N)>
|
||||||
|
HWY_API void StoreInterleaved4(const V v0, const V v1, const V v2, const V v3,
|
||||||
|
Simd<T, N, 0> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
// The interleaved vectors are A, B, C, D.
|
||||||
|
const auto A = InterleaveLower(d, v0, v1); // v1[0] v0[0]
|
||||||
|
const auto B = InterleaveLower(d, v2, v3);
|
||||||
|
const auto C = InterleaveUpper(d, v0, v1);
|
||||||
|
const auto D = InterleaveUpper(d, v2, v3);
|
||||||
|
detail::StoreTransposedBlocks4(A, B, C, D, d, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64-bit vector, 8..32-bit lanes
|
||||||
|
template <typename T, HWY_IF_NOT_LANE_SIZE(T, 8)>
|
||||||
|
HWY_API void StoreInterleaved4(const Vec64<T> part0, const Vec64<T> part1,
|
||||||
|
const Vec64<T> part2, const Vec64<T> part3,
|
||||||
|
Full64<T> /*tag*/, T* HWY_RESTRICT unaligned) {
|
||||||
|
constexpr size_t N = 16 / sizeof(T);
|
||||||
|
// Use full vectors to reduce the number of stores.
|
||||||
|
const Full128<T> d_full;
|
||||||
|
const RepartitionToWide<decltype(d_full)> dw;
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
const Vec128<T> v3{part3.raw};
|
||||||
|
const auto v10 = ZipLower(dw, v0, v1); // v1[0] v0[0]
|
||||||
|
const auto v32 = ZipLower(dw, v2, v3);
|
||||||
|
const auto A = BitCast(d_full, InterleaveLower(dw, v10, v32));
|
||||||
|
const auto B = BitCast(d_full, InterleaveUpper(dw, v10, v32));
|
||||||
|
StoreU(A, d_full, unaligned + 0 * N);
|
||||||
|
StoreU(B, d_full, unaligned + 1 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 64-bit vector, 64-bit lane
|
||||||
|
template <typename T, HWY_IF_LANE_SIZE(T, 8)>
|
||||||
|
HWY_API void StoreInterleaved4(const Vec64<T> part0, const Vec64<T> part1,
|
||||||
|
const Vec64<T> part2, const Vec64<T> part3,
|
||||||
|
Full64<T> /*tag*/, T* HWY_RESTRICT unaligned) {
|
||||||
|
constexpr size_t N = 16 / sizeof(T);
|
||||||
|
// Use full vectors to reduce the number of stores.
|
||||||
|
const Full128<T> d_full;
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
const Vec128<T> v3{part3.raw};
|
||||||
|
const auto A = InterleaveLower(d_full, v0, v1); // v1[0] v0[0]
|
||||||
|
const auto B = InterleaveLower(d_full, v2, v3);
|
||||||
|
StoreU(A, d_full, unaligned + 0 * N);
|
||||||
|
StoreU(B, d_full, unaligned + 1 * N);
|
||||||
|
}
|
||||||
|
|
||||||
|
// <= 32-bit vectors
|
||||||
|
template <typename T, size_t N, HWY_IF_LE32(T, N)>
|
||||||
|
HWY_API void StoreInterleaved4(const Vec128<T, N> part0,
|
||||||
|
const Vec128<T, N> part1,
|
||||||
|
const Vec128<T, N> part2,
|
||||||
|
const Vec128<T, N> part3, Simd<T, N, 0> /*tag*/,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
// Use full vectors to reduce the number of stores.
|
||||||
|
const Full128<T> d_full;
|
||||||
|
const RepartitionToWide<decltype(d_full)> dw;
|
||||||
|
const Vec128<T> v0{part0.raw};
|
||||||
|
const Vec128<T> v1{part1.raw};
|
||||||
|
const Vec128<T> v2{part2.raw};
|
||||||
|
const Vec128<T> v3{part3.raw};
|
||||||
|
const auto v10 = ZipLower(dw, v0, v1); // .. v1[0] v0[0]
|
||||||
|
const auto v32 = ZipLower(dw, v2, v3);
|
||||||
|
const auto v3210 = BitCast(d_full, InterleaveLower(dw, v10, v32));
|
||||||
|
alignas(16) T buf[16 / sizeof(T)];
|
||||||
|
StoreU(v3210, d_full, buf);
|
||||||
|
CopyBytes<4 * N * sizeof(T)>(buf, unaligned);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
|
||||||
// ------------------------------ AESRound
|
// ------------------------------ AESRound
|
||||||
|
|
||||||
// Cannot implement on scalar: need at least 16 bytes for TableLookupBytes.
|
// Cannot implement on scalar: need at least 16 bytes for TableLookupBytes.
|
||||||
|
@ -281,6 +1181,7 @@ HWY_API V CLMulUpper(V a, V b) {
|
||||||
#define HWY_NATIVE_POPCNT
|
#define HWY_NATIVE_POPCNT
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#undef HWY_MIN_POW2_FOR_128
|
||||||
#if HWY_TARGET == HWY_RVV
|
#if HWY_TARGET == HWY_RVV
|
||||||
#define HWY_MIN_POW2_FOR_128 1
|
#define HWY_MIN_POW2_FOR_128 1
|
||||||
#else
|
#else
|
||||||
|
@ -291,10 +1192,11 @@ HWY_API V CLMulUpper(V a, V b) {
|
||||||
|
|
||||||
// This algorithm requires vectors to be at least 16 bytes, which is the case
|
// This algorithm requires vectors to be at least 16 bytes, which is the case
|
||||||
// for LMUL >= 2. If not, use the fallback below.
|
// for LMUL >= 2. If not, use the fallback below.
|
||||||
template <typename V, HWY_IF_LANES_ARE(uint8_t, V), HWY_IF_GE128_D(DFromV<V>),
|
template <typename V, class D = DFromV<V>, HWY_IF_LANE_SIZE_D(D, 1),
|
||||||
HWY_IF_POW2_GE(DFromV<V>, HWY_MIN_POW2_FOR_128)>
|
HWY_IF_GE128_D(D), HWY_IF_POW2_GE(D, HWY_MIN_POW2_FOR_128)>
|
||||||
HWY_API V PopulationCount(V v) {
|
HWY_API V PopulationCount(V v) {
|
||||||
const DFromV<V> d;
|
static_assert(IsSame<TFromD<D>, uint8_t>(), "V must be u8");
|
||||||
|
const D d;
|
||||||
HWY_ALIGN constexpr uint8_t kLookup[16] = {
|
HWY_ALIGN constexpr uint8_t kLookup[16] = {
|
||||||
0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4,
|
0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4,
|
||||||
};
|
};
|
||||||
|
@ -307,9 +1209,10 @@ HWY_API V PopulationCount(V v) {
|
||||||
// RVV has a specialization that avoids the Set().
|
// RVV has a specialization that avoids the Set().
|
||||||
#if HWY_TARGET != HWY_RVV
|
#if HWY_TARGET != HWY_RVV
|
||||||
// Slower fallback for capped vectors.
|
// Slower fallback for capped vectors.
|
||||||
template <typename V, HWY_IF_LANES_ARE(uint8_t, V), HWY_IF_LT128_D(DFromV<V>)>
|
template <typename V, class D = DFromV<V>, HWY_IF_LANE_SIZE_D(D, 1), HWY_IF_LT128_D(D)>
|
||||||
HWY_API V PopulationCount(V v) {
|
HWY_API V PopulationCount(V v) {
|
||||||
const DFromV<V> d;
|
static_assert(IsSame<TFromD<D>, uint8_t>(), "V must be u8");
|
||||||
|
const D d;
|
||||||
// See https://arxiv.org/pdf/1611.07612.pdf, Figure 3
|
// See https://arxiv.org/pdf/1611.07612.pdf, Figure 3
|
||||||
v = Sub(v, And(ShiftRight<1>(v), Set(d, 0x55)));
|
v = Sub(v, And(ShiftRight<1>(v), Set(d, 0x55)));
|
||||||
v = Add(And(ShiftRight<2>(v), Set(d, 0x33)), And(v, Set(d, 0x33)));
|
v = Add(And(ShiftRight<2>(v), Set(d, 0x33)), And(v, Set(d, 0x33)));
|
||||||
|
@ -317,26 +1220,29 @@ HWY_API V PopulationCount(V v) {
|
||||||
}
|
}
|
||||||
#endif // HWY_TARGET != HWY_RVV
|
#endif // HWY_TARGET != HWY_RVV
|
||||||
|
|
||||||
template <typename V, HWY_IF_LANES_ARE(uint16_t, V)>
|
template <typename V, class D = DFromV<V>, HWY_IF_LANE_SIZE_D(D, 2)>
|
||||||
HWY_API V PopulationCount(V v) {
|
HWY_API V PopulationCount(V v) {
|
||||||
const DFromV<V> d;
|
static_assert(IsSame<TFromD<D>, uint16_t>(), "V must be u16");
|
||||||
|
const D d;
|
||||||
const Repartition<uint8_t, decltype(d)> d8;
|
const Repartition<uint8_t, decltype(d)> d8;
|
||||||
const auto vals = BitCast(d, PopulationCount(BitCast(d8, v)));
|
const auto vals = BitCast(d, PopulationCount(BitCast(d8, v)));
|
||||||
return Add(ShiftRight<8>(vals), And(vals, Set(d, 0xFF)));
|
return Add(ShiftRight<8>(vals), And(vals, Set(d, 0xFF)));
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename V, HWY_IF_LANES_ARE(uint32_t, V)>
|
template <typename V, class D = DFromV<V>, HWY_IF_LANE_SIZE_D(D, 4)>
|
||||||
HWY_API V PopulationCount(V v) {
|
HWY_API V PopulationCount(V v) {
|
||||||
const DFromV<V> d;
|
static_assert(IsSame<TFromD<D>, uint32_t>(), "V must be u32");
|
||||||
|
const D d;
|
||||||
Repartition<uint16_t, decltype(d)> d16;
|
Repartition<uint16_t, decltype(d)> d16;
|
||||||
auto vals = BitCast(d, PopulationCount(BitCast(d16, v)));
|
auto vals = BitCast(d, PopulationCount(BitCast(d16, v)));
|
||||||
return Add(ShiftRight<16>(vals), And(vals, Set(d, 0xFF)));
|
return Add(ShiftRight<16>(vals), And(vals, Set(d, 0xFF)));
|
||||||
}
|
}
|
||||||
|
|
||||||
#if HWY_HAVE_INTEGER64
|
#if HWY_HAVE_INTEGER64
|
||||||
template <typename V, HWY_IF_LANES_ARE(uint64_t, V)>
|
template <typename V, class D = DFromV<V>, HWY_IF_LANE_SIZE_D(D, 8)>
|
||||||
HWY_API V PopulationCount(V v) {
|
HWY_API V PopulationCount(V v) {
|
||||||
const DFromV<V> d;
|
static_assert(IsSame<TFromD<D>, uint64_t>(), "V must be u64");
|
||||||
|
const D d;
|
||||||
Repartition<uint32_t, decltype(d)> d32;
|
Repartition<uint32_t, decltype(d)> d32;
|
||||||
auto vals = BitCast(d, PopulationCount(BitCast(d32, v)));
|
auto vals = BitCast(d, PopulationCount(BitCast(d32, v)));
|
||||||
return Add(ShiftRight<32>(vals), And(vals, Set(d, 0xFF)));
|
return Add(ShiftRight<32>(vals), And(vals, Set(d, 0xFF)));
|
||||||
|
|
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -70,7 +71,7 @@ class Mask1 {
|
||||||
public:
|
public:
|
||||||
static HWY_INLINE Mask1<T> FromBool(bool b) {
|
static HWY_INLINE Mask1<T> FromBool(bool b) {
|
||||||
Mask1<T> mask;
|
Mask1<T> mask;
|
||||||
mask.bits = b ? ~Raw(0) : 0;
|
mask.bits = b ? static_cast<Raw>(~Raw{0}) : 0;
|
||||||
return mask;
|
return mask;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -127,6 +128,9 @@ HWY_API Vec1<T> Iota(const Sisd<T> /* tag */, const T2 first) {
|
||||||
return Vec1<T>(static_cast<T>(first));
|
return Vec1<T>(static_cast<T>(first));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class D>
|
||||||
|
using VFromD = decltype(Zero(D()));
|
||||||
|
|
||||||
// ================================================== LOGICAL
|
// ================================================== LOGICAL
|
||||||
|
|
||||||
// ------------------------------ Not
|
// ------------------------------ Not
|
||||||
|
@ -187,6 +191,13 @@ HWY_API Vec1<T> operator^(const Vec1<T> a, const Vec1<T> b) {
|
||||||
return Xor(a, b);
|
return Xor(a, b);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ Or3
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Vec1<T> Or3(Vec1<T> o1, Vec1<T> o2, Vec1<T> o3) {
|
||||||
|
return Or(o1, Or(o2, o3));
|
||||||
|
}
|
||||||
|
|
||||||
// ------------------------------ OrAnd
|
// ------------------------------ OrAnd
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -337,7 +348,8 @@ HWY_API Mask1<T> Xor(const Mask1<T> a, Mask1<T> b) {
|
||||||
template <int kBits, typename T>
|
template <int kBits, typename T>
|
||||||
HWY_API Vec1<T> ShiftLeft(const Vec1<T> v) {
|
HWY_API Vec1<T> ShiftLeft(const Vec1<T> v) {
|
||||||
static_assert(0 <= kBits && kBits < sizeof(T) * 8, "Invalid shift");
|
static_assert(0 <= kBits && kBits < sizeof(T) * 8, "Invalid shift");
|
||||||
return Vec1<T>(static_cast<hwy::MakeUnsigned<T>>(v.raw) << kBits);
|
return Vec1<T>(
|
||||||
|
static_cast<T>(static_cast<hwy::MakeUnsigned<T>>(v.raw) << kBits));
|
||||||
}
|
}
|
||||||
|
|
||||||
template <int kBits, typename T>
|
template <int kBits, typename T>
|
||||||
|
@ -346,7 +358,7 @@ HWY_API Vec1<T> ShiftRight(const Vec1<T> v) {
|
||||||
#if __cplusplus >= 202002L
|
#if __cplusplus >= 202002L
|
||||||
// Signed right shift is now guaranteed to be arithmetic (rounding toward
|
// Signed right shift is now guaranteed to be arithmetic (rounding toward
|
||||||
// negative infinity, i.e. shifting in the sign bit).
|
// negative infinity, i.e. shifting in the sign bit).
|
||||||
return Vec1<T>(v.raw >> kBits);
|
return Vec1<T>(static_cast<T>(v.raw >> kBits));
|
||||||
#else
|
#else
|
||||||
if (IsSigned<T>()) {
|
if (IsSigned<T>()) {
|
||||||
// Emulate arithmetic shift using only logical (unsigned) shifts, because
|
// Emulate arithmetic shift using only logical (unsigned) shifts, because
|
||||||
|
@ -355,28 +367,51 @@ HWY_API Vec1<T> ShiftRight(const Vec1<T> v) {
|
||||||
const Sisd<TU> du;
|
const Sisd<TU> du;
|
||||||
const TU shifted = BitCast(du, v).raw >> kBits;
|
const TU shifted = BitCast(du, v).raw >> kBits;
|
||||||
const TU sign = BitCast(du, BroadcastSignBit(v)).raw;
|
const TU sign = BitCast(du, BroadcastSignBit(v)).raw;
|
||||||
const TU upper = sign << (sizeof(TU) * 8 - 1 - kBits);
|
const size_t sign_shift =
|
||||||
|
static_cast<size_t>(static_cast<int>(sizeof(TU)) * 8 - 1 - kBits);
|
||||||
|
const TU upper = static_cast<TU>(sign << sign_shift);
|
||||||
return BitCast(Sisd<T>(), Vec1<TU>(shifted | upper));
|
return BitCast(Sisd<T>(), Vec1<TU>(shifted | upper));
|
||||||
} else {
|
} else { // T is unsigned
|
||||||
return Vec1<T>(v.raw >> kBits); // unsigned, logical shift
|
return Vec1<T>(static_cast<T>(v.raw >> kBits));
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ RotateRight (ShiftRight)
|
// ------------------------------ RotateRight (ShiftRight)
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// For partial specialization: kBits == 0 results in an invalid shift count
|
||||||
|
template <int kBits>
|
||||||
|
struct RotateRight {
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Vec1<T> operator()(const Vec1<T> v) const {
|
||||||
|
return Or(ShiftRight<kBits>(v), ShiftLeft<sizeof(T) * 8 - kBits>(v));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <>
|
||||||
|
struct RotateRight<0> {
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Vec1<T> operator()(const Vec1<T> v) const {
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
template <int kBits, typename T>
|
template <int kBits, typename T>
|
||||||
HWY_API Vec1<T> RotateRight(const Vec1<T> v) {
|
HWY_API Vec1<T> RotateRight(const Vec1<T> v) {
|
||||||
static_assert(0 <= kBits && kBits < sizeof(T) * 8, "Invalid shift");
|
static_assert(0 <= kBits && kBits < sizeof(T) * 8, "Invalid shift");
|
||||||
if (kBits == 0) return v;
|
return detail::RotateRight<kBits>()(v);
|
||||||
return Or(ShiftRight<kBits>(v), ShiftLeft<sizeof(T) * 8 - kBits>(v));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ ShiftLeftSame (BroadcastSignBit)
|
// ------------------------------ ShiftLeftSame (BroadcastSignBit)
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API Vec1<T> ShiftLeftSame(const Vec1<T> v, int bits) {
|
HWY_API Vec1<T> ShiftLeftSame(const Vec1<T> v, int bits) {
|
||||||
return Vec1<T>(static_cast<hwy::MakeUnsigned<T>>(v.raw) << bits);
|
return Vec1<T>(
|
||||||
|
static_cast<T>(static_cast<hwy::MakeUnsigned<T>>(v.raw) << bits));
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -384,7 +419,7 @@ HWY_API Vec1<T> ShiftRightSame(const Vec1<T> v, int bits) {
|
||||||
#if __cplusplus >= 202002L
|
#if __cplusplus >= 202002L
|
||||||
// Signed right shift is now guaranteed to be arithmetic (rounding toward
|
// Signed right shift is now guaranteed to be arithmetic (rounding toward
|
||||||
// negative infinity, i.e. shifting in the sign bit).
|
// negative infinity, i.e. shifting in the sign bit).
|
||||||
return Vec1<T>(v.raw >> bits);
|
return Vec1<T>(static_cast<T>(v.raw >> bits));
|
||||||
#else
|
#else
|
||||||
if (IsSigned<T>()) {
|
if (IsSigned<T>()) {
|
||||||
// Emulate arithmetic shift using only logical (unsigned) shifts, because
|
// Emulate arithmetic shift using only logical (unsigned) shifts, because
|
||||||
|
@ -393,10 +428,12 @@ HWY_API Vec1<T> ShiftRightSame(const Vec1<T> v, int bits) {
|
||||||
const Sisd<TU> du;
|
const Sisd<TU> du;
|
||||||
const TU shifted = BitCast(du, v).raw >> bits;
|
const TU shifted = BitCast(du, v).raw >> bits;
|
||||||
const TU sign = BitCast(du, BroadcastSignBit(v)).raw;
|
const TU sign = BitCast(du, BroadcastSignBit(v)).raw;
|
||||||
const TU upper = sign << (sizeof(TU) * 8 - 1 - bits);
|
const size_t sign_shift =
|
||||||
|
static_cast<size_t>(static_cast<int>(sizeof(TU)) * 8 - 1 - bits);
|
||||||
|
const TU upper = static_cast<TU>(sign << sign_shift);
|
||||||
return BitCast(Sisd<T>(), Vec1<TU>(shifted | upper));
|
return BitCast(Sisd<T>(), Vec1<TU>(shifted | upper));
|
||||||
} else {
|
} else { // T is unsigned
|
||||||
return Vec1<T>(v.raw >> bits); // unsigned, logical shift
|
return Vec1<T>(static_cast<T>(v.raw >> bits));
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
@ -601,6 +638,10 @@ HWY_API Vec1<uint16_t> MulHigh(const Vec1<uint16_t> a, const Vec1<uint16_t> b) {
|
||||||
(static_cast<uint32_t>(a.raw) * static_cast<uint32_t>(b.raw)) >> 16));
|
(static_cast<uint32_t>(a.raw) * static_cast<uint32_t>(b.raw)) >> 16));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
HWY_API Vec1<int16_t> MulFixedPoint15(Vec1<int16_t> a, Vec1<int16_t> b) {
|
||||||
|
return Vec1<int16_t>(static_cast<int16_t>((2 * a.raw * b.raw + 32768) >> 16));
|
||||||
|
}
|
||||||
|
|
||||||
// Multiplies even lanes (0, 2 ..) and returns the double-wide result.
|
// Multiplies even lanes (0, 2 ..) and returns the double-wide result.
|
||||||
HWY_API Vec1<int64_t> MulEven(const Vec1<int32_t> a, const Vec1<int32_t> b) {
|
HWY_API Vec1<int64_t> MulEven(const Vec1<int32_t> a, const Vec1<int32_t> b) {
|
||||||
const int64_t a64 = a.raw;
|
const int64_t a64 = a.raw;
|
||||||
|
@ -684,7 +725,7 @@ HWY_API Vec1<T> Round(const Vec1<T> v) {
|
||||||
const TI rounded = static_cast<TI>(v.raw + bias);
|
const TI rounded = static_cast<TI>(v.raw + bias);
|
||||||
if (rounded == 0) return CopySignToAbs(Vec1<T>(0), v);
|
if (rounded == 0) return CopySignToAbs(Vec1<T>(0), v);
|
||||||
// Round to even
|
// Round to even
|
||||||
if ((rounded & 1) && std::abs(rounded - v.raw) == T(0.5)) {
|
if ((rounded & 1) && std::abs(static_cast<T>(rounded) - v.raw) == T(0.5)) {
|
||||||
return Vec1<T>(static_cast<T>(rounded - (v.raw < T(0) ? -1 : 1)));
|
return Vec1<T>(static_cast<T>(rounded - (v.raw < T(0) ? -1 : 1)));
|
||||||
}
|
}
|
||||||
return Vec1<T>(static_cast<T>(rounded));
|
return Vec1<T>(static_cast<T>(rounded));
|
||||||
|
@ -842,6 +883,45 @@ HWY_API Mask1<T> operator>=(const Vec1<T> a, const Vec1<T> b) {
|
||||||
return Mask1<T>::FromBool(a.raw >= b.raw);
|
return Mask1<T>::FromBool(a.raw >= b.raw);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ Floating-point classification (==)
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Mask1<T> IsNaN(const Vec1<T> v) {
|
||||||
|
// std::isnan returns false for 0x7F..FF in clang AVX3 builds, so DIY.
|
||||||
|
MakeUnsigned<T> bits;
|
||||||
|
memcpy(&bits, &v, sizeof(v));
|
||||||
|
bits += bits;
|
||||||
|
bits >>= 1; // clear sign bit
|
||||||
|
// NaN if all exponent bits are set and the mantissa is not zero.
|
||||||
|
return Mask1<T>::FromBool(bits > ExponentMask<T>());
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Mask1<float> IsInf(const Vec1<float> v) {
|
||||||
|
const Sisd<float> d;
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
const Vec1<uint32_t> vu = BitCast(du, v);
|
||||||
|
// 'Shift left' to clear the sign bit, check for exponent=max and mantissa=0.
|
||||||
|
return RebindMask(d, (vu + vu) == Set(du, 0xFF000000u));
|
||||||
|
}
|
||||||
|
HWY_API Mask1<double> IsInf(const Vec1<double> v) {
|
||||||
|
const Sisd<double> d;
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
const Vec1<uint64_t> vu = BitCast(du, v);
|
||||||
|
// 'Shift left' to clear the sign bit, check for exponent=max and mantissa=0.
|
||||||
|
return RebindMask(d, (vu + vu) == Set(du, 0xFFE0000000000000ull));
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Mask1<float> IsFinite(const Vec1<float> v) {
|
||||||
|
const Vec1<uint32_t> vu = BitCast(Sisd<uint32_t>(), v);
|
||||||
|
// Shift left to clear the sign bit, check whether exponent != max value.
|
||||||
|
return Mask1<float>::FromBool((vu.raw << 1) < 0xFF000000u);
|
||||||
|
}
|
||||||
|
HWY_API Mask1<double> IsFinite(const Vec1<double> v) {
|
||||||
|
const Vec1<uint64_t> vu = BitCast(Sisd<uint64_t>(), v);
|
||||||
|
// Shift left to clear the sign bit, check whether exponent != max value.
|
||||||
|
return Mask1<double>::FromBool((vu.raw << 1) < 0xFFE0000000000000ull);
|
||||||
|
}
|
||||||
|
|
||||||
// ================================================== MEMORY
|
// ================================================== MEMORY
|
||||||
|
|
||||||
// ------------------------------ Load
|
// ------------------------------ Load
|
||||||
|
@ -883,20 +963,69 @@ HWY_API void StoreU(const Vec1<T> v, Sisd<T> d, T* HWY_RESTRICT p) {
|
||||||
return Store(v, d, p);
|
return Store(v, d, p);
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ StoreInterleaved3
|
template <typename T>
|
||||||
|
HWY_API void BlendedStore(const Vec1<T> v, Mask1<T> m, Sisd<T> d,
|
||||||
|
T* HWY_RESTRICT p) {
|
||||||
|
if (!m.bits) return;
|
||||||
|
StoreU(v, d, p);
|
||||||
|
}
|
||||||
|
|
||||||
HWY_API void StoreInterleaved3(const Vec1<uint8_t> v0, const Vec1<uint8_t> v1,
|
// ------------------------------ LoadInterleaved2/3/4
|
||||||
const Vec1<uint8_t> v2, Sisd<uint8_t> d,
|
|
||||||
uint8_t* HWY_RESTRICT unaligned) {
|
// Per-target flag to prevent generic_ops-inl.h from defining StoreInterleaved2.
|
||||||
|
#ifdef HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
#undef HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
#else
|
||||||
|
#define HWY_NATIVE_LOAD_STORE_INTERLEAVED
|
||||||
|
#endif
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void LoadInterleaved2(Sisd<T> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
Vec1<T>& v0, Vec1<T>& v1) {
|
||||||
|
v0 = LoadU(d, unaligned + 0);
|
||||||
|
v1 = LoadU(d, unaligned + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void LoadInterleaved3(Sisd<T> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
Vec1<T>& v0, Vec1<T>& v1, Vec1<T>& v2) {
|
||||||
|
v0 = LoadU(d, unaligned + 0);
|
||||||
|
v1 = LoadU(d, unaligned + 1);
|
||||||
|
v2 = LoadU(d, unaligned + 2);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void LoadInterleaved4(Sisd<T> d, const T* HWY_RESTRICT unaligned,
|
||||||
|
Vec1<T>& v0, Vec1<T>& v1, Vec1<T>& v2,
|
||||||
|
Vec1<T>& v3) {
|
||||||
|
v0 = LoadU(d, unaligned + 0);
|
||||||
|
v1 = LoadU(d, unaligned + 1);
|
||||||
|
v2 = LoadU(d, unaligned + 2);
|
||||||
|
v3 = LoadU(d, unaligned + 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ StoreInterleaved2/3/4
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void StoreInterleaved2(const Vec1<T> v0, const Vec1<T> v1, Sisd<T> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
StoreU(v0, d, unaligned + 0);
|
||||||
|
StoreU(v1, d, unaligned + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void StoreInterleaved3(const Vec1<T> v0, const Vec1<T> v1,
|
||||||
|
const Vec1<T> v2, Sisd<T> d,
|
||||||
|
T* HWY_RESTRICT unaligned) {
|
||||||
StoreU(v0, d, unaligned + 0);
|
StoreU(v0, d, unaligned + 0);
|
||||||
StoreU(v1, d, unaligned + 1);
|
StoreU(v1, d, unaligned + 1);
|
||||||
StoreU(v2, d, unaligned + 2);
|
StoreU(v2, d, unaligned + 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_API void StoreInterleaved4(const Vec1<uint8_t> v0, const Vec1<uint8_t> v1,
|
template <typename T>
|
||||||
const Vec1<uint8_t> v2, const Vec1<uint8_t> v3,
|
HWY_API void StoreInterleaved4(const Vec1<T> v0, const Vec1<T> v1,
|
||||||
Sisd<uint8_t> d,
|
const Vec1<T> v2, const Vec1<T> v3, Sisd<T> d,
|
||||||
uint8_t* HWY_RESTRICT unaligned) {
|
T* HWY_RESTRICT unaligned) {
|
||||||
StoreU(v0, d, unaligned + 0);
|
StoreU(v0, d, unaligned + 0);
|
||||||
StoreU(v1, d, unaligned + 1);
|
StoreU(v1, d, unaligned + 1);
|
||||||
StoreU(v2, d, unaligned + 2);
|
StoreU(v2, d, unaligned + 2);
|
||||||
|
@ -933,7 +1062,8 @@ template <typename T, typename Offset>
|
||||||
HWY_API Vec1<T> GatherOffset(Sisd<T> d, const T* base,
|
HWY_API Vec1<T> GatherOffset(Sisd<T> d, const T* base,
|
||||||
const Vec1<Offset> offset) {
|
const Vec1<Offset> offset) {
|
||||||
static_assert(sizeof(T) == sizeof(Offset), "Must match for portability");
|
static_assert(sizeof(T) == sizeof(Offset), "Must match for portability");
|
||||||
const uintptr_t addr = reinterpret_cast<uintptr_t>(base) + offset.raw;
|
const intptr_t addr =
|
||||||
|
reinterpret_cast<intptr_t>(base) + static_cast<intptr_t>(offset.raw);
|
||||||
return Load(d, reinterpret_cast<const T*>(addr));
|
return Load(d, reinterpret_cast<const T*>(addr));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -988,12 +1118,8 @@ HWY_API Vec1<ToT> DemoteTo(Sisd<ToT> /* tag */, Vec1<FromT> from) {
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_API Vec1<float> PromoteTo(Sisd<float> /* tag */, const Vec1<float16_t> v) {
|
HWY_API Vec1<float> PromoteTo(Sisd<float> /* tag */, const Vec1<float16_t> v) {
|
||||||
#if HWY_NATIVE_FLOAT16
|
|
||||||
uint16_t bits16;
|
uint16_t bits16;
|
||||||
CopyBytes<2>(&v.raw, &bits16);
|
CopyBytes<2>(&v.raw, &bits16);
|
||||||
#else
|
|
||||||
const uint16_t bits16 = v.raw.bits;
|
|
||||||
#endif
|
|
||||||
const uint32_t sign = static_cast<uint32_t>(bits16 >> 15);
|
const uint32_t sign = static_cast<uint32_t>(bits16 >> 15);
|
||||||
const uint32_t biased_exp = (bits16 >> 10) & 0x1F;
|
const uint32_t biased_exp = (bits16 >> 10) & 0x1F;
|
||||||
const uint32_t mantissa = bits16 & 0x3FF;
|
const uint32_t mantissa = bits16 & 0x3FF;
|
||||||
|
@ -1031,12 +1157,8 @@ HWY_API Vec1<float16_t> DemoteTo(Sisd<float16_t> /* tag */,
|
||||||
// Tiny or zero => zero.
|
// Tiny or zero => zero.
|
||||||
Vec1<float16_t> out;
|
Vec1<float16_t> out;
|
||||||
if (exp < -24) {
|
if (exp < -24) {
|
||||||
#if HWY_NATIVE_FLOAT16
|
|
||||||
const uint16_t zero = 0;
|
const uint16_t zero = 0;
|
||||||
CopyBytes<2>(&zero, &out.raw);
|
CopyBytes<2>(&zero, &out.raw);
|
||||||
#else
|
|
||||||
out.raw.bits = 0;
|
|
||||||
#endif
|
|
||||||
return out;
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1059,12 +1181,8 @@ HWY_API Vec1<float16_t> DemoteTo(Sisd<float16_t> /* tag */,
|
||||||
HWY_DASSERT(mantissa16 < 1024);
|
HWY_DASSERT(mantissa16 < 1024);
|
||||||
const uint32_t bits16 = (sign << 15) | (biased_exp16 << 10) | mantissa16;
|
const uint32_t bits16 = (sign << 15) | (biased_exp16 << 10) | mantissa16;
|
||||||
HWY_DASSERT(bits16 < 0x10000);
|
HWY_DASSERT(bits16 < 0x10000);
|
||||||
#if HWY_NATIVE_FLOAT16
|
|
||||||
const uint16_t narrowed = static_cast<uint16_t>(bits16); // big-endian safe
|
const uint16_t narrowed = static_cast<uint16_t>(bits16); // big-endian safe
|
||||||
CopyBytes<2>(&narrowed, &out.raw);
|
CopyBytes<2>(&narrowed, &out.raw);
|
||||||
#else
|
|
||||||
out.raw.bits = static_cast<uint16_t>(bits16);
|
|
||||||
#endif
|
|
||||||
return out;
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1097,6 +1215,38 @@ HWY_API Vec1<uint8_t> U8FromU32(const Vec1<uint32_t> v) {
|
||||||
return DemoteTo(Sisd<uint8_t>(), v);
|
return DemoteTo(Sisd<uint8_t>(), v);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ Truncations
|
||||||
|
|
||||||
|
HWY_API Vec1<uint8_t> TruncateTo(Sisd<uint8_t> /* tag */,
|
||||||
|
const Vec1<uint64_t> v) {
|
||||||
|
return Vec1<uint8_t>{static_cast<uint8_t>(v.raw & 0xFF)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec1<uint16_t> TruncateTo(Sisd<uint16_t> /* tag */,
|
||||||
|
const Vec1<uint64_t> v) {
|
||||||
|
return Vec1<uint16_t>{static_cast<uint16_t>(v.raw & 0xFFFF)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec1<uint32_t> TruncateTo(Sisd<uint32_t> /* tag */,
|
||||||
|
const Vec1<uint64_t> v) {
|
||||||
|
return Vec1<uint32_t>{static_cast<uint32_t>(v.raw & 0xFFFFFFFFu)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec1<uint8_t> TruncateTo(Sisd<uint8_t> /* tag */,
|
||||||
|
const Vec1<uint32_t> v) {
|
||||||
|
return Vec1<uint8_t>{static_cast<uint8_t>(v.raw & 0xFF)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec1<uint16_t> TruncateTo(Sisd<uint16_t> /* tag */,
|
||||||
|
const Vec1<uint32_t> v) {
|
||||||
|
return Vec1<uint16_t>{static_cast<uint16_t>(v.raw & 0xFFFF)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec1<uint8_t> TruncateTo(Sisd<uint8_t> /* tag */,
|
||||||
|
const Vec1<uint16_t> v) {
|
||||||
|
return Vec1<uint8_t>{static_cast<uint8_t>(v.raw & 0xFF)};
|
||||||
|
}
|
||||||
|
|
||||||
// ================================================== COMBINE
|
// ================================================== COMBINE
|
||||||
// UpperHalf, ZeroExtendVector, Combine, Concat* are unsupported.
|
// UpperHalf, ZeroExtendVector, Combine, Concat* are unsupported.
|
||||||
|
|
||||||
|
@ -1117,6 +1267,21 @@ HWY_API T GetLane(const Vec1<T> v) {
|
||||||
return v.raw;
|
return v.raw;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API T ExtractLane(const Vec1<T> v, size_t i) {
|
||||||
|
HWY_DASSERT(i == 0);
|
||||||
|
(void)i;
|
||||||
|
return v.raw;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Vec1<T> InsertLane(Vec1<T> v, size_t i, T t) {
|
||||||
|
HWY_DASSERT(i == 0);
|
||||||
|
(void)i;
|
||||||
|
v.raw = t;
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API Vec1<T> DupEven(Vec1<T> v) {
|
HWY_API Vec1<T> DupEven(Vec1<T> v) {
|
||||||
return v;
|
return v;
|
||||||
|
@ -1157,7 +1322,7 @@ HWY_API Indices1<T> IndicesFromVec(Sisd<T>, Vec1<TI> vec) {
|
||||||
|
|
||||||
template <typename T, typename TI>
|
template <typename T, typename TI>
|
||||||
HWY_API Indices1<T> SetTableIndices(Sisd<T> d, const TI* idx) {
|
HWY_API Indices1<T> SetTableIndices(Sisd<T> d, const TI* idx) {
|
||||||
return IndicesFromVec(d, LoadU(idx));
|
return IndicesFromVec(d, LoadU(Sisd<TI>(), idx));
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -1180,6 +1345,7 @@ HWY_API Vec1<T> Reverse(Sisd<T> /* tag */, const Vec1<T> v) {
|
||||||
return v;
|
return v;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Must not be called:
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API Vec1<T> Reverse2(Sisd<T> /* tag */, const Vec1<T> v) {
|
HWY_API Vec1<T> Reverse2(Sisd<T> /* tag */, const Vec1<T> v) {
|
||||||
return v;
|
return v;
|
||||||
|
@ -1304,19 +1470,24 @@ HWY_API intptr_t FindFirstTrue(Sisd<T> /* tag */, const Mask1<T> mask) {
|
||||||
|
|
||||||
// ------------------------------ Compress, CompressBits
|
// ------------------------------ Compress, CompressBits
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
struct CompressIsPartition {
|
||||||
|
enum { value = 1 };
|
||||||
|
};
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API Vec1<T> Compress(Vec1<T> v, const Mask1<T> /* mask */) {
|
HWY_API Vec1<T> Compress(Vec1<T> v, const Mask1<T> /* mask */) {
|
||||||
// Upper lanes are undefined, so result is the same independent of mask.
|
// A single lane is already partitioned by definition.
|
||||||
return v;
|
return v;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API Vec1<T> Compress(Vec1<T> v, const uint8_t* HWY_RESTRICT /* bits */) {
|
HWY_API Vec1<T> CompressNot(Vec1<T> v, const Mask1<T> /* mask */) {
|
||||||
|
// A single lane is already partitioned by definition.
|
||||||
return v;
|
return v;
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ CompressStore
|
// ------------------------------ CompressStore
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API size_t CompressStore(Vec1<T> v, const Mask1<T> mask, Sisd<T> d,
|
HWY_API size_t CompressStore(Vec1<T> v, const Mask1<T> mask, Sisd<T> d,
|
||||||
T* HWY_RESTRICT unaligned) {
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
@ -1325,7 +1496,6 @@ HWY_API size_t CompressStore(Vec1<T> v, const Mask1<T> mask, Sisd<T> d,
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ CompressBlendedStore
|
// ------------------------------ CompressBlendedStore
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API size_t CompressBlendedStore(Vec1<T> v, const Mask1<T> mask, Sisd<T> d,
|
HWY_API size_t CompressBlendedStore(Vec1<T> v, const Mask1<T> mask, Sisd<T> d,
|
||||||
T* HWY_RESTRICT unaligned) {
|
T* HWY_RESTRICT unaligned) {
|
||||||
|
@ -1334,8 +1504,13 @@ HWY_API size_t CompressBlendedStore(Vec1<T> v, const Mask1<T> mask, Sisd<T> d,
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ CompressBitsStore
|
// ------------------------------ CompressBits
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Vec1<T> CompressBits(Vec1<T> v, const uint8_t* HWY_RESTRICT /*bits*/) {
|
||||||
|
return v;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ CompressBitsStore
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API size_t CompressBitsStore(Vec1<T> v, const uint8_t* HWY_RESTRICT bits,
|
HWY_API size_t CompressBitsStore(Vec1<T> v, const uint8_t* HWY_RESTRICT bits,
|
||||||
Sisd<T> d, T* HWY_RESTRICT unaligned) {
|
Sisd<T> d, T* HWY_RESTRICT unaligned) {
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -36,6 +37,8 @@
|
||||||
#undef HWY_HAVE_INTEGER64
|
#undef HWY_HAVE_INTEGER64
|
||||||
#undef HWY_HAVE_FLOAT16
|
#undef HWY_HAVE_FLOAT16
|
||||||
#undef HWY_HAVE_FLOAT64
|
#undef HWY_HAVE_FLOAT64
|
||||||
|
#undef HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
#undef HWY_NATIVE_FMA
|
||||||
#undef HWY_CAP_GE256
|
#undef HWY_CAP_GE256
|
||||||
#undef HWY_CAP_GE512
|
#undef HWY_CAP_GE512
|
||||||
|
|
||||||
|
@ -71,6 +74,7 @@
|
||||||
|
|
||||||
// Before include guard so we redefine HWY_TARGET_STR on each include,
|
// Before include guard so we redefine HWY_TARGET_STR on each include,
|
||||||
// governed by the current HWY_TARGET.
|
// governed by the current HWY_TARGET.
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------
|
||||||
// SSSE3
|
// SSSE3
|
||||||
#if HWY_TARGET == HWY_SSSE3
|
#if HWY_TARGET == HWY_SSSE3
|
||||||
|
@ -84,11 +88,13 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
#define HWY_CAP_AES 0
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
#define HWY_TARGET_STR HWY_TARGET_STR_SSSE3
|
#define HWY_TARGET_STR HWY_TARGET_STR_SSSE3
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------
|
||||||
// SSE4
|
// SSE4
|
||||||
#elif HWY_TARGET == HWY_SSE4
|
#elif HWY_TARGET == HWY_SSE4
|
||||||
|
@ -102,6 +108,8 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
|
@ -120,6 +128,14 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
|
||||||
|
#ifdef HWY_DISABLE_BMI2_FMA
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
|
#else
|
||||||
|
#define HWY_NATIVE_FMA 1
|
||||||
|
#endif
|
||||||
|
|
||||||
#define HWY_CAP_GE256 1
|
#define HWY_CAP_GE256 1
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
|
@ -137,6 +153,8 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 0
|
||||||
|
#define HWY_NATIVE_FMA 1
|
||||||
#define HWY_CAP_GE256 1
|
#define HWY_CAP_GE256 1
|
||||||
#define HWY_CAP_GE512 1
|
#define HWY_CAP_GE512 1
|
||||||
|
|
||||||
|
@ -150,7 +168,8 @@
|
||||||
#define HWY_NAMESPACE N_AVX3_DL
|
#define HWY_NAMESPACE N_AVX3_DL
|
||||||
#define HWY_TARGET_STR \
|
#define HWY_TARGET_STR \
|
||||||
HWY_TARGET_STR_AVX3 \
|
HWY_TARGET_STR_AVX3 \
|
||||||
",vpclmulqdq,avx512vbmi2,vaes,avxvnni,avx512bitalg,avx512vpopcntdq"
|
",vpclmulqdq,avx512vbmi,avx512vbmi2,vaes,avxvnni,avx512bitalg," \
|
||||||
|
"avx512vpopcntdq"
|
||||||
|
|
||||||
#else
|
#else
|
||||||
#error "Logic error"
|
#error "Logic error"
|
||||||
|
@ -168,6 +187,8 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 0
|
#define HWY_HAVE_FLOAT16 0
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#define HWY_NATIVE_FMA 1
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
|
@ -186,8 +207,6 @@
|
||||||
#define HWY_HAVE_SCALABLE 0
|
#define HWY_HAVE_SCALABLE 0
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_CAP_GE256 0
|
|
||||||
#define HWY_CAP_GE512 0
|
|
||||||
|
|
||||||
#if HWY_ARCH_ARM_A64
|
#if HWY_ARCH_ARM_A64
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
@ -195,19 +214,38 @@
|
||||||
#define HWY_HAVE_FLOAT64 0
|
#define HWY_HAVE_FLOAT64 0
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
|
||||||
|
#if defined(__ARM_VFPV4__) || HWY_ARCH_ARM_A64
|
||||||
|
#define HWY_NATIVE_FMA 1
|
||||||
|
#else
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define HWY_CAP_GE256 0
|
||||||
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
#define HWY_NAMESPACE N_NEON
|
#define HWY_NAMESPACE N_NEON
|
||||||
|
|
||||||
// HWY_TARGET_STR remains undefined so HWY_ATTR is a no-op.
|
// Can use pragmas instead of -march compiler flag
|
||||||
|
#if HWY_HAVE_RUNTIME_DISPATCH
|
||||||
|
#if HWY_ARCH_ARM_V7
|
||||||
|
#define HWY_TARGET_STR "+neon-vfpv4"
|
||||||
|
#else
|
||||||
|
#define HWY_TARGET_STR "+crypto"
|
||||||
|
#endif // HWY_ARCH_ARM_V7
|
||||||
|
#else
|
||||||
|
// HWY_TARGET_STR remains undefined
|
||||||
|
#endif
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------
|
||||||
// SVE[2]
|
// SVE[2]
|
||||||
#elif HWY_TARGET == HWY_SVE2 || HWY_TARGET == HWY_SVE
|
#elif HWY_TARGET == HWY_SVE2 || HWY_TARGET == HWY_SVE || \
|
||||||
|
HWY_TARGET == HWY_SVE_256 || HWY_TARGET == HWY_SVE2_128
|
||||||
|
|
||||||
// SVE only requires lane alignment, not natural alignment of the entire vector.
|
// SVE only requires lane alignment, not natural alignment of the entire vector.
|
||||||
#define HWY_ALIGN alignas(8)
|
#define HWY_ALIGN alignas(8)
|
||||||
|
|
||||||
#define HWY_MAX_BYTES 256
|
|
||||||
|
|
||||||
// Value ensures MaxLanes() is the tightest possible upper bound to reduce
|
// Value ensures MaxLanes() is the tightest possible upper bound to reduce
|
||||||
// overallocation.
|
// overallocation.
|
||||||
#define HWY_LANES(T) ((HWY_MAX_BYTES) / sizeof(T))
|
#define HWY_LANES(T) ((HWY_MAX_BYTES) / sizeof(T))
|
||||||
|
@ -216,16 +254,35 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 0
|
||||||
|
#define HWY_NATIVE_FMA 1
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
#if HWY_TARGET == HWY_SVE2
|
#if HWY_TARGET == HWY_SVE2
|
||||||
#define HWY_NAMESPACE N_SVE2
|
#define HWY_NAMESPACE N_SVE2
|
||||||
|
#define HWY_MAX_BYTES 256
|
||||||
|
#elif HWY_TARGET == HWY_SVE_256
|
||||||
|
#define HWY_NAMESPACE N_SVE_256
|
||||||
|
#define HWY_MAX_BYTES 32
|
||||||
|
#elif HWY_TARGET == HWY_SVE2_128
|
||||||
|
#define HWY_NAMESPACE N_SVE2_128
|
||||||
|
#define HWY_MAX_BYTES 16
|
||||||
#else
|
#else
|
||||||
#define HWY_NAMESPACE N_SVE
|
#define HWY_NAMESPACE N_SVE
|
||||||
|
#define HWY_MAX_BYTES 256
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Can use pragmas instead of -march compiler flag
|
||||||
|
#if HWY_HAVE_RUNTIME_DISPATCH
|
||||||
|
#if HWY_TARGET == HWY_SVE2 || HWY_TARGET == HWY_SVE2_128
|
||||||
|
#define HWY_TARGET_STR "+sve2-aes"
|
||||||
|
#else
|
||||||
|
#define HWY_TARGET_STR "+sve"
|
||||||
|
#endif
|
||||||
|
#else
|
||||||
// HWY_TARGET_STR remains undefined
|
// HWY_TARGET_STR remains undefined
|
||||||
|
#endif
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------
|
||||||
// WASM
|
// WASM
|
||||||
|
@ -239,6 +296,8 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 0
|
#define HWY_HAVE_FLOAT64 0
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
|
@ -247,8 +306,8 @@
|
||||||
#define HWY_TARGET_STR "simd128"
|
#define HWY_TARGET_STR "simd128"
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------
|
||||||
// WASM2
|
// WASM_EMU256
|
||||||
#elif HWY_TARGET == HWY_WASM2
|
#elif HWY_TARGET == HWY_WASM_EMU256
|
||||||
|
|
||||||
#define HWY_ALIGN alignas(32)
|
#define HWY_ALIGN alignas(32)
|
||||||
#define HWY_MAX_BYTES 32
|
#define HWY_MAX_BYTES 32
|
||||||
|
@ -258,10 +317,12 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 0
|
#define HWY_HAVE_FLOAT64 0
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
#define HWY_NAMESPACE N_WASM2
|
#define HWY_NAMESPACE N_WASM_EMU256
|
||||||
|
|
||||||
#define HWY_TARGET_STR "simd128"
|
#define HWY_TARGET_STR "simd128"
|
||||||
|
|
||||||
|
@ -283,10 +344,12 @@
|
||||||
#define HWY_HAVE_SCALABLE 1
|
#define HWY_HAVE_SCALABLE 1
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 0
|
||||||
|
#define HWY_NATIVE_FMA 1
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
#if defined(__riscv_zfh)
|
#if defined(__riscv_zvfh)
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#else
|
#else
|
||||||
#define HWY_HAVE_FLOAT16 0
|
#define HWY_HAVE_FLOAT16 0
|
||||||
|
@ -297,6 +360,27 @@
|
||||||
// HWY_TARGET_STR remains undefined so HWY_ATTR is a no-op.
|
// HWY_TARGET_STR remains undefined so HWY_ATTR is a no-op.
|
||||||
// (rv64gcv is not a valid target)
|
// (rv64gcv is not a valid target)
|
||||||
|
|
||||||
|
//-----------------------------------------------------------------------------
|
||||||
|
// EMU128
|
||||||
|
#elif HWY_TARGET == HWY_EMU128
|
||||||
|
|
||||||
|
#define HWY_ALIGN alignas(16)
|
||||||
|
#define HWY_MAX_BYTES 16
|
||||||
|
#define HWY_LANES(T) (16 / sizeof(T))
|
||||||
|
|
||||||
|
#define HWY_HAVE_SCALABLE 0
|
||||||
|
#define HWY_HAVE_INTEGER64 1
|
||||||
|
#define HWY_HAVE_FLOAT16 1
|
||||||
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
|
#define HWY_CAP_GE256 0
|
||||||
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
|
#define HWY_NAMESPACE N_EMU128
|
||||||
|
|
||||||
|
// HWY_TARGET_STR remains undefined so HWY_ATTR is a no-op.
|
||||||
|
|
||||||
//-----------------------------------------------------------------------------
|
//-----------------------------------------------------------------------------
|
||||||
// SCALAR
|
// SCALAR
|
||||||
#elif HWY_TARGET == HWY_SCALAR
|
#elif HWY_TARGET == HWY_SCALAR
|
||||||
|
@ -309,6 +393,8 @@
|
||||||
#define HWY_HAVE_INTEGER64 1
|
#define HWY_HAVE_INTEGER64 1
|
||||||
#define HWY_HAVE_FLOAT16 1
|
#define HWY_HAVE_FLOAT16 1
|
||||||
#define HWY_HAVE_FLOAT64 1
|
#define HWY_HAVE_FLOAT64 1
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 0
|
||||||
|
#define HWY_NATIVE_FMA 0
|
||||||
#define HWY_CAP_GE256 0
|
#define HWY_CAP_GE256 0
|
||||||
#define HWY_CAP_GE512 0
|
#define HWY_CAP_GE512 0
|
||||||
|
|
||||||
|
@ -320,6 +406,12 @@
|
||||||
#pragma message("HWY_TARGET does not match any known target")
|
#pragma message("HWY_TARGET does not match any known target")
|
||||||
#endif // HWY_TARGET
|
#endif // HWY_TARGET
|
||||||
|
|
||||||
|
// Override this to 1 in asan/msan builds, which will still fault.
|
||||||
|
#if HWY_IS_ASAN || HWY_IS_MSAN
|
||||||
|
#undef HWY_MEM_OPS_MIGHT_FAULT
|
||||||
|
#define HWY_MEM_OPS_MIGHT_FAULT 1
|
||||||
|
#endif
|
||||||
|
|
||||||
// Clang <9 requires this be invoked at file scope, before any namespace.
|
// Clang <9 requires this be invoked at file scope, before any namespace.
|
||||||
#undef HWY_BEFORE_NAMESPACE
|
#undef HWY_BEFORE_NAMESPACE
|
||||||
#if defined(HWY_TARGET_STR)
|
#if defined(HWY_TARGET_STR)
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -98,21 +99,21 @@ struct Simd {
|
||||||
|
|
||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
#if HWY_HAVE_SCALABLE
|
|
||||||
|
|
||||||
template <typename T, size_t N, int kPow2>
|
template <typename T, size_t N, int kPow2>
|
||||||
constexpr bool IsFull(Simd<T, N, kPow2> /* d */) {
|
constexpr bool IsFull(Simd<T, N, kPow2> /* d */) {
|
||||||
return N == HWY_LANES(T) && kPow2 == 0;
|
return N == HWY_LANES(T) && kPow2 == 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Returns the number of lanes (possibly zero) after applying a shift:
|
// Returns the number of lanes (possibly zero) after applying a shift:
|
||||||
// - 0: no change;
|
// - 0: no change;
|
||||||
// - [1,3]: a group of 2,4,8 [fractional] vectors;
|
// - [1,3]: a group of 2,4,8 [fractional] vectors;
|
||||||
// - [-3,-1]: a fraction of a vector from 1/8 to 1/2.
|
// - [-3,-1]: a fraction of a vector from 1/8 to 1/2.
|
||||||
constexpr size_t ScaleByPower(size_t N, int pow2) {
|
constexpr size_t ScaleByPower(size_t N, int pow2) {
|
||||||
|
#if HWY_TARGET == HWY_RVV
|
||||||
return pow2 >= 0 ? (N << pow2) : (N >> (-pow2));
|
return pow2 >= 0 ? (N << pow2) : (N >> (-pow2));
|
||||||
|
#else
|
||||||
|
return pow2 >= 0 ? N : (N >> (-pow2));
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
// Struct wrappers enable validation of arguments via static_assert.
|
// Struct wrappers enable validation of arguments via static_assert.
|
||||||
|
@ -136,17 +137,16 @@ struct ScalableTagChecker {
|
||||||
template <typename T, size_t kLimit>
|
template <typename T, size_t kLimit>
|
||||||
struct CappedTagChecker {
|
struct CappedTagChecker {
|
||||||
static_assert(kLimit != 0, "Does not make sense to have zero lanes");
|
static_assert(kLimit != 0, "Does not make sense to have zero lanes");
|
||||||
using type = Simd<T, HWY_MIN(kLimit, HWY_MAX_BYTES / sizeof(T)), 0>;
|
// Safely handle non-power-of-two inputs by rounding down, which is allowed by
|
||||||
|
// CappedTag. Otherwise, Simd<T, 3, 0> would static_assert.
|
||||||
|
static constexpr size_t kLimitPow2 = size_t{1} << hwy::FloorLog2(kLimit);
|
||||||
|
using type = Simd<T, HWY_MIN(kLimitPow2, HWY_LANES(T)), 0>;
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename T, size_t kNumLanes>
|
template <typename T, size_t kNumLanes>
|
||||||
struct FixedTagChecker {
|
struct FixedTagChecker {
|
||||||
static_assert(kNumLanes != 0, "Does not make sense to have zero lanes");
|
static_assert(kNumLanes != 0, "Does not make sense to have zero lanes");
|
||||||
static_assert(kNumLanes * sizeof(T) <= HWY_MAX_BYTES, "Too many lanes");
|
static_assert(kNumLanes <= HWY_LANES(T), "Too many lanes");
|
||||||
#if HWY_TARGET == HWY_SCALAR
|
|
||||||
// HWY_MAX_BYTES would still allow uint8x8, which is not supported.
|
|
||||||
static_assert(kNumLanes == 1, "Scalar only supports one lane");
|
|
||||||
#endif
|
|
||||||
using type = Simd<T, kNumLanes, 0>;
|
using type = Simd<T, kNumLanes, 0>;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -172,8 +172,8 @@ template <typename T, size_t kLimit>
|
||||||
using CappedTag = typename detail::CappedTagChecker<T, kLimit>::type;
|
using CappedTag = typename detail::CappedTagChecker<T, kLimit>::type;
|
||||||
|
|
||||||
// Alias for a tag describing a vector with *exactly* kNumLanes active lanes,
|
// Alias for a tag describing a vector with *exactly* kNumLanes active lanes,
|
||||||
// even on targets with scalable vectors. HWY_SCALAR only supports one lane.
|
// even on targets with scalable vectors. Requires `kNumLanes` to be a power of
|
||||||
// All other targets allow kNumLanes up to HWY_MAX_BYTES / sizeof(T).
|
// two not exceeding `HWY_LANES(T)`.
|
||||||
//
|
//
|
||||||
// NOTE: if the application does not need to support HWY_SCALAR (+), use this
|
// NOTE: if the application does not need to support HWY_SCALAR (+), use this
|
||||||
// instead of CappedTag to emphasize that there will be exactly kNumLanes lanes.
|
// instead of CappedTag to emphasize that there will be exactly kNumLanes lanes.
|
||||||
|
@ -218,6 +218,15 @@ using Half = typename D::Half;
|
||||||
template <class D>
|
template <class D>
|
||||||
using Twice = typename D::Twice;
|
using Twice = typename D::Twice;
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
using Full32 = Simd<T, 4 / sizeof(T), 0>;
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
using Full64 = Simd<T, 8 / sizeof(T), 0>;
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
using Full128 = Simd<T, 16 / sizeof(T), 0>;
|
||||||
|
|
||||||
// Same as base.h macros but with a Simd<T, N, kPow2> argument instead of T.
|
// Same as base.h macros but with a Simd<T, N, kPow2> argument instead of T.
|
||||||
#define HWY_IF_UNSIGNED_D(D) HWY_IF_UNSIGNED(TFromD<D>)
|
#define HWY_IF_UNSIGNED_D(D) HWY_IF_UNSIGNED(TFromD<D>)
|
||||||
#define HWY_IF_SIGNED_D(D) HWY_IF_SIGNED(TFromD<D>)
|
#define HWY_IF_SIGNED_D(D) HWY_IF_SIGNED(TFromD<D>)
|
||||||
|
@ -232,15 +241,12 @@ using Twice = typename D::Twice;
|
||||||
#define HWY_IF_GE128_D(D) \
|
#define HWY_IF_GE128_D(D) \
|
||||||
hwy::EnableIf<D::kPrivateN * sizeof(TFromD<D>) >= 16>* = nullptr
|
hwy::EnableIf<D::kPrivateN * sizeof(TFromD<D>) >= 16>* = nullptr
|
||||||
|
|
||||||
// Same, but with a vector argument.
|
// Same, but with a vector argument. ops/*-inl.h define their own TFromV.
|
||||||
#define HWY_IF_UNSIGNED_V(V) HWY_IF_UNSIGNED(TFromV<V>)
|
#define HWY_IF_UNSIGNED_V(V) HWY_IF_UNSIGNED(TFromV<V>)
|
||||||
#define HWY_IF_SIGNED_V(V) HWY_IF_SIGNED(TFromV<V>)
|
#define HWY_IF_SIGNED_V(V) HWY_IF_SIGNED(TFromV<V>)
|
||||||
#define HWY_IF_FLOAT_V(V) HWY_IF_FLOAT(TFromV<V>)
|
#define HWY_IF_FLOAT_V(V) HWY_IF_FLOAT(TFromV<V>)
|
||||||
#define HWY_IF_LANE_SIZE_V(V, bytes) HWY_IF_LANE_SIZE(TFromV<V>, bytes)
|
#define HWY_IF_LANE_SIZE_V(V, bytes) HWY_IF_LANE_SIZE(TFromV<V>, bytes)
|
||||||
|
#define HWY_IF_NOT_LANE_SIZE_V(V, bytes) HWY_IF_NOT_LANE_SIZE(TFromV<V>, bytes)
|
||||||
// For implementing functions for a specific type.
|
|
||||||
// IsSame<...>() in template arguments is broken on MSVC2015.
|
|
||||||
#define HWY_IF_LANES_ARE(T, V) EnableIf<IsSameT<T, TFromV<V>>::value>* = nullptr
|
|
||||||
|
|
||||||
template <class D>
|
template <class D>
|
||||||
HWY_INLINE HWY_MAYBE_UNUSED constexpr int Pow2(D /* d */) {
|
HWY_INLINE HWY_MAYBE_UNUSED constexpr int Pow2(D /* d */) {
|
||||||
|
@ -291,8 +297,7 @@ HWY_INLINE HWY_MAYBE_UNUSED size_t Lanes(Simd<T, N, kPow2>) {
|
||||||
// We therefore pass by const& only on GCC and (Windows or ARM64). This alias
|
// We therefore pass by const& only on GCC and (Windows or ARM64). This alias
|
||||||
// must be used for all vector/mask parameters of functions marked HWY_NOINLINE,
|
// must be used for all vector/mask parameters of functions marked HWY_NOINLINE,
|
||||||
// and possibly also other functions that are not inlined.
|
// and possibly also other functions that are not inlined.
|
||||||
#if HWY_COMPILER_GCC && !HWY_COMPILER_CLANG && \
|
#if HWY_COMPILER_GCC_ACTUAL && (HWY_OS_WIN || HWY_ARCH_ARM_A64)
|
||||||
((defined(_WIN32) || defined(_WIN64)) || HWY_ARCH_ARM_A64)
|
|
||||||
template <class V>
|
template <class V>
|
||||||
using VecArg = const V&;
|
using VecArg = const V&;
|
||||||
#else
|
#else
|
||||||
|
|
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2021 Google LLC
|
// Copyright 2021 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -590,6 +591,10 @@ HWY_API Vec256<int16_t> MulHigh(const Vec256<int16_t> a,
|
||||||
return Vec256<int16_t>{wasm_i16x8_shuffle(l, h, 1, 3, 5, 7, 9, 11, 13, 15)};
|
return Vec256<int16_t>{wasm_i16x8_shuffle(l, h, 1, 3, 5, 7, 9, 11, 13, 15)};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
HWY_API Vec256<int16_t> MulFixedPoint15(Vec256<int16_t>, Vec256<int16_t>) {
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
}
|
||||||
|
|
||||||
// Multiplies even lanes (0, 2 ..) and returns the double-width result.
|
// Multiplies even lanes (0, 2 ..) and returns the double-width result.
|
||||||
HWY_API Vec256<int64_t> MulEven(const Vec256<int32_t> a,
|
HWY_API Vec256<int64_t> MulEven(const Vec256<int32_t> a,
|
||||||
const Vec256<int32_t> b) {
|
const Vec256<int32_t> b) {
|
||||||
|
@ -717,6 +722,37 @@ HWY_API Vec256<float> Floor(const Vec256<float> v) {
|
||||||
return Vec256<float>{wasm_f32x4_floor(v.raw)};
|
return Vec256<float>{wasm_f32x4_floor(v.raw)};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ Floating-point classification
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Mask256<T> IsNaN(const Vec256<T> v) {
|
||||||
|
return v != v;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T, HWY_IF_FLOAT(T)>
|
||||||
|
HWY_API Mask256<T> IsInf(const Vec256<T> v) {
|
||||||
|
const Full256<T> d;
|
||||||
|
const RebindToSigned<decltype(d)> di;
|
||||||
|
const VFromD<decltype(di)> vi = BitCast(di, v);
|
||||||
|
// 'Shift left' to clear the sign bit, check for exponent=max and mantissa=0.
|
||||||
|
return RebindMask(d, Eq(Add(vi, vi), Set(di, hwy::MaxExponentTimes2<T>())));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns whether normal/subnormal/zero.
|
||||||
|
template <typename T, HWY_IF_FLOAT(T)>
|
||||||
|
HWY_API Mask256<T> IsFinite(const Vec256<T> v) {
|
||||||
|
const Full256<T> d;
|
||||||
|
const RebindToUnsigned<decltype(d)> du;
|
||||||
|
const RebindToSigned<decltype(d)> di; // cheaper than unsigned comparison
|
||||||
|
const VFromD<decltype(du)> vu = BitCast(du, v);
|
||||||
|
// 'Shift left' to clear the sign bit, then right so we can compare with the
|
||||||
|
// max exponent (cannot compare with MaxExponentTimes2 directly because it is
|
||||||
|
// negative and non-negative floats would be greater).
|
||||||
|
const VFromD<decltype(di)> exp =
|
||||||
|
BitCast(di, ShiftRight<hwy::MantissaBits<T>() + 1>(Add(vu, vu)));
|
||||||
|
return RebindMask(d, Lt(exp, Set(di, hwy::MaxExponentField<T>())));
|
||||||
|
}
|
||||||
|
|
||||||
// ================================================== COMPARE
|
// ================================================== COMPARE
|
||||||
|
|
||||||
// Comparisons fill a lane with 1-bits if the condition is true, else 0.
|
// Comparisons fill a lane with 1-bits if the condition is true, else 0.
|
||||||
|
@ -910,6 +946,13 @@ HWY_API Vec256<T> Xor(Vec256<T> a, Vec256<T> b) {
|
||||||
return Vec256<T>{wasm_v128_xor(a.raw, b.raw)};
|
return Vec256<T>{wasm_v128_xor(a.raw, b.raw)};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ Or3
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Vec256<T> Or3(Vec256<T> o1, Vec256<T> o2, Vec256<T> o3) {
|
||||||
|
return Or(o1, Or(o2, o3));
|
||||||
|
}
|
||||||
|
|
||||||
// ------------------------------ OrAnd
|
// ------------------------------ OrAnd
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -1201,6 +1244,12 @@ HWY_API void StoreU(Vec256<T> v, Full256<T> d, T* HWY_RESTRICT p) {
|
||||||
Store(v, d, p);
|
Store(v, d, p);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_API void BlendedStore(Vec256<T> v, Mask256<T> m, Full256<T> d,
|
||||||
|
T* HWY_RESTRICT p) {
|
||||||
|
StoreU(IfThenElse(m, v, LoadU(d, p)), d, p);
|
||||||
|
}
|
||||||
|
|
||||||
// ------------------------------ Non-temporal stores
|
// ------------------------------ Non-temporal stores
|
||||||
|
|
||||||
// Same as aligned stores on non-x86.
|
// Same as aligned stores on non-x86.
|
||||||
|
@ -1281,8 +1330,19 @@ HWY_API Vec256<T> GatherIndex(const Full256<T> d, const T* HWY_RESTRICT base,
|
||||||
|
|
||||||
// ================================================== SWIZZLE
|
// ================================================== SWIZZLE
|
||||||
|
|
||||||
// ------------------------------ Extract lane
|
// ------------------------------ ExtractLane
|
||||||
|
template <typename T, size_t N>
|
||||||
|
HWY_API T ExtractLane(const Vec128<T, N> v, size_t i) {
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ InsertLane
|
||||||
|
template <typename T, size_t N>
|
||||||
|
HWY_API Vec128<T, N> InsertLane(const Vec128<T, N> v, size_t i, T t) {
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ GetLane
|
||||||
// Gets the single value stored in a vector/part.
|
// Gets the single value stored in a vector/part.
|
||||||
HWY_API uint8_t GetLane(const Vec256<uint8_t> v) {
|
HWY_API uint8_t GetLane(const Vec256<uint8_t> v) {
|
||||||
return wasm_i8x16_extract_lane(v.raw, 0);
|
return wasm_i8x16_extract_lane(v.raw, 0);
|
||||||
|
@ -2244,6 +2304,50 @@ HWY_API Vec256<uint8_t> U8FromU32(const Vec256<uint32_t> v) {
|
||||||
return Vec256<uint8_t>{wasm_u8x16_narrow_i16x8(intermediate, intermediate)};
|
return Vec256<uint8_t>{wasm_u8x16_narrow_i16x8(intermediate, intermediate)};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ Truncations
|
||||||
|
|
||||||
|
HWY_API Vec256<uint8_t, 4> TruncateTo(Simd<uint8_t, 4, 0> /* tag */,
|
||||||
|
const Vec256<uint64_t> v) {
|
||||||
|
return Vec256<uint8_t, 4>{wasm_i8x16_shuffle(v.v0.raw, v.v1.raw, 0, 8, 16, 24,
|
||||||
|
0, 8, 16, 24, 0, 8, 16, 24, 0, 8,
|
||||||
|
16, 24)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec256<uint16_t, 4> TruncateTo(Simd<uint16_t, 4, 0> /* tag */,
|
||||||
|
const Vec256<uint64_t> v) {
|
||||||
|
return Vec256<uint16_t, 4>{wasm_i8x16_shuffle(v.v0.raw, v.v1.raw, 0, 1, 8, 9,
|
||||||
|
16, 17, 24, 25, 0, 1, 8, 9, 16,
|
||||||
|
17, 24, 25)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec256<uint32_t, 4> TruncateTo(Simd<uint32_t, 4, 0> /* tag */,
|
||||||
|
const Vec256<uint64_t> v) {
|
||||||
|
return Vec256<uint32_t, 4>{wasm_i8x16_shuffle(v.v0.raw, v.v1.raw, 0, 1, 2, 3,
|
||||||
|
8, 9, 10, 11, 16, 17, 18, 19,
|
||||||
|
24, 25, 26, 27)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec256<uint8_t, 8> TruncateTo(Simd<uint8_t, 8, 0> /* tag */,
|
||||||
|
const Vec256<uint32_t> v) {
|
||||||
|
return Vec256<uint8_t, 8>{wasm_i8x16_shuffle(v.v0.raw, v.v1.raw, 0, 4, 8, 12,
|
||||||
|
16, 20, 24, 28, 0, 4, 8, 12, 16,
|
||||||
|
20, 24, 28)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec256<uint16_t, 8> TruncateTo(Simd<uint16_t, 8, 0> /* tag */,
|
||||||
|
const Vec256<uint32_t> v) {
|
||||||
|
return Vec256<uint16_t, 8>{wasm_i8x16_shuffle(v.v0.raw, v.v1.raw, 0, 1, 4, 5,
|
||||||
|
8, 9, 12, 13, 16, 17, 20, 21,
|
||||||
|
24, 25, 28, 29)};
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_API Vec256<uint8_t, 16> TruncateTo(Simd<uint8_t, 16, 0> /* tag */,
|
||||||
|
const Vec256<uint16_t> v) {
|
||||||
|
return Vec256<uint8_t, 16>{wasm_i8x16_shuffle(v.v0.raw, v.v1.raw, 0, 2, 4, 6,
|
||||||
|
8, 10, 12, 14, 16, 18, 20, 22,
|
||||||
|
24, 26, 28, 30)};
|
||||||
|
}
|
||||||
|
|
||||||
// ------------------------------ Convert i32 <=> f32 (Round)
|
// ------------------------------ Convert i32 <=> f32 (Round)
|
||||||
|
|
||||||
HWY_API Vec256<float> ConvertTo(Full256<float> /* tag */,
|
HWY_API Vec256<float> ConvertTo(Full256<float> /* tag */,
|
||||||
|
@ -2687,12 +2791,29 @@ HWY_INLINE Vec256<uint64_t> Compress(hwy::SizeTag<8> /*tag*/,
|
||||||
|
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
struct CompressIsPartition {
|
||||||
|
enum { value = 1 };
|
||||||
|
};
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_API Vec256<T> Compress(Vec256<T> v, const Mask256<T> mask) {
|
HWY_API Vec256<T> Compress(Vec256<T> v, const Mask256<T> mask) {
|
||||||
const uint64_t mask_bits = detail::BitsFromMask(mask);
|
const uint64_t mask_bits = detail::BitsFromMask(mask);
|
||||||
return detail::Compress(hwy::SizeTag<sizeof(T)>(), v, mask_bits);
|
return detail::Compress(hwy::SizeTag<sizeof(T)>(), v, mask_bits);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------ CompressNot
|
||||||
|
template <typename T>
|
||||||
|
HWY_API Vec256<T> Compress(Vec256<T> v, const Mask256<T> mask) {
|
||||||
|
return Compress(v, Not(mask));
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------ CompressBlocksNot
|
||||||
|
HWY_API Vec256<uint64_t> CompressBlocksNot(Vec256<uint64_t> v,
|
||||||
|
Mask256<uint64_t> mask) {
|
||||||
|
HWY_ASSERT(0);
|
||||||
|
}
|
||||||
|
|
||||||
// ------------------------------ CompressBits
|
// ------------------------------ CompressBits
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
|
@ -2750,76 +2871,10 @@ HWY_API size_t CompressBitsStore(Vec256<T> v, const uint8_t* HWY_RESTRICT bits,
|
||||||
return PopCount(mask_bits);
|
return PopCount(mask_bits);
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------ StoreInterleaved3 (CombineShiftRightBytes,
|
// ------------------------------ StoreInterleaved2/3/4
|
||||||
// TableLookupBytes)
|
|
||||||
|
|
||||||
HWY_API void StoreInterleaved3(const Vec256<uint8_t> a, const Vec256<uint8_t> b,
|
// HWY_NATIVE_LOAD_STORE_INTERLEAVED not set, hence defined in
|
||||||
const Vec256<uint8_t> c, Full256<uint8_t> d,
|
// generic_ops-inl.h.
|
||||||
uint8_t* HWY_RESTRICT unaligned) {
|
|
||||||
const auto k5 = Set(d, 5);
|
|
||||||
const auto k6 = Set(d, 6);
|
|
||||||
|
|
||||||
// Shuffle (a,b,c) vector bytes to (MSB on left): r5, bgr[4:0].
|
|
||||||
// 0x80 so lanes to be filled from other vectors are 0 for blending.
|
|
||||||
alignas(32) static constexpr uint8_t tbl_r0[16] = {
|
|
||||||
0, 0x80, 0x80, 1, 0x80, 0x80, 2, 0x80, 0x80, //
|
|
||||||
3, 0x80, 0x80, 4, 0x80, 0x80, 5};
|
|
||||||
alignas(32) static constexpr uint8_t tbl_g0[16] = {
|
|
||||||
0x80, 0, 0x80, 0x80, 1, 0x80, //
|
|
||||||
0x80, 2, 0x80, 0x80, 3, 0x80, 0x80, 4, 0x80, 0x80};
|
|
||||||
const auto shuf_r0 = Load(d, tbl_r0);
|
|
||||||
const auto shuf_g0 = Load(d, tbl_g0); // cannot reuse r0 due to 5 in MSB
|
|
||||||
const auto shuf_b0 = CombineShiftRightBytes<15>(d, shuf_g0, shuf_g0);
|
|
||||||
const auto r0 = TableLookupBytes(a, shuf_r0); // 5..4..3..2..1..0
|
|
||||||
const auto g0 = TableLookupBytes(b, shuf_g0); // ..4..3..2..1..0.
|
|
||||||
const auto b0 = TableLookupBytes(c, shuf_b0); // .4..3..2..1..0..
|
|
||||||
const auto int0 = r0 | g0 | b0;
|
|
||||||
StoreU(int0, d, unaligned + 0 * 16);
|
|
||||||
|
|
||||||
// Second vector: g10,r10, bgr[9:6], b5,g5
|
|
||||||
const auto shuf_r1 = shuf_b0 + k6; // .A..9..8..7..6..
|
|
||||||
const auto shuf_g1 = shuf_r0 + k5; // A..9..8..7..6..5
|
|
||||||
const auto shuf_b1 = shuf_g0 + k5; // ..9..8..7..6..5.
|
|
||||||
const auto r1 = TableLookupBytes(a, shuf_r1);
|
|
||||||
const auto g1 = TableLookupBytes(b, shuf_g1);
|
|
||||||
const auto b1 = TableLookupBytes(c, shuf_b1);
|
|
||||||
const auto int1 = r1 | g1 | b1;
|
|
||||||
StoreU(int1, d, unaligned + 1 * 16);
|
|
||||||
|
|
||||||
// Third vector: bgr[15:11], b10
|
|
||||||
const auto shuf_r2 = shuf_b1 + k6; // ..F..E..D..C..B.
|
|
||||||
const auto shuf_g2 = shuf_r1 + k5; // .F..E..D..C..B..
|
|
||||||
const auto shuf_b2 = shuf_g1 + k5; // F..E..D..C..B..A
|
|
||||||
const auto r2 = TableLookupBytes(a, shuf_r2);
|
|
||||||
const auto g2 = TableLookupBytes(b, shuf_g2);
|
|
||||||
const auto b2 = TableLookupBytes(c, shuf_b2);
|
|
||||||
const auto int2 = r2 | g2 | b2;
|
|
||||||
StoreU(int2, d, unaligned + 2 * 16);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------ StoreInterleaved4
|
|
||||||
|
|
||||||
HWY_API void StoreInterleaved4(const Vec256<uint8_t> v0,
|
|
||||||
const Vec256<uint8_t> v1,
|
|
||||||
const Vec256<uint8_t> v2,
|
|
||||||
const Vec256<uint8_t> v3, Full256<uint8_t> d8,
|
|
||||||
uint8_t* HWY_RESTRICT unaligned) {
|
|
||||||
const RepartitionToWide<decltype(d8)> d16;
|
|
||||||
const RepartitionToWide<decltype(d16)> d32;
|
|
||||||
// let a,b,c,d denote v0..3.
|
|
||||||
const auto ba0 = ZipLower(d16, v0, v1); // b7 a7 .. b0 a0
|
|
||||||
const auto dc0 = ZipLower(d16, v2, v3); // d7 c7 .. d0 c0
|
|
||||||
const auto ba8 = ZipUpper(d16, v0, v1);
|
|
||||||
const auto dc8 = ZipUpper(d16, v2, v3);
|
|
||||||
const auto dcba_0 = ZipLower(d32, ba0, dc0); // d..a3 d..a0
|
|
||||||
const auto dcba_4 = ZipUpper(d32, ba0, dc0); // d..a7 d..a4
|
|
||||||
const auto dcba_8 = ZipLower(d32, ba8, dc8); // d..aB d..a8
|
|
||||||
const auto dcba_C = ZipUpper(d32, ba8, dc8); // d..aF d..aC
|
|
||||||
StoreU(BitCast(d8, dcba_0), d8, unaligned + 0 * 16);
|
|
||||||
StoreU(BitCast(d8, dcba_4), d8, unaligned + 1 * 16);
|
|
||||||
StoreU(BitCast(d8, dcba_8), d8, unaligned + 2 * 16);
|
|
||||||
StoreU(BitCast(d8, dcba_C), d8, unaligned + 3 * 16);
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------ MulEven/Odd (Load)
|
// ------------------------------ MulEven/Odd (Load)
|
||||||
|
|
||||||
|
@ -2952,12 +3007,27 @@ HWY_API Vec256<T> MaxOfLanes(Full256<T> /* tag */, const Vec256<T> v) {
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_INLINE Mask256<T> Lt128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
HWY_INLINE Mask256<T> Lt128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Mask256<T> Lt128Upper(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Mask256<T> Eq128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Mask256<T> Eq128Upper(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_INLINE Vec256<T> Min128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
HWY_INLINE Vec256<T> Min128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
HWY_INLINE Vec256<T> Max128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
HWY_INLINE Vec256<T> Max128(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Vec256<T> Min128Upper(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE Vec256<T> Max128Upper(Full256<T> d, Vec256<T> a, Vec256<T> b) {}
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
} // namespace HWY_NAMESPACE
|
} // namespace HWY_NAMESPACE
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
|
@ -0,0 +1,50 @@
|
||||||
|
// Copyright 2022 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/per_target.h"
|
||||||
|
|
||||||
|
#undef HWY_TARGET_INCLUDE
|
||||||
|
#define HWY_TARGET_INCLUDE "hwy/per_target.cc"
|
||||||
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
|
#include "hwy/highway.h"
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
// On SVE, Lanes rounds down to a power of two, but we want to know the actual
|
||||||
|
// size here. Otherwise, hypothetical SVE with 48 bytes would round down to 32
|
||||||
|
// and we'd enable HWY_SVE_256, and then fail reverse_test because Reverse on
|
||||||
|
// HWY_SVE_256 requires the actual vector to be a power of two.
|
||||||
|
#if HWY_TARGET == HWY_SVE || HWY_TARGET == HWY_SVE2 || HWY_TARGET == HWY_SVE_256
|
||||||
|
size_t GetVectorBytes() { return detail::AllHardwareLanes(hwy::SizeTag<1>()); }
|
||||||
|
#else
|
||||||
|
size_t GetVectorBytes() { return Lanes(ScalableTag<uint8_t>()); }
|
||||||
|
#endif
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#if HWY_ONCE
|
||||||
|
namespace hwy {
|
||||||
|
namespace {
|
||||||
|
HWY_EXPORT(GetVectorBytes); // Local function.
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
size_t VectorBytes() { return HWY_DYNAMIC_DISPATCH(GetVectorBytes)(); }
|
||||||
|
|
||||||
|
} // namespace hwy
|
||||||
|
#endif // HWY_ONCE
|
|
@ -0,0 +1,37 @@
|
||||||
|
// Copyright 2022 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.
|
||||||
|
|
||||||
|
#ifndef HIGHWAY_HWY_PER_TARGET_H_
|
||||||
|
#define HIGHWAY_HWY_PER_TARGET_H_
|
||||||
|
|
||||||
|
#include <stddef.h>
|
||||||
|
|
||||||
|
// Per-target functions.
|
||||||
|
|
||||||
|
namespace hwy {
|
||||||
|
|
||||||
|
// Returns size in bytes of a vector, i.e. `Lanes(ScalableTag<uint8_t>())`.
|
||||||
|
//
|
||||||
|
// Do not cache the result, which may change after calling DisableTargets, or
|
||||||
|
// if software requests a different vector size (e.g. when entering/exiting SME
|
||||||
|
// streaming mode). Instead call this right before the code that depends on the
|
||||||
|
// result, without any DisableTargets or SME transition in-between. Note that
|
||||||
|
// this involves an indirect call, so prefer not to call this frequently nor
|
||||||
|
// unnecessarily.
|
||||||
|
size_t VectorBytes();
|
||||||
|
|
||||||
|
} // namespace hwy
|
||||||
|
|
||||||
|
#endif // HIGHWAY_HWY_PER_TARGET_H_
|
|
@ -0,0 +1,56 @@
|
||||||
|
// Copyright 2022 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.
|
||||||
|
|
||||||
|
// Print() function
|
||||||
|
|
||||||
|
#include <inttypes.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
#include "hwy/aligned_allocator.h"
|
||||||
|
#include "hwy/highway.h"
|
||||||
|
#include "hwy/print.h"
|
||||||
|
|
||||||
|
// Per-target include guard
|
||||||
|
#if defined(HIGHWAY_HWY_PRINT_INL_H_) == \
|
||||||
|
defined(HWY_TARGET_TOGGLE)
|
||||||
|
#ifdef HIGHWAY_HWY_PRINT_INL_H_
|
||||||
|
#undef HIGHWAY_HWY_PRINT_INL_H_
|
||||||
|
#else
|
||||||
|
#define HIGHWAY_HWY_PRINT_INL_H_
|
||||||
|
#endif
|
||||||
|
|
||||||
|
HWY_BEFORE_NAMESPACE();
|
||||||
|
namespace hwy {
|
||||||
|
namespace HWY_NAMESPACE {
|
||||||
|
|
||||||
|
// Prints lanes around `lane`, in memory order.
|
||||||
|
template <class D, class V = Vec<D>>
|
||||||
|
void Print(const D d, const char* caption, VecArg<V> v, size_t lane_u = 0,
|
||||||
|
size_t max_lanes = 7) {
|
||||||
|
const size_t N = Lanes(d);
|
||||||
|
using T = TFromD<D>;
|
||||||
|
auto lanes = AllocateAligned<T>(N);
|
||||||
|
Store(v, d, lanes.get());
|
||||||
|
|
||||||
|
const auto info = hwy::detail::MakeTypeInfo<T>();
|
||||||
|
hwy::detail::PrintArray(info, caption, lanes.get(), N, lane_u, max_lanes);
|
||||||
|
}
|
||||||
|
|
||||||
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
} // namespace HWY_NAMESPACE
|
||||||
|
} // namespace hwy
|
||||||
|
HWY_AFTER_NAMESPACE();
|
||||||
|
|
||||||
|
#endif // per-target include guard
|
|
@ -0,0 +1,107 @@
|
||||||
|
// Copyright 2022 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/print.h"
|
||||||
|
|
||||||
|
#include <inttypes.h>
|
||||||
|
#include <stddef.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
#include "hwy/base.h"
|
||||||
|
|
||||||
|
namespace hwy {
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void TypeName(const TypeInfo& info, size_t N, char* string100) {
|
||||||
|
const char prefix = info.is_float ? 'f' : (info.is_signed ? 'i' : 'u');
|
||||||
|
// Omit the xN suffix for scalars.
|
||||||
|
if (N == 1) {
|
||||||
|
// NOLINTNEXTLINE
|
||||||
|
snprintf(string100, 64, "%c%d", prefix,
|
||||||
|
static_cast<int>(info.sizeof_t * 8));
|
||||||
|
} else {
|
||||||
|
// NOLINTNEXTLINE
|
||||||
|
snprintf(string100, 64, "%c%dx%d", prefix,
|
||||||
|
static_cast<int>(info.sizeof_t * 8), static_cast<int>(N));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void ToString(const TypeInfo& info, const void* ptr,
|
||||||
|
char* string100) {
|
||||||
|
if (info.sizeof_t == 1) {
|
||||||
|
uint8_t byte;
|
||||||
|
CopyBytes<1>(ptr, &byte); // endian-safe: we ensured sizeof(T)=1.
|
||||||
|
snprintf(string100, 100, "0x%02X", byte); // NOLINT
|
||||||
|
} else if (info.sizeof_t == 2) {
|
||||||
|
uint16_t bits;
|
||||||
|
CopyBytes<2>(ptr, &bits);
|
||||||
|
snprintf(string100, 100, "0x%04X", bits); // NOLINT
|
||||||
|
} else if (info.sizeof_t == 4) {
|
||||||
|
if (info.is_float) {
|
||||||
|
float value;
|
||||||
|
CopyBytes<4>(ptr, &value);
|
||||||
|
snprintf(string100, 100, "%g", static_cast<double>(value)); // NOLINT
|
||||||
|
} else if (info.is_signed) {
|
||||||
|
int32_t value;
|
||||||
|
CopyBytes<4>(ptr, &value);
|
||||||
|
snprintf(string100, 100, "%d", value); // NOLINT
|
||||||
|
} else {
|
||||||
|
uint32_t value;
|
||||||
|
CopyBytes<4>(ptr, &value);
|
||||||
|
snprintf(string100, 100, "%u", value); // NOLINT
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
HWY_ASSERT(info.sizeof_t == 8);
|
||||||
|
if (info.is_float) {
|
||||||
|
double value;
|
||||||
|
CopyBytes<8>(ptr, &value);
|
||||||
|
snprintf(string100, 100, "%g", value); // NOLINT
|
||||||
|
} else if (info.is_signed) {
|
||||||
|
int64_t value;
|
||||||
|
CopyBytes<8>(ptr, &value);
|
||||||
|
snprintf(string100, 100, "%" PRIi64 "", value); // NOLINT
|
||||||
|
} else {
|
||||||
|
uint64_t value;
|
||||||
|
CopyBytes<8>(ptr, &value);
|
||||||
|
snprintf(string100, 100, "%" PRIu64 "", value); // NOLINT
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void PrintArray(const TypeInfo& info, const char* caption,
|
||||||
|
const void* array_void, size_t N, size_t lane_u,
|
||||||
|
size_t max_lanes) {
|
||||||
|
const uint8_t* array_bytes = reinterpret_cast<const uint8_t*>(array_void);
|
||||||
|
|
||||||
|
char type_name[100];
|
||||||
|
TypeName(info, N, type_name);
|
||||||
|
|
||||||
|
const intptr_t lane = intptr_t(lane_u);
|
||||||
|
const size_t begin = static_cast<size_t>(HWY_MAX(0, lane - 2));
|
||||||
|
const size_t end = HWY_MIN(begin + max_lanes, N);
|
||||||
|
fprintf(stderr, "%s %s [%" PRIu64 "+ ->]:\n ", type_name, caption,
|
||||||
|
static_cast<uint64_t>(begin));
|
||||||
|
for (size_t i = begin; i < end; ++i) {
|
||||||
|
const void* ptr = array_bytes + i * info.sizeof_t;
|
||||||
|
char str[100];
|
||||||
|
ToString(info, ptr, str);
|
||||||
|
fprintf(stderr, "%s,", str);
|
||||||
|
}
|
||||||
|
if (begin >= end) fprintf(stderr, "(out of bounds)");
|
||||||
|
fprintf(stderr, "\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
} // namespace hwy
|
|
@ -0,0 +1,73 @@
|
||||||
|
// Copyright 2022 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.
|
||||||
|
|
||||||
|
#ifndef HWY_PRINT_H_
|
||||||
|
#define HWY_PRINT_H_
|
||||||
|
|
||||||
|
// Helpers for printing vector lanes.
|
||||||
|
|
||||||
|
#include <stddef.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
#include "hwy/base.h"
|
||||||
|
#include "hwy/highway_export.h"
|
||||||
|
|
||||||
|
namespace hwy {
|
||||||
|
|
||||||
|
namespace detail {
|
||||||
|
|
||||||
|
// For implementing value comparisons etc. as type-erased functions to reduce
|
||||||
|
// template bloat.
|
||||||
|
struct TypeInfo {
|
||||||
|
size_t sizeof_t;
|
||||||
|
bool is_float;
|
||||||
|
bool is_signed;
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_INLINE TypeInfo MakeTypeInfo() {
|
||||||
|
TypeInfo info;
|
||||||
|
info.sizeof_t = sizeof(T);
|
||||||
|
info.is_float = IsFloat<T>();
|
||||||
|
info.is_signed = IsSigned<T>();
|
||||||
|
return info;
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void TypeName(const TypeInfo& info, size_t N, char* string100);
|
||||||
|
HWY_DLLEXPORT void ToString(const TypeInfo& info, const void* ptr,
|
||||||
|
char* string100);
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void PrintArray(const TypeInfo& info, const char* caption,
|
||||||
|
const void* array_void, size_t N,
|
||||||
|
size_t lane_u = 0, size_t max_lanes = 7);
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_NOINLINE void PrintValue(T value) {
|
||||||
|
char str[100];
|
||||||
|
detail::ToString(hwy::detail::MakeTypeInfo<T>(), &value, str);
|
||||||
|
fprintf(stderr, "%s,", str);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
HWY_NOINLINE void PrintArray(const T* value, size_t count) {
|
||||||
|
detail::PrintArray(hwy::detail::MakeTypeInfo<T>(), "", value, count, 0,
|
||||||
|
count);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace hwy
|
||||||
|
|
||||||
|
#endif // HWY_PRINT_H_
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -14,6 +15,7 @@
|
||||||
|
|
||||||
#include "hwy/targets.h"
|
#include "hwy/targets.h"
|
||||||
|
|
||||||
|
#include <inttypes.h> // PRIx64
|
||||||
#include <stdarg.h>
|
#include <stdarg.h>
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
@ -21,7 +23,7 @@
|
||||||
|
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
|
||||||
#include "hwy/base.h"
|
#include "hwy/per_target.h"
|
||||||
|
|
||||||
#if HWY_IS_ASAN || HWY_IS_MSAN || HWY_IS_TSAN
|
#if HWY_IS_ASAN || HWY_IS_MSAN || HWY_IS_TSAN
|
||||||
#include "sanitizer/common_interface_defs.h" // __sanitizer_print_stack_trace
|
#include "sanitizer/common_interface_defs.h" // __sanitizer_print_stack_trace
|
||||||
|
@ -36,7 +38,11 @@
|
||||||
#else // !HWY_COMPILER_MSVC
|
#else // !HWY_COMPILER_MSVC
|
||||||
#include <cpuid.h>
|
#include <cpuid.h>
|
||||||
#endif // HWY_COMPILER_MSVC
|
#endif // HWY_COMPILER_MSVC
|
||||||
#endif // HWY_ARCH_X86
|
|
||||||
|
#elif HWY_ARCH_ARM && HWY_OS_LINUX
|
||||||
|
#include <asm/hwcap.h>
|
||||||
|
#include <sys/auxv.h>
|
||||||
|
#endif // HWY_ARCH_*
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
namespace {
|
namespace {
|
||||||
|
@ -87,15 +93,12 @@ uint32_t ReadXCR0() {
|
||||||
|
|
||||||
#endif // HWY_ARCH_X86
|
#endif // HWY_ARCH_X86
|
||||||
|
|
||||||
// Not function-local => no compiler-generated locking.
|
|
||||||
std::atomic<uint32_t> supported_{0}; // Not yet initialized
|
|
||||||
|
|
||||||
// When running tests, this value can be set to the mocked supported targets
|
// When running tests, this value can be set to the mocked supported targets
|
||||||
// mask. Only written to from a single thread before the test starts.
|
// mask. Only written to from a single thread before the test starts.
|
||||||
uint32_t supported_targets_for_test_ = 0;
|
int64_t supported_targets_for_test_ = 0;
|
||||||
|
|
||||||
// Mask of targets disabled at runtime with DisableTargets.
|
// Mask of targets disabled at runtime with DisableTargets.
|
||||||
uint32_t supported_mask_{LimitsMax<uint32_t>()};
|
int64_t supported_mask_ = LimitsMax<int64_t>();
|
||||||
|
|
||||||
#if HWY_ARCH_X86
|
#if HWY_ARCH_X86
|
||||||
// Arbritrary bit indices indicating which instruction set extensions are
|
// Arbritrary bit indices indicating which instruction set extensions are
|
||||||
|
@ -126,6 +129,7 @@ enum class FeatureIndex : uint32_t {
|
||||||
|
|
||||||
kVNNI,
|
kVNNI,
|
||||||
kVPCLMULQDQ,
|
kVPCLMULQDQ,
|
||||||
|
kVBMI,
|
||||||
kVBMI2,
|
kVBMI2,
|
||||||
kVAES,
|
kVAES,
|
||||||
kPOPCNTDQ,
|
kPOPCNTDQ,
|
||||||
|
@ -176,77 +180,19 @@ constexpr uint64_t kGroupAVX3 =
|
||||||
|
|
||||||
constexpr uint64_t kGroupAVX3_DL =
|
constexpr uint64_t kGroupAVX3_DL =
|
||||||
Bit(FeatureIndex::kVNNI) | Bit(FeatureIndex::kVPCLMULQDQ) |
|
Bit(FeatureIndex::kVNNI) | Bit(FeatureIndex::kVPCLMULQDQ) |
|
||||||
Bit(FeatureIndex::kVBMI2) | Bit(FeatureIndex::kVAES) |
|
Bit(FeatureIndex::kVBMI) | Bit(FeatureIndex::kVBMI2) |
|
||||||
Bit(FeatureIndex::kPOPCNTDQ) | Bit(FeatureIndex::kBITALG) | kGroupAVX3;
|
Bit(FeatureIndex::kVAES) | Bit(FeatureIndex::kPOPCNTDQ) |
|
||||||
|
Bit(FeatureIndex::kBITALG) | kGroupAVX3;
|
||||||
|
|
||||||
#endif // HWY_ARCH_X86
|
#endif // HWY_ARCH_X86
|
||||||
|
|
||||||
} // namespace
|
// Returns targets supported by the CPU, independently of DisableTargets.
|
||||||
|
// Factored out of SupportedTargets to make its structure more obvious. Note
|
||||||
HWY_NORETURN void HWY_FORMAT(3, 4)
|
// that x86 CPUID may take several hundred cycles.
|
||||||
Abort(const char* file, int line, const char* format, ...) {
|
int64_t DetectTargets() {
|
||||||
char buf[2000];
|
// Apps will use only one of these (the default is EMU128), but compile flags
|
||||||
va_list args;
|
// for this TU may differ from that of the app, so allow both.
|
||||||
va_start(args, format);
|
int64_t bits = HWY_SCALAR | HWY_EMU128;
|
||||||
vsnprintf(buf, sizeof(buf), format, args);
|
|
||||||
va_end(args);
|
|
||||||
|
|
||||||
fprintf(stderr, "Abort at %s:%d: %s\n", file, line, buf);
|
|
||||||
|
|
||||||
// If compiled with any sanitizer, they can also print a stack trace.
|
|
||||||
#if HWY_IS_ASAN || HWY_IS_MSAN || HWY_IS_TSAN
|
|
||||||
__sanitizer_print_stack_trace();
|
|
||||||
#endif // HWY_IS_*
|
|
||||||
fflush(stderr);
|
|
||||||
|
|
||||||
// Now terminate the program:
|
|
||||||
#if HWY_ARCH_RVV
|
|
||||||
exit(1); // trap/abort just freeze Spike.
|
|
||||||
#elif HWY_IS_DEBUG_BUILD && !HWY_COMPILER_MSVC
|
|
||||||
// Facilitates breaking into a debugger, but don't use this in non-debug
|
|
||||||
// builds because it looks like "illegal instruction", which is misleading.
|
|
||||||
__builtin_trap();
|
|
||||||
#else
|
|
||||||
abort(); // Compile error without this due to HWY_NORETURN.
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void DisableTargets(uint32_t disabled_targets) {
|
|
||||||
supported_mask_ = ~(disabled_targets & ~uint32_t(HWY_ENABLED_BASELINE));
|
|
||||||
// We can call Update() here to initialize the mask but that will trigger a
|
|
||||||
// call to SupportedTargets() which we use in tests to tell whether any of the
|
|
||||||
// highway dynamic dispatch functions were used.
|
|
||||||
GetChosenTarget().DeInit();
|
|
||||||
}
|
|
||||||
|
|
||||||
void SetSupportedTargetsForTest(uint32_t targets) {
|
|
||||||
// Reset the cached supported_ value to 0 to force a re-evaluation in the
|
|
||||||
// next call to SupportedTargets() which will use the mocked value set here
|
|
||||||
// if not zero.
|
|
||||||
supported_.store(0, std::memory_order_release);
|
|
||||||
supported_targets_for_test_ = targets;
|
|
||||||
GetChosenTarget().DeInit();
|
|
||||||
}
|
|
||||||
|
|
||||||
bool SupportedTargetsCalledForTest() {
|
|
||||||
return supported_.load(std::memory_order_acquire) != 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
uint32_t SupportedTargets() {
|
|
||||||
uint32_t bits = supported_.load(std::memory_order_acquire);
|
|
||||||
// Already initialized?
|
|
||||||
if (HWY_LIKELY(bits != 0)) {
|
|
||||||
return bits & supported_mask_;
|
|
||||||
}
|
|
||||||
|
|
||||||
// When running tests, this allows to mock the current supported targets.
|
|
||||||
if (HWY_UNLIKELY(supported_targets_for_test_ != 0)) {
|
|
||||||
// Store the value to signal that this was used.
|
|
||||||
supported_.store(supported_targets_for_test_, std::memory_order_release);
|
|
||||||
return supported_targets_for_test_ & supported_mask_;
|
|
||||||
}
|
|
||||||
|
|
||||||
bits = HWY_SCALAR;
|
|
||||||
|
|
||||||
#if HWY_ARCH_X86
|
#if HWY_ARCH_X86
|
||||||
bool has_osxsave = false;
|
bool has_osxsave = false;
|
||||||
|
@ -288,6 +234,7 @@ uint32_t SupportedTargets() {
|
||||||
flags |= IsBitSet(abcd[1], 30) ? Bit(FeatureIndex::kAVX512BW) : 0;
|
flags |= IsBitSet(abcd[1], 30) ? Bit(FeatureIndex::kAVX512BW) : 0;
|
||||||
flags |= IsBitSet(abcd[1], 31) ? Bit(FeatureIndex::kAVX512VL) : 0;
|
flags |= IsBitSet(abcd[1], 31) ? Bit(FeatureIndex::kAVX512VL) : 0;
|
||||||
|
|
||||||
|
flags |= IsBitSet(abcd[2], 1) ? Bit(FeatureIndex::kVBMI) : 0;
|
||||||
flags |= IsBitSet(abcd[2], 6) ? Bit(FeatureIndex::kVBMI2) : 0;
|
flags |= IsBitSet(abcd[2], 6) ? Bit(FeatureIndex::kVBMI2) : 0;
|
||||||
flags |= IsBitSet(abcd[2], 9) ? Bit(FeatureIndex::kVAES) : 0;
|
flags |= IsBitSet(abcd[2], 9) ? Bit(FeatureIndex::kVAES) : 0;
|
||||||
flags |= IsBitSet(abcd[2], 10) ? Bit(FeatureIndex::kVPCLMULQDQ) : 0;
|
flags |= IsBitSet(abcd[2], 10) ? Bit(FeatureIndex::kVPCLMULQDQ) : 0;
|
||||||
|
@ -318,33 +265,162 @@ uint32_t SupportedTargets() {
|
||||||
// are not preserved across context switches.
|
// are not preserved across context switches.
|
||||||
if (has_osxsave) {
|
if (has_osxsave) {
|
||||||
const uint32_t xcr0 = ReadXCR0();
|
const uint32_t xcr0 = ReadXCR0();
|
||||||
|
const int64_t min_avx3 = HWY_AVX3 | HWY_AVX3_DL;
|
||||||
|
const int64_t min_avx2 = HWY_AVX2 | min_avx3;
|
||||||
// XMM
|
// XMM
|
||||||
if (!IsBitSet(xcr0, 1)) {
|
if (!IsBitSet(xcr0, 1)) {
|
||||||
bits &=
|
bits &= ~(HWY_SSSE3 | HWY_SSE4 | min_avx2);
|
||||||
~uint32_t(HWY_SSSE3 | HWY_SSE4 | HWY_AVX2 | HWY_AVX3 | HWY_AVX3_DL);
|
|
||||||
}
|
}
|
||||||
// YMM
|
// YMM
|
||||||
if (!IsBitSet(xcr0, 2)) {
|
if (!IsBitSet(xcr0, 2)) {
|
||||||
bits &= ~uint32_t(HWY_AVX2 | HWY_AVX3 | HWY_AVX3_DL);
|
bits &= ~min_avx2;
|
||||||
}
|
}
|
||||||
// ZMM + opmask
|
// opmask, ZMM lo/hi
|
||||||
if ((xcr0 & 0x70) != 0x70) {
|
if (!IsBitSet(xcr0, 5) || !IsBitSet(xcr0, 6) || !IsBitSet(xcr0, 7)) {
|
||||||
bits &= ~uint32_t(HWY_AVX3 | HWY_AVX3_DL);
|
bits &= ~min_avx3;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#else
|
|
||||||
// TODO(janwas): detect for other platforms
|
|
||||||
bits = HWY_ENABLED_BASELINE;
|
|
||||||
#endif // HWY_ARCH_X86
|
|
||||||
|
|
||||||
if ((bits & HWY_ENABLED_BASELINE) != HWY_ENABLED_BASELINE) {
|
if ((bits & HWY_ENABLED_BASELINE) != HWY_ENABLED_BASELINE) {
|
||||||
fprintf(stderr, "WARNING: CPU supports %zx but software requires %x\n",
|
fprintf(stderr,
|
||||||
size_t(bits), HWY_ENABLED_BASELINE);
|
"WARNING: CPU supports %" PRIx64 " but software requires %" PRIx64
|
||||||
|
"\n",
|
||||||
|
bits, static_cast<int64_t>(HWY_ENABLED_BASELINE));
|
||||||
}
|
}
|
||||||
|
|
||||||
supported_.store(bits, std::memory_order_release);
|
#elif HWY_ARCH_ARM && HWY_HAVE_RUNTIME_DISPATCH
|
||||||
return bits & supported_mask_;
|
using CapBits = unsigned long; // NOLINT
|
||||||
|
const CapBits hw = getauxval(AT_HWCAP);
|
||||||
|
(void)hw;
|
||||||
|
|
||||||
|
#if HWY_ARCH_ARM_A64
|
||||||
|
|
||||||
|
#if defined(HWCAP_AES)
|
||||||
|
// aarch64 always has NEON and VFPv4, but not necessarily AES, which we
|
||||||
|
// require and thus must still check for.
|
||||||
|
if (hw & HWCAP_AES) {
|
||||||
|
bits |= HWY_NEON;
|
||||||
|
}
|
||||||
|
#endif // HWCAP_AES
|
||||||
|
|
||||||
|
#if defined(HWCAP_SVE)
|
||||||
|
if (hw & HWCAP_SVE) {
|
||||||
|
bits |= HWY_SVE;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if defined(HWCAP2_SVE2) && defined(HWCAP2_SVEAES)
|
||||||
|
const CapBits hw2 = getauxval(AT_HWCAP2);
|
||||||
|
if ((hw2 & HWCAP2_SVE2) && (hw2 & HWCAP2_SVEAES)) {
|
||||||
|
bits |= HWY_SVE2;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#else // HWY_ARCH_ARM_A64
|
||||||
|
|
||||||
|
// Some old auxv.h / hwcap.h do not define these. If not, treat as unsupported.
|
||||||
|
// Note that AES has a different HWCAP bit compared to aarch64.
|
||||||
|
#if defined(HWCAP_NEON) && defined(HWCAP_VFPv4)
|
||||||
|
if ((hw & HWCAP_NEON) && (hw & HWCAP_VFPv4)) {
|
||||||
|
bits |= HWY_NEON;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif // HWY_ARCH_ARM_A64
|
||||||
|
if ((bits & HWY_ENABLED_BASELINE) != HWY_ENABLED_BASELINE) {
|
||||||
|
fprintf(stderr,
|
||||||
|
"WARNING: CPU supports %" PRIx64 " but software requires %" PRIx64
|
||||||
|
"\n",
|
||||||
|
bits, static_cast<int64_t>(HWY_ENABLED_BASELINE));
|
||||||
|
}
|
||||||
|
#else // HWY_ARCH_ARM && HWY_HAVE_RUNTIME_DISPATCH
|
||||||
|
// TODO(janwas): detect for other platforms and check for baseline
|
||||||
|
// This file is typically compiled without HWY_IS_TEST, but targets_test has
|
||||||
|
// it set, and will expect all of its HWY_TARGETS (= all attainable) to be
|
||||||
|
// supported.
|
||||||
|
bits |= HWY_ENABLED_BASELINE;
|
||||||
|
#endif // HWY_ARCH_X86
|
||||||
|
|
||||||
|
return bits;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
HWY_DLLEXPORT HWY_NORETURN void HWY_FORMAT(3, 4)
|
||||||
|
Abort(const char* file, int line, const char* format, ...) {
|
||||||
|
char buf[2000];
|
||||||
|
va_list args;
|
||||||
|
va_start(args, format);
|
||||||
|
vsnprintf(buf, sizeof(buf), format, args);
|
||||||
|
va_end(args);
|
||||||
|
|
||||||
|
fprintf(stderr, "Abort at %s:%d: %s\n", file, line, buf);
|
||||||
|
|
||||||
|
// If compiled with any sanitizer, they can also print a stack trace.
|
||||||
|
#if HWY_IS_ASAN || HWY_IS_MSAN || HWY_IS_TSAN
|
||||||
|
__sanitizer_print_stack_trace();
|
||||||
|
#endif // HWY_IS_*
|
||||||
|
fflush(stderr);
|
||||||
|
|
||||||
|
// Now terminate the program:
|
||||||
|
#if HWY_ARCH_RVV
|
||||||
|
exit(1); // trap/abort just freeze Spike.
|
||||||
|
#elif HWY_IS_DEBUG_BUILD && !HWY_COMPILER_MSVC
|
||||||
|
// Facilitates breaking into a debugger, but don't use this in non-debug
|
||||||
|
// builds because it looks like "illegal instruction", which is misleading.
|
||||||
|
__builtin_trap();
|
||||||
|
#else
|
||||||
|
abort(); // Compile error without this due to HWY_NORETURN.
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void DisableTargets(int64_t disabled_targets) {
|
||||||
|
supported_mask_ = static_cast<int64_t>(~disabled_targets);
|
||||||
|
// This will take effect on the next call to SupportedTargets, which is
|
||||||
|
// called right before GetChosenTarget::Update. However, calling Update here
|
||||||
|
// would make it appear that HWY_DYNAMIC_DISPATCH was called, which we want
|
||||||
|
// to check in tests. We instead de-initialize such that the next
|
||||||
|
// HWY_DYNAMIC_DISPATCH calls GetChosenTarget::Update via FunctionCache.
|
||||||
|
GetChosenTarget().DeInit();
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_DLLEXPORT void SetSupportedTargetsForTest(int64_t targets) {
|
||||||
|
supported_targets_for_test_ = targets;
|
||||||
|
GetChosenTarget().DeInit(); // see comment above
|
||||||
|
}
|
||||||
|
|
||||||
|
HWY_DLLEXPORT int64_t SupportedTargets() {
|
||||||
|
int64_t targets = supported_targets_for_test_;
|
||||||
|
if (HWY_LIKELY(targets == 0)) {
|
||||||
|
// Mock not active. Re-detect instead of caching just in case we're on a
|
||||||
|
// heterogeneous ISA (also requires some app support to pin threads). This
|
||||||
|
// is only reached on the first HWY_DYNAMIC_DISPATCH or after each call to
|
||||||
|
// DisableTargets or SetSupportedTargetsForTest.
|
||||||
|
targets = DetectTargets();
|
||||||
|
|
||||||
|
// VectorBytes invokes HWY_DYNAMIC_DISPATCH. To prevent infinite recursion,
|
||||||
|
// first set up ChosenTarget. No need to Update() again afterwards with the
|
||||||
|
// final targets - that will be done by a caller of this function.
|
||||||
|
GetChosenTarget().Update(targets);
|
||||||
|
|
||||||
|
// Now that we can call VectorBytes, check for targets with specific sizes.
|
||||||
|
if (HWY_ARCH_ARM_A64) {
|
||||||
|
const size_t vec_bytes = VectorBytes(); // uncached, see declaration
|
||||||
|
if ((targets & HWY_SVE) && vec_bytes == 32) {
|
||||||
|
targets = static_cast<int64_t>(targets | HWY_SVE_256);
|
||||||
|
} else {
|
||||||
|
targets = static_cast<int64_t>(targets & ~HWY_SVE_256);
|
||||||
|
}
|
||||||
|
if ((targets & HWY_SVE2) && vec_bytes == 16) {
|
||||||
|
targets = static_cast<int64_t>(targets | HWY_SVE2_128);
|
||||||
|
} else {
|
||||||
|
targets = static_cast<int64_t>(targets & ~HWY_SVE2_128);
|
||||||
|
}
|
||||||
|
} // HWY_ARCH_ARM_A64
|
||||||
|
}
|
||||||
|
|
||||||
|
targets &= supported_mask_;
|
||||||
|
return targets == 0 ? HWY_STATIC_TARGET : targets;
|
||||||
}
|
}
|
||||||
|
|
||||||
HWY_DLLEXPORT ChosenTarget& GetChosenTarget() {
|
HWY_DLLEXPORT ChosenTarget& GetChosenTarget() {
|
||||||
|
@ -352,14 +428,4 @@ HWY_DLLEXPORT ChosenTarget& GetChosenTarget() {
|
||||||
return chosen_target;
|
return chosen_target;
|
||||||
}
|
}
|
||||||
|
|
||||||
void ChosenTarget::Update() {
|
|
||||||
// The supported variable contains the current CPU supported targets shifted
|
|
||||||
// to the location expected by the ChosenTarget mask. We enabled SCALAR
|
|
||||||
// regardless of whether it was compiled since it is also used as the
|
|
||||||
// fallback mechanism to the baseline target.
|
|
||||||
uint32_t supported = HWY_CHOSEN_TARGET_SHIFT(hwy::SupportedTargets()) |
|
|
||||||
HWY_CHOSEN_TARGET_MASK_SCALAR;
|
|
||||||
mask_.store(supported);
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -24,13 +25,18 @@
|
||||||
#include "hwy/detect_targets.h"
|
#include "hwy/detect_targets.h"
|
||||||
#include "hwy/highway_export.h"
|
#include "hwy/highway_export.h"
|
||||||
|
|
||||||
|
#if !HWY_ARCH_RVV
|
||||||
|
#include <atomic>
|
||||||
|
#endif
|
||||||
|
|
||||||
namespace hwy {
|
namespace hwy {
|
||||||
|
|
||||||
// Returns (cached) bitfield of enabled targets that are supported on this CPU.
|
// Returns bitfield of enabled targets that are supported on this CPU; there is
|
||||||
// Implemented in targets.cc; unconditionally compiled to support the use case
|
// always at least one such target, hence the return value is never 0. The
|
||||||
// of binary-only distributions. The HWY_SUPPORTED_TARGETS wrapper may allow
|
// targets returned may change after calling DisableTargets. This function is
|
||||||
// eliding calls to this function.
|
// always defined, but the HWY_SUPPORTED_TARGETS wrapper may allow eliding
|
||||||
HWY_DLLEXPORT uint32_t SupportedTargets();
|
// calls to it if there is only a single target enabled.
|
||||||
|
HWY_DLLEXPORT int64_t SupportedTargets();
|
||||||
|
|
||||||
// Evaluates to a function call, or literal if there is a single target.
|
// Evaluates to a function call, or literal if there is a single target.
|
||||||
#if (HWY_TARGETS & (HWY_TARGETS - 1)) == 0
|
#if (HWY_TARGETS & (HWY_TARGETS - 1)) == 0
|
||||||
|
@ -39,40 +45,36 @@ HWY_DLLEXPORT uint32_t SupportedTargets();
|
||||||
#define HWY_SUPPORTED_TARGETS hwy::SupportedTargets()
|
#define HWY_SUPPORTED_TARGETS hwy::SupportedTargets()
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
// Disable from runtime dispatch the mask of compiled in targets. Targets that
|
// Subsequent SupportedTargets will not return targets whose bit(s) are set in
|
||||||
// were not enabled at compile time are ignored. This function is useful to
|
// `disabled_targets`. Exception: if SupportedTargets would return 0, it will
|
||||||
// disable a target supported by the CPU that is known to have bugs or when a
|
// instead return HWY_STATIC_TARGET (there must always be one target to call).
|
||||||
// lower target is desired. For this reason, attempts to disable targets which
|
//
|
||||||
// are in HWY_ENABLED_BASELINE have no effect so SupportedTargets() always
|
// This function is useful for disabling targets known to be buggy, or if the
|
||||||
// returns at least the baseline target.
|
// best available target is undesirable (perhaps due to throttling or memory
|
||||||
HWY_DLLEXPORT void DisableTargets(uint32_t disabled_targets);
|
// bandwidth limitations). Use SetSupportedTargetsForTest instead of this
|
||||||
|
// function for iteratively enabling specific targets for testing.
|
||||||
|
HWY_DLLEXPORT void DisableTargets(int64_t disabled_targets);
|
||||||
|
|
||||||
// Set the mock mask of CPU supported targets instead of the actual CPU
|
// Subsequent SupportedTargets will return the given set of targets, except
|
||||||
// supported targets computed in SupportedTargets(). The return value of
|
// those disabled via DisableTargets. Call with a mask of 0 to disable the mock
|
||||||
// SupportedTargets() will still be affected by the DisableTargets() mask
|
// and return to the normal SupportedTargets behavior. Used to run tests for
|
||||||
// regardless of this mock, to prevent accidentally adding targets that are
|
// all targets.
|
||||||
// known to be buggy in the current CPU. Call with a mask of 0 to disable the
|
HWY_DLLEXPORT void SetSupportedTargetsForTest(int64_t targets);
|
||||||
// mock and use the actual CPU supported targets instead.
|
|
||||||
HWY_DLLEXPORT void SetSupportedTargetsForTest(uint32_t targets);
|
|
||||||
|
|
||||||
// Returns whether the SupportedTargets() function was called since the last
|
|
||||||
// SetSupportedTargetsForTest() call.
|
|
||||||
HWY_DLLEXPORT bool SupportedTargetsCalledForTest();
|
|
||||||
|
|
||||||
// Return the list of targets in HWY_TARGETS supported by the CPU as a list of
|
// Return the list of targets in HWY_TARGETS supported by the CPU as a list of
|
||||||
// individual HWY_* target macros such as HWY_SCALAR or HWY_NEON. This list
|
// individual HWY_* target macros such as HWY_SCALAR or HWY_NEON. This list
|
||||||
// is affected by the current SetSupportedTargetsForTest() mock if any.
|
// is affected by the current SetSupportedTargetsForTest() mock if any.
|
||||||
HWY_INLINE std::vector<uint32_t> SupportedAndGeneratedTargets() {
|
HWY_INLINE std::vector<int64_t> SupportedAndGeneratedTargets() {
|
||||||
std::vector<uint32_t> ret;
|
std::vector<int64_t> ret;
|
||||||
for (uint32_t targets = SupportedTargets() & HWY_TARGETS; targets != 0;
|
for (int64_t targets = SupportedTargets() & HWY_TARGETS; targets != 0;
|
||||||
targets = targets & (targets - 1)) {
|
targets = targets & (targets - 1)) {
|
||||||
uint32_t current_target = targets & ~(targets - 1);
|
int64_t current_target = targets & ~(targets - 1);
|
||||||
ret.push_back(current_target);
|
ret.push_back(current_target);
|
||||||
}
|
}
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
static inline HWY_MAYBE_UNUSED const char* TargetName(int64_t target) {
|
||||||
switch (target) {
|
switch (target) {
|
||||||
#if HWY_ARCH_X86
|
#if HWY_ARCH_X86
|
||||||
case HWY_SSSE3:
|
case HWY_SSSE3:
|
||||||
|
@ -88,22 +90,28 @@ static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_ARCH_ARM
|
#if HWY_ARCH_ARM
|
||||||
|
case HWY_SVE2_128:
|
||||||
|
return "SVE2_128";
|
||||||
|
case HWY_SVE_256:
|
||||||
|
return "SVE_256";
|
||||||
case HWY_SVE2:
|
case HWY_SVE2:
|
||||||
return "SVE2";
|
return "SVE2";
|
||||||
case HWY_SVE:
|
case HWY_SVE:
|
||||||
return "SVE";
|
return "SVE";
|
||||||
case HWY_NEON:
|
case HWY_NEON:
|
||||||
return "Neon";
|
return "NEON";
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_ARCH_PPC
|
#if HWY_ARCH_PPC
|
||||||
case HWY_PPC8:
|
case HWY_PPC8:
|
||||||
return "Power8";
|
return "PPC8";
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_ARCH_WASM
|
#if HWY_ARCH_WASM
|
||||||
case HWY_WASM:
|
case HWY_WASM:
|
||||||
return "Wasm";
|
return "WASM";
|
||||||
|
case HWY_WASM_EMU256:
|
||||||
|
return "WASM_EMU256";
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if HWY_ARCH_RVV
|
#if HWY_ARCH_RVV
|
||||||
|
@ -111,8 +119,10 @@ static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
||||||
return "RVV";
|
return "RVV";
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
case HWY_EMU128:
|
||||||
|
return "EMU128";
|
||||||
case HWY_SCALAR:
|
case HWY_SCALAR:
|
||||||
return "Scalar";
|
return "SCALAR";
|
||||||
|
|
||||||
default:
|
default:
|
||||||
return "Unknown"; // must satisfy gtest IsValidParamName()
|
return "Unknown"; // must satisfy gtest IsValidParamName()
|
||||||
|
@ -125,37 +135,37 @@ static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
||||||
// For the ChosenTarget mask and index we use a different bit arrangement than
|
// For the ChosenTarget mask and index we use a different bit arrangement than
|
||||||
// in the HWY_TARGETS mask. Only the targets involved in the current
|
// in the HWY_TARGETS mask. Only the targets involved in the current
|
||||||
// architecture are used in this mask, and therefore only the least significant
|
// architecture are used in this mask, and therefore only the least significant
|
||||||
// (HWY_MAX_DYNAMIC_TARGETS + 2) bits of the uint32_t mask are used. The least
|
// (HWY_MAX_DYNAMIC_TARGETS + 2) bits of the int64_t mask are used. The least
|
||||||
// significant bit is set when the mask is not initialized, the next
|
// significant bit is set when the mask is not initialized, the next
|
||||||
// HWY_MAX_DYNAMIC_TARGETS more significant bits are a range of bits from the
|
// HWY_MAX_DYNAMIC_TARGETS more significant bits are a range of bits from the
|
||||||
// HWY_TARGETS or SupportedTargets() mask for the given architecture shifted to
|
// HWY_TARGETS or SupportedTargets() mask for the given architecture shifted to
|
||||||
// that position and the next more significant bit is used for the scalar
|
// that position and the next more significant bit is used for HWY_SCALAR (if
|
||||||
// target. Because of this we need to define equivalent values for HWY_TARGETS
|
// HWY_COMPILE_ONLY_SCALAR is defined) or HWY_EMU128. Because of this we need to
|
||||||
// in this representation.
|
// define equivalent values for HWY_TARGETS in this representation.
|
||||||
// This mask representation allows to use ctz() on this mask and obtain a small
|
// This mask representation allows to use ctz() on this mask and obtain a small
|
||||||
// number that's used as an index of the table for dynamic dispatch. In this
|
// number that's used as an index of the table for dynamic dispatch. In this
|
||||||
// way the first entry is used when the mask is uninitialized, the following
|
// way the first entry is used when the mask is uninitialized, the following
|
||||||
// HWY_MAX_DYNAMIC_TARGETS are for dynamic dispatch and the last one is for
|
// HWY_MAX_DYNAMIC_TARGETS are for dynamic dispatch and the last one is for
|
||||||
// scalar.
|
// scalar.
|
||||||
|
|
||||||
// The HWY_SCALAR bit in the ChosenTarget mask format.
|
// The HWY_SCALAR/HWY_EMU128 bit in the ChosenTarget mask format.
|
||||||
#define HWY_CHOSEN_TARGET_MASK_SCALAR (1u << (HWY_MAX_DYNAMIC_TARGETS + 1))
|
#define HWY_CHOSEN_TARGET_MASK_SCALAR (1LL << (HWY_MAX_DYNAMIC_TARGETS + 1))
|
||||||
|
|
||||||
// Converts from a HWY_TARGETS mask to a ChosenTarget mask format for the
|
// Converts from a HWY_TARGETS mask to a ChosenTarget mask format for the
|
||||||
// current architecture.
|
// current architecture.
|
||||||
#define HWY_CHOSEN_TARGET_SHIFT(X) \
|
#define HWY_CHOSEN_TARGET_SHIFT(X) \
|
||||||
((((X) >> (HWY_HIGHEST_TARGET_BIT + 1 - HWY_MAX_DYNAMIC_TARGETS)) & \
|
((((X) >> (HWY_HIGHEST_TARGET_BIT + 1 - HWY_MAX_DYNAMIC_TARGETS)) & \
|
||||||
((1u << HWY_MAX_DYNAMIC_TARGETS) - 1)) \
|
((1LL << HWY_MAX_DYNAMIC_TARGETS) - 1)) \
|
||||||
<< 1)
|
<< 1)
|
||||||
|
|
||||||
// The HWY_TARGETS mask in the ChosenTarget mask format.
|
// The HWY_TARGETS mask in the ChosenTarget mask format.
|
||||||
#define HWY_CHOSEN_TARGET_MASK_TARGETS \
|
#define HWY_CHOSEN_TARGET_MASK_TARGETS \
|
||||||
(HWY_CHOSEN_TARGET_SHIFT(HWY_TARGETS) | HWY_CHOSEN_TARGET_MASK_SCALAR | 1u)
|
(HWY_CHOSEN_TARGET_SHIFT(HWY_TARGETS) | HWY_CHOSEN_TARGET_MASK_SCALAR | 1LL)
|
||||||
|
|
||||||
#if HWY_ARCH_X86
|
#if HWY_ARCH_X86
|
||||||
// Maximum number of dynamic targets, changing this value is an ABI incompatible
|
// Maximum number of dynamic targets, changing this value is an ABI incompatible
|
||||||
// change
|
// change
|
||||||
#define HWY_MAX_DYNAMIC_TARGETS 10
|
#define HWY_MAX_DYNAMIC_TARGETS 15
|
||||||
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_X86
|
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_X86
|
||||||
// These must match the order in which the HWY_TARGETS are defined
|
// These must match the order in which the HWY_TARGETS are defined
|
||||||
// starting by the least significant (HWY_HIGHEST_TARGET_BIT + 1 -
|
// starting by the least significant (HWY_HIGHEST_TARGET_BIT + 1 -
|
||||||
|
@ -163,6 +173,11 @@ static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
||||||
// HWY_MAX_DYNAMIC_TARGETS elements and does not include SCALAR. The first entry
|
// HWY_MAX_DYNAMIC_TARGETS elements and does not include SCALAR. The first entry
|
||||||
// corresponds to the best target. Don't include a "," at the end of the list.
|
// corresponds to the best target. Don't include a "," at the end of the list.
|
||||||
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
nullptr, /* reserved */ \
|
nullptr, /* reserved */ \
|
||||||
nullptr, /* reserved */ \
|
nullptr, /* reserved */ \
|
||||||
HWY_CHOOSE_AVX3_DL(func_name), /* AVX3_DL */ \
|
HWY_CHOOSE_AVX3_DL(func_name), /* AVX3_DL */ \
|
||||||
|
@ -171,49 +186,74 @@ static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
||||||
nullptr, /* AVX */ \
|
nullptr, /* AVX */ \
|
||||||
HWY_CHOOSE_SSE4(func_name), /* SSE4 */ \
|
HWY_CHOOSE_SSE4(func_name), /* SSE4 */ \
|
||||||
HWY_CHOOSE_SSSE3(func_name), /* SSSE3 */ \
|
HWY_CHOOSE_SSSE3(func_name), /* SSSE3 */ \
|
||||||
nullptr, /* SSE3 */ \
|
nullptr , /* reserved - SSE3? */ \
|
||||||
nullptr /* SSE2 */
|
nullptr /* reserved - SSE2? */
|
||||||
|
|
||||||
#elif HWY_ARCH_ARM
|
#elif HWY_ARCH_ARM
|
||||||
// See HWY_ARCH_X86 above for details.
|
// See HWY_ARCH_X86 above for details.
|
||||||
#define HWY_MAX_DYNAMIC_TARGETS 4
|
#define HWY_MAX_DYNAMIC_TARGETS 15
|
||||||
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_ARM
|
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_ARM
|
||||||
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
HWY_CHOOSE_SVE2_128(func_name), /* SVE2 128-bit */ \
|
||||||
|
HWY_CHOOSE_SVE_256(func_name), /* SVE 256-bit */ \
|
||||||
HWY_CHOOSE_SVE2(func_name), /* SVE2 */ \
|
HWY_CHOOSE_SVE2(func_name), /* SVE2 */ \
|
||||||
HWY_CHOOSE_SVE(func_name), /* SVE */ \
|
HWY_CHOOSE_SVE(func_name), /* SVE */ \
|
||||||
nullptr, /* reserved */ \
|
HWY_CHOOSE_NEON(func_name), /* NEON */ \
|
||||||
HWY_CHOOSE_NEON(func_name) /* NEON */
|
nullptr /* reserved - Helium? */
|
||||||
|
|
||||||
#elif HWY_ARCH_PPC
|
|
||||||
// See HWY_ARCH_X86 above for details.
|
|
||||||
#define HWY_MAX_DYNAMIC_TARGETS 5
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_PPC
|
|
||||||
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
|
||||||
nullptr, /* reserved */ \
|
|
||||||
nullptr, /* reserved */ \
|
|
||||||
HWY_CHOOSE_PPC8(func_name), /* PPC8 */ \
|
|
||||||
nullptr, /* VSX */ \
|
|
||||||
nullptr /* AltiVec */
|
|
||||||
|
|
||||||
#elif HWY_ARCH_WASM
|
|
||||||
// See HWY_ARCH_X86 above for details.
|
|
||||||
#define HWY_MAX_DYNAMIC_TARGETS 4
|
|
||||||
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_WASM
|
|
||||||
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
|
||||||
nullptr, /* reserved */ \
|
|
||||||
nullptr, /* reserved */ \
|
|
||||||
HWY_CHOOSE_WASM2(func_name), /* WASM2 */ \
|
|
||||||
HWY_CHOOSE_WASM(func_name) /* WASM */
|
|
||||||
|
|
||||||
#elif HWY_ARCH_RVV
|
#elif HWY_ARCH_RVV
|
||||||
// See HWY_ARCH_X86 above for details.
|
// See HWY_ARCH_X86 above for details.
|
||||||
#define HWY_MAX_DYNAMIC_TARGETS 4
|
#define HWY_MAX_DYNAMIC_TARGETS 9
|
||||||
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_RVV
|
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_RVV
|
||||||
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
||||||
nullptr, /* reserved */ \
|
nullptr, /* reserved */ \
|
||||||
nullptr, /* reserved */ \
|
nullptr, /* reserved */ \
|
||||||
nullptr, /* reserved */ \
|
nullptr, /* reserved */ \
|
||||||
HWY_CHOOSE_RVV(func_name) /* RVV */
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
HWY_CHOOSE_RVV(func_name), /* RVV */ \
|
||||||
|
nullptr /* reserved */
|
||||||
|
|
||||||
|
#elif HWY_ARCH_PPC
|
||||||
|
// See HWY_ARCH_X86 above for details.
|
||||||
|
#define HWY_MAX_DYNAMIC_TARGETS 9
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_PPC
|
||||||
|
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
HWY_CHOOSE_PPC8(func_name), /* PPC8 */ \
|
||||||
|
nullptr, /* reserved (VSX or AltiVec) */ \
|
||||||
|
nullptr /* reserved (VSX or AltiVec) */
|
||||||
|
|
||||||
|
#elif HWY_ARCH_WASM
|
||||||
|
// See HWY_ARCH_X86 above for details.
|
||||||
|
#define HWY_MAX_DYNAMIC_TARGETS 9
|
||||||
|
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_WASM
|
||||||
|
#define HWY_CHOOSE_TARGET_LIST(func_name) \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
nullptr, /* reserved */ \
|
||||||
|
HWY_CHOOSE_WASM_EMU256(func_name), /* WASM_EMU256 */ \
|
||||||
|
HWY_CHOOSE_WASM(func_name), /* WASM */ \
|
||||||
|
nullptr /* reserved */
|
||||||
|
|
||||||
#else
|
#else
|
||||||
// Unknown architecture, will use HWY_SCALAR without dynamic dispatch, though
|
// Unknown architecture, will use HWY_SCALAR without dynamic dispatch, though
|
||||||
|
@ -222,32 +262,52 @@ static inline HWY_MAYBE_UNUSED const char* TargetName(uint32_t target) {
|
||||||
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_SCALAR
|
#define HWY_HIGHEST_TARGET_BIT HWY_HIGHEST_TARGET_BIT_SCALAR
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Bitfield of supported and enabled targets. The format differs from that of
|
||||||
|
// HWY_TARGETS; the lowest bit governs the first function pointer (which is
|
||||||
|
// special in that it calls FunctionCache, then Update, then dispatches to the
|
||||||
|
// actual implementation) in the tables created by HWY_EXPORT. Monostate (see
|
||||||
|
// GetChosenTarget), thread-safe except on RVV.
|
||||||
struct ChosenTarget {
|
struct ChosenTarget {
|
||||||
public:
|
public:
|
||||||
// Update the ChosenTarget mask based on the current CPU supported
|
// Reset bits according to `targets` (typically the return value of
|
||||||
// targets.
|
// SupportedTargets()). Postcondition: IsInitialized() == true.
|
||||||
HWY_DLLEXPORT void Update();
|
void Update(int64_t targets) {
|
||||||
|
// These are `targets` shifted downwards, see above. Also include SCALAR
|
||||||
|
// (corresponds to the last entry in the function table) as fallback.
|
||||||
|
StoreMask(HWY_CHOSEN_TARGET_SHIFT(targets) | HWY_CHOSEN_TARGET_MASK_SCALAR);
|
||||||
|
}
|
||||||
|
|
||||||
// Reset the ChosenTarget to the uninitialized state.
|
// Reset to the uninitialized state, so that FunctionCache will call Update
|
||||||
void DeInit() { mask_.store(1); }
|
// during the next HWY_DYNAMIC_DISPATCH, and IsInitialized returns false.
|
||||||
|
void DeInit() { StoreMask(1); }
|
||||||
|
|
||||||
// Whether the ChosenTarget was initialized. This is useful to know whether
|
// Whether Update was called. This indicates whether any HWY_DYNAMIC_DISPATCH
|
||||||
// any HWY_DYNAMIC_DISPATCH function was called.
|
// function was called, which we check in tests.
|
||||||
bool IsInitialized() const { return mask_.load() != 1; }
|
bool IsInitialized() const { return LoadMask() != 1; }
|
||||||
|
|
||||||
// Return the index in the dynamic dispatch table to be used by the current
|
// Return the index in the dynamic dispatch table to be used by the current
|
||||||
// CPU. Note that this method must be in the header file so it uses the value
|
// CPU. Note that this method must be in the header file so it uses the value
|
||||||
// of HWY_CHOSEN_TARGET_MASK_TARGETS defined in the translation unit that
|
// of HWY_CHOSEN_TARGET_MASK_TARGETS defined in the translation unit that
|
||||||
// calls it, which may be different from others. This allows to only consider
|
// calls it, which may be different from others. This means we only enable
|
||||||
// those targets that were actually compiled in this module.
|
// those targets that were actually compiled in this module.
|
||||||
size_t HWY_INLINE GetIndex() const {
|
size_t HWY_INLINE GetIndex() const {
|
||||||
return hwy::Num0BitsBelowLS1Bit_Nonzero32(mask_.load() &
|
return hwy::Num0BitsBelowLS1Bit_Nonzero64(
|
||||||
HWY_CHOSEN_TARGET_MASK_TARGETS);
|
static_cast<uint64_t>(LoadMask() & HWY_CHOSEN_TARGET_MASK_TARGETS));
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// Initialized to 1 so GetIndex() returns 0.
|
// TODO(janwas): remove #if once <atomic> is available
|
||||||
std::atomic<uint32_t> mask_{1};
|
#if HWY_ARCH_RVV
|
||||||
|
int64_t LoadMask() const { return mask_; }
|
||||||
|
void StoreMask(int64_t mask) { mask_ = mask; }
|
||||||
|
|
||||||
|
int64_t mask_{1}; // Initialized to 1 so GetIndex() returns 0.
|
||||||
|
#else
|
||||||
|
int64_t LoadMask() const { return mask_.load(); }
|
||||||
|
void StoreMask(int64_t mask) { mask_.store(mask); }
|
||||||
|
|
||||||
|
std::atomic<int64_t> mask_{1}; // Initialized to 1 so GetIndex() returns 0.
|
||||||
|
#endif // HWY_ARCH_RVV
|
||||||
};
|
};
|
||||||
|
|
||||||
// For internal use (e.g. by FunctionCache and DisableTargets).
|
// For internal use (e.g. by FunctionCache and DisableTargets).
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2020 Google LLC
|
// Copyright 2020 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -20,7 +21,8 @@ namespace fake {
|
||||||
|
|
||||||
#define DECLARE_FUNCTION(TGT) \
|
#define DECLARE_FUNCTION(TGT) \
|
||||||
namespace N_##TGT { \
|
namespace N_##TGT { \
|
||||||
uint32_t FakeFunction(int) { return HWY_##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_DL)
|
||||||
|
@ -31,41 +33,62 @@ DECLARE_FUNCTION(SSSE3)
|
||||||
DECLARE_FUNCTION(NEON)
|
DECLARE_FUNCTION(NEON)
|
||||||
DECLARE_FUNCTION(SVE)
|
DECLARE_FUNCTION(SVE)
|
||||||
DECLARE_FUNCTION(SVE2)
|
DECLARE_FUNCTION(SVE2)
|
||||||
|
DECLARE_FUNCTION(SVE_256)
|
||||||
|
DECLARE_FUNCTION(SVE2_128)
|
||||||
DECLARE_FUNCTION(PPC8)
|
DECLARE_FUNCTION(PPC8)
|
||||||
DECLARE_FUNCTION(WASM)
|
DECLARE_FUNCTION(WASM)
|
||||||
DECLARE_FUNCTION(RVV)
|
DECLARE_FUNCTION(RVV)
|
||||||
DECLARE_FUNCTION(SCALAR)
|
DECLARE_FUNCTION(SCALAR)
|
||||||
|
DECLARE_FUNCTION(EMU128)
|
||||||
|
|
||||||
HWY_EXPORT(FakeFunction);
|
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() {
|
void CheckFakeFunction() {
|
||||||
#define CHECK_ARRAY_ENTRY(TGT) \
|
// When adding a target, also add to DECLARE_FUNCTION above.
|
||||||
if ((HWY_TARGETS & HWY_##TGT) != 0) { \
|
CallFunctionForTarget(HWY_AVX3_DL, __LINE__);
|
||||||
hwy::SetSupportedTargetsForTest(HWY_##TGT); \
|
CallFunctionForTarget(HWY_AVX3, __LINE__);
|
||||||
/* Calling Update() first to make &HWY_DYNAMIC_DISPATCH() return */ \
|
CallFunctionForTarget(HWY_AVX2, __LINE__);
|
||||||
/* the pointer to the already cached function. */ \
|
CallFunctionForTarget(HWY_SSE4, __LINE__);
|
||||||
hwy::GetChosenTarget().Update(); \
|
CallFunctionForTarget(HWY_SSSE3, __LINE__);
|
||||||
EXPECT_EQ(uint32_t(HWY_##TGT), HWY_DYNAMIC_DISPATCH(FakeFunction)(42)); \
|
CallFunctionForTarget(HWY_NEON, __LINE__);
|
||||||
/* Calling DeInit() will test that the initializer function */ \
|
CallFunctionForTarget(HWY_SVE, __LINE__);
|
||||||
/* also calls the right function. */ \
|
CallFunctionForTarget(HWY_SVE2, __LINE__);
|
||||||
hwy::GetChosenTarget().DeInit(); \
|
CallFunctionForTarget(HWY_SVE_256, __LINE__);
|
||||||
EXPECT_EQ(uint32_t(HWY_##TGT), HWY_DYNAMIC_DISPATCH(FakeFunction)(42)); \
|
CallFunctionForTarget(HWY_SVE2_128, __LINE__);
|
||||||
/* Second call uses the cached value from the previous call. */ \
|
CallFunctionForTarget(HWY_PPC8, __LINE__);
|
||||||
EXPECT_EQ(uint32_t(HWY_##TGT), HWY_DYNAMIC_DISPATCH(FakeFunction)(42)); \
|
CallFunctionForTarget(HWY_WASM, __LINE__);
|
||||||
}
|
CallFunctionForTarget(HWY_RVV, __LINE__);
|
||||||
CHECK_ARRAY_ENTRY(AVX3_DL)
|
// The tables only have space for either HWY_SCALAR or HWY_EMU128; the former
|
||||||
CHECK_ARRAY_ENTRY(AVX3)
|
// is opt-in only.
|
||||||
CHECK_ARRAY_ENTRY(AVX2)
|
#if defined(HWY_COMPILE_ONLY_SCALAR) || HWY_BROKEN_EMU128
|
||||||
CHECK_ARRAY_ENTRY(SSE4)
|
CallFunctionForTarget(HWY_SCALAR, __LINE__);
|
||||||
CHECK_ARRAY_ENTRY(SSSE3)
|
#else
|
||||||
CHECK_ARRAY_ENTRY(NEON)
|
CallFunctionForTarget(HWY_EMU128, __LINE__);
|
||||||
CHECK_ARRAY_ENTRY(SVE)
|
#endif
|
||||||
CHECK_ARRAY_ENTRY(SVE2)
|
|
||||||
CHECK_ARRAY_ENTRY(PPC8)
|
|
||||||
CHECK_ARRAY_ENTRY(WASM)
|
|
||||||
CHECK_ARRAY_ENTRY(RVV)
|
|
||||||
CHECK_ARRAY_ENTRY(SCALAR)
|
|
||||||
#undef CHECK_ARRAY_ENTRY
|
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace fake
|
} // namespace fake
|
||||||
|
@ -86,31 +109,27 @@ class HwyTargetsTest : public testing::Test {
|
||||||
TEST_F(HwyTargetsTest, ChosenTargetOrderTest) { fake::CheckFakeFunction(); }
|
TEST_F(HwyTargetsTest, ChosenTargetOrderTest) { fake::CheckFakeFunction(); }
|
||||||
|
|
||||||
TEST_F(HwyTargetsTest, DisabledTargetsTest) {
|
TEST_F(HwyTargetsTest, DisabledTargetsTest) {
|
||||||
DisableTargets(~0u);
|
DisableTargets(~0LL);
|
||||||
// Check that the baseline can't be disabled.
|
// Check that disabling everything at least leaves the static target.
|
||||||
HWY_ASSERT(HWY_ENABLED_BASELINE == SupportedTargets());
|
HWY_ASSERT(HWY_STATIC_TARGET == SupportedTargets());
|
||||||
|
|
||||||
DisableTargets(0); // Reset the mask.
|
DisableTargets(0); // Reset the mask.
|
||||||
uint32_t current_targets = SupportedTargets();
|
const int64_t current_targets = SupportedTargets();
|
||||||
if ((current_targets & ~uint32_t(HWY_ENABLED_BASELINE)) == 0) {
|
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.
|
// We can't test anything else if the only compiled target is the baseline.
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get the lowest bit in the mask (the best target) and disable that one.
|
// Get the lowest bit in the mask (the best target) and disable that one.
|
||||||
uint32_t lowest_target = current_targets & (~current_targets + 1);
|
const int64_t best_target = current_targets & (~current_targets + 1);
|
||||||
// The lowest target shouldn't be one in the baseline.
|
DisableTargets(best_target);
|
||||||
HWY_ASSERT((lowest_target & ~uint32_t(HWY_ENABLED_BASELINE)) != 0);
|
|
||||||
DisableTargets(lowest_target);
|
|
||||||
|
|
||||||
// Check that the other targets are still enabled.
|
// Check that the other targets are still enabled.
|
||||||
HWY_ASSERT((lowest_target ^ current_targets) == SupportedTargets());
|
HWY_ASSERT((best_target ^ current_targets) == SupportedTargets());
|
||||||
DisableTargets(0); // Reset the mask.
|
DisableTargets(0); // Reset the mask.
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char **argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
// Copyright 2019 Google LLC
|
// Copyright 2019 Google LLC
|
||||||
|
// SPDX-License-Identifier: Apache-2.0
|
||||||
//
|
//
|
||||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
// you may not use this file except in compliance with the License.
|
// you may not use this file except in compliance with the License.
|
||||||
|
@ -16,12 +17,9 @@
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
|
||||||
#include <algorithm>
|
|
||||||
#include <limits>
|
|
||||||
|
|
||||||
#undef HWY_TARGET_INCLUDE
|
#undef HWY_TARGET_INCLUDE
|
||||||
#define HWY_TARGET_INCLUDE "tests/arithmetic_test.cc"
|
#define HWY_TARGET_INCLUDE "tests/arithmetic_test.cc"
|
||||||
#include "hwy/foreach_target.h"
|
#include "hwy/foreach_target.h" // IWYU pragma: keep
|
||||||
#include "hwy/highway.h"
|
#include "hwy/highway.h"
|
||||||
#include "hwy/tests/test_util-inl.h"
|
#include "hwy/tests/test_util-inl.h"
|
||||||
|
|
||||||
|
@ -177,6 +175,23 @@ HWY_NOINLINE void TestAllAbs() {
|
||||||
ForFloatTypes(ForPartialVectors<TestFloatAbs>());
|
ForFloatTypes(ForPartialVectors<TestFloatAbs>());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct TestNeg {
|
||||||
|
template <typename T, class D>
|
||||||
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
|
const auto v0 = Zero(d);
|
||||||
|
const auto vn = Set(d, T(-3));
|
||||||
|
const auto vp = Set(d, T(3));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v0, Neg(v0));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, vp, Neg(vn));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, vn, Neg(vp));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
HWY_NOINLINE void TestAllNeg() {
|
||||||
|
ForSignedTypes(ForPartialVectors<TestNeg>());
|
||||||
|
ForFloatTypes(ForPartialVectors<TestNeg>());
|
||||||
|
}
|
||||||
|
|
||||||
struct TestUnsignedMinMax {
|
struct TestUnsignedMinMax {
|
||||||
template <typename T, class D>
|
template <typename T, class D>
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
|
@ -263,16 +278,15 @@ HWY_NOINLINE void TestAllMinMax() {
|
||||||
ForFloatTypes(ForPartialVectors<TestFloatMinMax>());
|
ForFloatTypes(ForPartialVectors<TestFloatMinMax>());
|
||||||
}
|
}
|
||||||
|
|
||||||
class TestMinMax128 {
|
template <class D>
|
||||||
template <class D>
|
static HWY_NOINLINE Vec<D> Make128(D d, uint64_t hi, uint64_t lo) {
|
||||||
static HWY_NOINLINE Vec<D> Make128(D d, uint64_t hi, uint64_t lo) {
|
|
||||||
alignas(16) uint64_t in[2];
|
alignas(16) uint64_t in[2];
|
||||||
in[0] = lo;
|
in[0] = lo;
|
||||||
in[1] = hi;
|
in[1] = hi;
|
||||||
return LoadDup128(d, in);
|
return LoadDup128(d, in);
|
||||||
}
|
}
|
||||||
|
|
||||||
public:
|
struct TestMinMax128 {
|
||||||
template <typename T, class D>
|
template <typename T, class D>
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
using V = Vec<D>;
|
using V = Vec<D>;
|
||||||
|
@ -341,779 +355,73 @@ HWY_NOINLINE void TestAllMinMax128() {
|
||||||
ForGEVectors<128, TestMinMax128>()(uint64_t());
|
ForGEVectors<128, TestMinMax128>()(uint64_t());
|
||||||
}
|
}
|
||||||
|
|
||||||
struct TestUnsignedMul {
|
struct TestMinMax128Upper {
|
||||||
template <typename T, class D>
|
template <typename T, class D>
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
||||||
const auto v0 = Zero(d);
|
using V = Vec<D>;
|
||||||
const auto v1 = Set(d, T(1));
|
|
||||||
const auto vi = Iota(d, 1);
|
|
||||||
const auto vj = Iota(d, 3);
|
|
||||||
const size_t N = Lanes(d);
|
const size_t N = Lanes(d);
|
||||||
auto expected = AllocateAligned<T>(N);
|
auto a_lanes = AllocateAligned<T>(N);
|
||||||
|
auto b_lanes = AllocateAligned<T>(N);
|
||||||
HWY_ASSERT_VEC_EQ(d, v0, Mul(v0, v0));
|
auto min_lanes = AllocateAligned<T>(N);
|
||||||
HWY_ASSERT_VEC_EQ(d, v1, Mul(v1, v1));
|
auto max_lanes = AllocateAligned<T>(N);
|
||||||
HWY_ASSERT_VEC_EQ(d, vi, Mul(v1, vi));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vi, Mul(vi, v1));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((1 + i) * (1 + i));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), Mul(vi, vi));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((1 + i) * (3 + i));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), Mul(vi, vj));
|
|
||||||
|
|
||||||
const T max = LimitsMax<T>();
|
|
||||||
const auto vmax = Set(d, max);
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vmax, Mul(vmax, v1));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vmax, Mul(v1, vmax));
|
|
||||||
|
|
||||||
const size_t bits = sizeof(T) * 8;
|
|
||||||
const uint64_t mask = (1ull << bits) - 1;
|
|
||||||
const T max2 = (uint64_t(max) * max) & mask;
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, max2), Mul(vmax, vmax));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
struct TestSignedMul {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto expected = AllocateAligned<T>(N);
|
|
||||||
|
|
||||||
const auto v0 = Zero(d);
|
|
||||||
const auto v1 = Set(d, T(1));
|
|
||||||
const auto vi = Iota(d, 1);
|
|
||||||
const auto vn = Iota(d, -T(N)); // no i8 supported, so no wraparound
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v0, Mul(v0, v0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v1, Mul(v1, v1));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vi, Mul(v1, vi));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vi, Mul(vi, v1));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((1 + i) * (1 + i));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), Mul(vi, vi));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((-T(N) + T(i)) * T(1u + i));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), Mul(vn, vi));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), Mul(vi, vn));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllMul() {
|
|
||||||
const ForPartialVectors<TestUnsignedMul> test_unsigned;
|
|
||||||
// No u8.
|
|
||||||
test_unsigned(uint16_t());
|
|
||||||
test_unsigned(uint32_t());
|
|
||||||
// No u64.
|
|
||||||
|
|
||||||
const ForPartialVectors<TestSignedMul> test_signed;
|
|
||||||
// No i8.
|
|
||||||
test_signed(int16_t());
|
|
||||||
test_signed(int32_t());
|
|
||||||
// No i64.
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestMulHigh {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
using Wide = MakeWide<T>;
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto in_lanes = AllocateAligned<T>(N);
|
|
||||||
auto expected_lanes = AllocateAligned<T>(N);
|
|
||||||
|
|
||||||
const auto vi = Iota(d, 1);
|
|
||||||
const auto vni = Iota(d, -T(N)); // no i8 supported, so no wraparound
|
|
||||||
|
|
||||||
const auto v0 = Zero(d);
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v0, MulHigh(v0, v0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v0, MulHigh(v0, vi));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v0, MulHigh(vi, v0));
|
|
||||||
|
|
||||||
// Large positive squared
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = T(LimitsMax<T>() >> i);
|
|
||||||
expected_lanes[i] = T((Wide(in_lanes[i]) * in_lanes[i]) >> 16);
|
|
||||||
}
|
|
||||||
auto v = Load(d, in_lanes.get());
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_lanes.get(), MulHigh(v, v));
|
|
||||||
|
|
||||||
// Large positive * small positive
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected_lanes[i] = T((Wide(in_lanes[i]) * T(1u + i)) >> 16);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_lanes.get(), MulHigh(v, vi));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_lanes.get(), MulHigh(vi, v));
|
|
||||||
|
|
||||||
// Large positive * small negative
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected_lanes[i] = T((Wide(in_lanes[i]) * T(i - N)) >> 16);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_lanes.get(), MulHigh(v, vni));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_lanes.get(), MulHigh(vni, v));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllMulHigh() {
|
|
||||||
ForPartialVectors<TestMulHigh> test;
|
|
||||||
test(int16_t());
|
|
||||||
test(uint16_t());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestMulEven {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
using Wide = MakeWide<T>;
|
|
||||||
const Repartition<Wide, D> d2;
|
|
||||||
const auto v0 = Zero(d);
|
|
||||||
HWY_ASSERT_VEC_EQ(d2, Zero(d2), MulEven(v0, v0));
|
|
||||||
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto in_lanes = AllocateAligned<T>(N);
|
|
||||||
auto expected = AllocateAligned<Wide>(Lanes(d2));
|
|
||||||
for (size_t i = 0; i < N; i += 2) {
|
|
||||||
in_lanes[i + 0] = LimitsMax<T>() >> i;
|
|
||||||
if (N != 1) {
|
|
||||||
in_lanes[i + 1] = 1; // unused
|
|
||||||
}
|
|
||||||
expected[i / 2] = Wide(in_lanes[i + 0]) * in_lanes[i + 0];
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto v = Load(d, in_lanes.get());
|
|
||||||
HWY_ASSERT_VEC_EQ(d2, expected.get(), MulEven(v, v));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
struct TestMulEvenOdd64 {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
#if HWY_TARGET != HWY_SCALAR
|
|
||||||
const auto v0 = Zero(d);
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Zero(d), MulEven(v0, v0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Zero(d), MulOdd(v0, v0));
|
|
||||||
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
if (N == 1) return;
|
|
||||||
|
|
||||||
auto in1 = AllocateAligned<T>(N);
|
|
||||||
auto in2 = AllocateAligned<T>(N);
|
|
||||||
auto expected_even = AllocateAligned<T>(N);
|
|
||||||
auto expected_odd = AllocateAligned<T>(N);
|
|
||||||
|
|
||||||
// Random inputs in each lane
|
|
||||||
RandomState rng;
|
RandomState rng;
|
||||||
|
|
||||||
|
const V v00 = Zero(d);
|
||||||
|
const V v01 = Make128(d, 0, 1);
|
||||||
|
const V v10 = Make128(d, 1, 0);
|
||||||
|
const V v11 = Add(v01, v10);
|
||||||
|
|
||||||
|
// Same arg
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v00, Min128Upper(d, v00, v00));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v01, Min128Upper(d, v01, v01));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v10, Min128Upper(d, v10, v10));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v11, Min128Upper(d, v11, v11));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v00, Max128Upper(d, v00, v00));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v01, Max128Upper(d, v01, v01));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v10, Max128Upper(d, v10, v10));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v11, Max128Upper(d, v11, v11));
|
||||||
|
|
||||||
|
// Equivalent but not equal (chooses second arg)
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v01, Min128Upper(d, v00, v01));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v11, Min128Upper(d, v10, v11));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v00, Min128Upper(d, v01, v00));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v10, Min128Upper(d, v11, v10));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v00, Max128Upper(d, v01, v00));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v10, Max128Upper(d, v11, v10));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v01, Max128Upper(d, v00, v01));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v11, Max128Upper(d, v10, v11));
|
||||||
|
|
||||||
|
// First arg less
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v01, Min128Upper(d, v01, v10));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v10, Max128Upper(d, v01, v10));
|
||||||
|
|
||||||
|
// Second arg less
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v01, Min128Upper(d, v10, v01));
|
||||||
|
HWY_ASSERT_VEC_EQ(d, v10, Max128Upper(d, v10, v01));
|
||||||
|
|
||||||
|
// Also check 128-bit blocks are independent
|
||||||
for (size_t rep = 0; rep < AdjustedReps(1000); ++rep) {
|
for (size_t rep = 0; rep < AdjustedReps(1000); ++rep) {
|
||||||
for (size_t i = 0; i < N; ++i) {
|
for (size_t i = 0; i < N; ++i) {
|
||||||
in1[i] = Random64(&rng);
|
a_lanes[i] = Random64(&rng);
|
||||||
in2[i] = Random64(&rng);
|
b_lanes[i] = Random64(&rng);
|
||||||
}
|
}
|
||||||
|
const V a = Load(d, a_lanes.get());
|
||||||
|
const V b = Load(d, b_lanes.get());
|
||||||
for (size_t i = 0; i < N; i += 2) {
|
for (size_t i = 0; i < N; i += 2) {
|
||||||
expected_even[i] = Mul128(in1[i], in2[i], &expected_even[i + 1]);
|
const bool lt = a_lanes[i + 1] < b_lanes[i + 1];
|
||||||
expected_odd[i] = Mul128(in1[i + 1], in2[i + 1], &expected_odd[i + 1]);
|
min_lanes[i + 0] = lt ? a_lanes[i + 0] : b_lanes[i + 0];
|
||||||
|
min_lanes[i + 1] = lt ? a_lanes[i + 1] : b_lanes[i + 1];
|
||||||
|
max_lanes[i + 0] = lt ? b_lanes[i + 0] : a_lanes[i + 0];
|
||||||
|
max_lanes[i + 1] = lt ? b_lanes[i + 1] : a_lanes[i + 1];
|
||||||
}
|
}
|
||||||
|
HWY_ASSERT_VEC_EQ(d, min_lanes.get(), Min128Upper(d, a, b));
|
||||||
const auto a = Load(d, in1.get());
|
HWY_ASSERT_VEC_EQ(d, max_lanes.get(), Max128Upper(d, a, b));
|
||||||
const auto b = Load(d, in2.get());
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_even.get(), MulEven(a, b));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected_odd.get(), MulOdd(a, b));
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
(void)d;
|
|
||||||
#endif // HWY_TARGET != HWY_SCALAR
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllMulEven() {
|
|
||||||
ForGEVectors<64, TestMulEven> test;
|
|
||||||
test(int32_t());
|
|
||||||
test(uint32_t());
|
|
||||||
|
|
||||||
ForGEVectors<128, TestMulEvenOdd64>()(uint64_t());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestMulAdd {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const auto k0 = Zero(d);
|
|
||||||
const auto kNeg0 = Set(d, T(-0.0));
|
|
||||||
const auto v1 = Iota(d, 1);
|
|
||||||
const auto v2 = Iota(d, 2);
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto expected = AllocateAligned<T>(N);
|
|
||||||
HWY_ASSERT_VEC_EQ(d, k0, MulAdd(k0, k0, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v2, MulAdd(k0, v1, v2));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v2, MulAdd(v1, k0, v2));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, k0, NegMulAdd(k0, k0, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v2, NegMulAdd(k0, v1, v2));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v2, NegMulAdd(v1, k0, v2));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((i + 1) * (i + 2));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulAdd(v2, v1, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulAdd(v1, v2, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulAdd(Neg(v2), v1, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulAdd(v1, Neg(v2), k0));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((i + 2) * (i + 2) + (i + 1));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulAdd(v2, v2, v1));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulAdd(Neg(v2), v2, v1));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] =
|
|
||||||
T(-T(i + 2u) * static_cast<T>(i + 2) + static_cast<T>(1 + i));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulAdd(v2, v2, v1));
|
|
||||||
|
|
||||||
HWY_ASSERT_VEC_EQ(d, k0, MulSub(k0, k0, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, kNeg0, NegMulSub(k0, k0, k0));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = -T(i + 2);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulSub(k0, v1, v2));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulSub(v1, k0, v2));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulSub(Neg(k0), v1, v2));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulSub(v1, Neg(k0), v2));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((i + 1) * (i + 2));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulSub(v1, v2, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulSub(v2, v1, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulSub(Neg(v1), v2, k0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulSub(v2, Neg(v1), k0));
|
|
||||||
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = static_cast<T>((i + 2) * (i + 2) - (1 + i));
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), MulSub(v2, v2, v1));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), NegMulSub(Neg(v2), v2, v1));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllMulAdd() {
|
|
||||||
ForFloatTypes(ForPartialVectors<TestMulAdd>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestReorderWidenMulAccumulate {
|
|
||||||
template <typename TN, class DN>
|
|
||||||
HWY_NOINLINE void operator()(TN /*unused*/, DN dn) {
|
|
||||||
using TW = MakeWide<TN>;
|
|
||||||
const RepartitionToWide<DN> dw;
|
|
||||||
const auto f0 = Zero(dw);
|
|
||||||
const auto f1 = Set(dw, 1.0f);
|
|
||||||
const auto fi = Iota(dw, 1);
|
|
||||||
const auto bf0 = ReorderDemote2To(dn, f0, f0);
|
|
||||||
const auto bf1 = ReorderDemote2To(dn, f1, f1);
|
|
||||||
const auto bfi = ReorderDemote2To(dn, fi, fi);
|
|
||||||
const size_t NW = Lanes(dw);
|
|
||||||
auto delta = AllocateAligned<TW>(2 * NW);
|
|
||||||
for (size_t i = 0; i < 2 * NW; ++i) {
|
|
||||||
delta[i] = 0.0f;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Any input zero => both outputs zero
|
|
||||||
auto sum1 = f0;
|
|
||||||
HWY_ASSERT_VEC_EQ(dw, f0,
|
|
||||||
ReorderWidenMulAccumulate(dw, bf0, bf0, f0, sum1));
|
|
||||||
HWY_ASSERT_VEC_EQ(dw, f0, sum1);
|
|
||||||
HWY_ASSERT_VEC_EQ(dw, f0,
|
|
||||||
ReorderWidenMulAccumulate(dw, bf0, bfi, f0, sum1));
|
|
||||||
HWY_ASSERT_VEC_EQ(dw, f0, sum1);
|
|
||||||
HWY_ASSERT_VEC_EQ(dw, f0,
|
|
||||||
ReorderWidenMulAccumulate(dw, bfi, bf0, f0, sum1));
|
|
||||||
HWY_ASSERT_VEC_EQ(dw, f0, sum1);
|
|
||||||
|
|
||||||
// delta[p] := 1.0, all others zero. For each p: Dot(delta, all-ones) == 1.
|
|
||||||
for (size_t p = 0; p < 2 * NW; ++p) {
|
|
||||||
delta[p] = 1.0f;
|
|
||||||
const auto delta0 = Load(dw, delta.get() + 0);
|
|
||||||
const auto delta1 = Load(dw, delta.get() + NW);
|
|
||||||
delta[p] = 0.0f;
|
|
||||||
const auto bf_delta = ReorderDemote2To(dn, delta0, delta1);
|
|
||||||
|
|
||||||
{
|
|
||||||
sum1 = f0;
|
|
||||||
const auto sum0 =
|
|
||||||
ReorderWidenMulAccumulate(dw, bf_delta, bf1, f0, sum1);
|
|
||||||
HWY_ASSERT_EQ(1.0f, GetLane(SumOfLanes(dw, Add(sum0, sum1))));
|
|
||||||
}
|
|
||||||
// Swapped arg order
|
|
||||||
{
|
|
||||||
sum1 = f0;
|
|
||||||
const auto sum0 =
|
|
||||||
ReorderWidenMulAccumulate(dw, bf1, bf_delta, f0, sum1);
|
|
||||||
HWY_ASSERT_EQ(1.0f, GetLane(SumOfLanes(dw, Add(sum0, sum1))));
|
|
||||||
}
|
|
||||||
// Start with nonzero sum0 or sum1
|
|
||||||
{
|
|
||||||
sum1 = delta1;
|
|
||||||
const auto sum0 =
|
|
||||||
ReorderWidenMulAccumulate(dw, bf_delta, bf1, delta0, sum1);
|
|
||||||
HWY_ASSERT_EQ(2.0f, GetLane(SumOfLanes(dw, Add(sum0, sum1))));
|
|
||||||
}
|
|
||||||
// Start with nonzero sum0 or sum1, and swap arg order
|
|
||||||
{
|
|
||||||
sum1 = delta1;
|
|
||||||
const auto sum0 =
|
|
||||||
ReorderWidenMulAccumulate(dw, bf1, bf_delta, delta0, sum1);
|
|
||||||
HWY_ASSERT_EQ(2.0f, GetLane(SumOfLanes(dw, Add(sum0, sum1))));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllReorderWidenMulAccumulate() {
|
|
||||||
ForShrinkableVectors<TestReorderWidenMulAccumulate>()(bfloat16_t());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestDiv {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const auto v = Iota(d, T(-2));
|
|
||||||
const auto v1 = Set(d, T(1));
|
|
||||||
|
|
||||||
// Unchanged after division by 1.
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v, Div(v, v1));
|
|
||||||
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto expected = AllocateAligned<T>(N);
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
expected[i] = (T(i) - 2) / T(2);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected.get(), Div(v, Set(d, T(2))));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllDiv() { ForFloatTypes(ForPartialVectors<TestDiv>()); }
|
|
||||||
|
|
||||||
struct TestApproximateReciprocal {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const auto v = Iota(d, T(-2));
|
|
||||||
const auto nonzero = IfThenElse(Eq(v, Zero(d)), Set(d, T(1)), v);
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto input = AllocateAligned<T>(N);
|
|
||||||
Store(nonzero, d, input.get());
|
|
||||||
|
|
||||||
auto actual = AllocateAligned<T>(N);
|
|
||||||
Store(ApproximateReciprocal(nonzero), d, actual.get());
|
|
||||||
|
|
||||||
double max_l1 = 0.0;
|
|
||||||
double worst_expected = 0.0;
|
|
||||||
double worst_actual = 0.0;
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
const double expected = 1.0 / input[i];
|
|
||||||
const double l1 = std::abs(expected - actual[i]);
|
|
||||||
if (l1 > max_l1) {
|
|
||||||
max_l1 = l1;
|
|
||||||
worst_expected = expected;
|
|
||||||
worst_actual = actual[i];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
const double abs_worst_expected = std::abs(worst_expected);
|
|
||||||
if (abs_worst_expected > 1E-5) {
|
|
||||||
const double max_rel = max_l1 / abs_worst_expected;
|
|
||||||
fprintf(stderr, "max l1 %f rel %f (%f vs %f)\n", max_l1, max_rel,
|
|
||||||
worst_expected, worst_actual);
|
|
||||||
HWY_ASSERT(max_rel < 0.004);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllApproximateReciprocal() {
|
HWY_NOINLINE void TestAllMinMax128Upper() {
|
||||||
ForPartialVectors<TestApproximateReciprocal>()(float());
|
ForGEVectors<128, TestMinMax128Upper>()(uint64_t());
|
||||||
}
|
|
||||||
|
|
||||||
struct TestSquareRoot {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const auto vi = Iota(d, 0);
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vi, Sqrt(Mul(vi, vi)));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllSquareRoot() {
|
|
||||||
ForFloatTypes(ForPartialVectors<TestSquareRoot>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestReciprocalSquareRoot {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const auto v = Set(d, 123.0f);
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto lanes = AllocateAligned<T>(N);
|
|
||||||
Store(ApproximateReciprocalSqrt(v), d, lanes.get());
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
float err = lanes[i] - 0.090166f;
|
|
||||||
if (err < 0.0f) err = -err;
|
|
||||||
if (err >= 4E-4f) {
|
|
||||||
HWY_ABORT("Lane %" PRIu64 "(%" PRIu64 "): actual %f err %f\n",
|
|
||||||
static_cast<uint64_t>(i), static_cast<uint64_t>(N), lanes[i],
|
|
||||||
err);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllReciprocalSquareRoot() {
|
|
||||||
ForPartialVectors<TestReciprocalSquareRoot>()(float());
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T, class D>
|
|
||||||
AlignedFreeUniquePtr<T[]> RoundTestCases(T /*unused*/, D d, size_t& padded) {
|
|
||||||
const T eps = std::numeric_limits<T>::epsilon();
|
|
||||||
const T test_cases[] = {
|
|
||||||
// +/- 1
|
|
||||||
T(1),
|
|
||||||
T(-1),
|
|
||||||
// +/- 0
|
|
||||||
T(0),
|
|
||||||
T(-0),
|
|
||||||
// near 0
|
|
||||||
T(0.4),
|
|
||||||
T(-0.4),
|
|
||||||
// +/- integer
|
|
||||||
T(4),
|
|
||||||
T(-32),
|
|
||||||
// positive near limit
|
|
||||||
MantissaEnd<T>() - T(1.5),
|
|
||||||
MantissaEnd<T>() + T(1.5),
|
|
||||||
// negative near limit
|
|
||||||
-MantissaEnd<T>() - T(1.5),
|
|
||||||
-MantissaEnd<T>() + T(1.5),
|
|
||||||
// positive tiebreak
|
|
||||||
T(1.5),
|
|
||||||
T(2.5),
|
|
||||||
// negative tiebreak
|
|
||||||
T(-1.5),
|
|
||||||
T(-2.5),
|
|
||||||
// positive +/- delta
|
|
||||||
T(2.0001),
|
|
||||||
T(3.9999),
|
|
||||||
// negative +/- delta
|
|
||||||
T(-999.9999),
|
|
||||||
T(-998.0001),
|
|
||||||
// positive +/- epsilon
|
|
||||||
T(1) + eps,
|
|
||||||
T(1) - eps,
|
|
||||||
// negative +/- epsilon
|
|
||||||
T(-1) + eps,
|
|
||||||
T(-1) - eps,
|
|
||||||
// +/- huge (but still fits in float)
|
|
||||||
T(1E34),
|
|
||||||
T(-1E35),
|
|
||||||
// +/- infinity
|
|
||||||
std::numeric_limits<T>::infinity(),
|
|
||||||
-std::numeric_limits<T>::infinity(),
|
|
||||||
// qNaN
|
|
||||||
GetLane(NaN(d))
|
|
||||||
};
|
|
||||||
const size_t kNumTestCases = sizeof(test_cases) / sizeof(test_cases[0]);
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
padded = RoundUpTo(kNumTestCases, N); // allow loading whole vectors
|
|
||||||
auto in = AllocateAligned<T>(padded);
|
|
||||||
auto expected = AllocateAligned<T>(padded);
|
|
||||||
std::copy(test_cases, test_cases + kNumTestCases, in.get());
|
|
||||||
std::fill(in.get() + kNumTestCases, in.get() + padded, T(0));
|
|
||||||
return in;
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestRound {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T t, D d) {
|
|
||||||
size_t padded;
|
|
||||||
auto in = RoundTestCases(t, d, padded);
|
|
||||||
auto expected = AllocateAligned<T>(padded);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < padded; ++i) {
|
|
||||||
// Avoid [std::]round, which does not round to nearest *even*.
|
|
||||||
// NOTE: std:: version from C++11 cmath is not defined in RVV GCC, see
|
|
||||||
// https://lists.freebsd.org/pipermail/freebsd-current/2014-January/048130.html
|
|
||||||
expected[i] = static_cast<T>(nearbyint(in[i]));
|
|
||||||
}
|
|
||||||
for (size_t i = 0; i < padded; i += Lanes(d)) {
|
|
||||||
HWY_ASSERT_VEC_EQ(d, &expected[i], Round(Load(d, &in[i])));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllRound() {
|
|
||||||
ForFloatTypes(ForPartialVectors<TestRound>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestNearestInt {
|
|
||||||
template <typename TF, class DF>
|
|
||||||
HWY_NOINLINE void operator()(TF tf, const DF df) {
|
|
||||||
using TI = MakeSigned<TF>;
|
|
||||||
const RebindToSigned<DF> di;
|
|
||||||
|
|
||||||
size_t padded;
|
|
||||||
auto in = RoundTestCases(tf, df, padded);
|
|
||||||
auto expected = AllocateAligned<TI>(padded);
|
|
||||||
|
|
||||||
constexpr double max = static_cast<double>(LimitsMax<TI>());
|
|
||||||
for (size_t i = 0; i < padded; ++i) {
|
|
||||||
if (std::isnan(in[i])) {
|
|
||||||
// We replace NaN with 0 below (no_nan)
|
|
||||||
expected[i] = 0;
|
|
||||||
} else if (std::isinf(in[i]) || double(std::abs(in[i])) >= max) {
|
|
||||||
// Avoid undefined result for lrintf
|
|
||||||
expected[i] = std::signbit(in[i]) ? LimitsMin<TI>() : LimitsMax<TI>();
|
|
||||||
} else {
|
|
||||||
expected[i] = static_cast<TI>(lrintf(in[i]));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for (size_t i = 0; i < padded; i += Lanes(df)) {
|
|
||||||
const auto v = Load(df, &in[i]);
|
|
||||||
const auto no_nan = IfThenElse(Eq(v, v), v, Zero(df));
|
|
||||||
HWY_ASSERT_VEC_EQ(di, &expected[i], NearestInt(no_nan));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllNearestInt() {
|
|
||||||
ForPartialVectors<TestNearestInt>()(float());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestTrunc {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T t, D d) {
|
|
||||||
size_t padded;
|
|
||||||
auto in = RoundTestCases(t, d, padded);
|
|
||||||
auto expected = AllocateAligned<T>(padded);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < padded; ++i) {
|
|
||||||
// NOTE: std:: version from C++11 cmath is not defined in RVV GCC, see
|
|
||||||
// https://lists.freebsd.org/pipermail/freebsd-current/2014-January/048130.html
|
|
||||||
expected[i] = static_cast<T>(trunc(in[i]));
|
|
||||||
}
|
|
||||||
for (size_t i = 0; i < padded; i += Lanes(d)) {
|
|
||||||
HWY_ASSERT_VEC_EQ(d, &expected[i], Trunc(Load(d, &in[i])));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllTrunc() {
|
|
||||||
ForFloatTypes(ForPartialVectors<TestTrunc>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestCeil {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T t, D d) {
|
|
||||||
size_t padded;
|
|
||||||
auto in = RoundTestCases(t, d, padded);
|
|
||||||
auto expected = AllocateAligned<T>(padded);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < padded; ++i) {
|
|
||||||
expected[i] = std::ceil(in[i]);
|
|
||||||
}
|
|
||||||
for (size_t i = 0; i < padded; i += Lanes(d)) {
|
|
||||||
HWY_ASSERT_VEC_EQ(d, &expected[i], Ceil(Load(d, &in[i])));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllCeil() {
|
|
||||||
ForFloatTypes(ForPartialVectors<TestCeil>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestFloor {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T t, D d) {
|
|
||||||
size_t padded;
|
|
||||||
auto in = RoundTestCases(t, d, padded);
|
|
||||||
auto expected = AllocateAligned<T>(padded);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < padded; ++i) {
|
|
||||||
expected[i] = std::floor(in[i]);
|
|
||||||
}
|
|
||||||
for (size_t i = 0; i < padded; i += Lanes(d)) {
|
|
||||||
HWY_ASSERT_VEC_EQ(d, &expected[i], Floor(Load(d, &in[i])));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllFloor() {
|
|
||||||
ForFloatTypes(ForPartialVectors<TestFloor>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestSumOfLanes {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto in_lanes = AllocateAligned<T>(N);
|
|
||||||
|
|
||||||
// Lane i = bit i, higher lanes 0
|
|
||||||
double sum = 0.0;
|
|
||||||
// Avoid setting sign bit and cap at double precision
|
|
||||||
constexpr size_t kBits = HWY_MIN(sizeof(T) * 8 - 1, 51);
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = i < kBits ? static_cast<T>(1ull << i) : 0;
|
|
||||||
sum += static_cast<double>(in_lanes[i]);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, T(sum)),
|
|
||||||
SumOfLanes(d, Load(d, in_lanes.get())));
|
|
||||||
|
|
||||||
// Lane i = i (iota) to include upper lanes
|
|
||||||
sum = 0.0;
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
sum += static_cast<double>(i);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, T(sum)), SumOfLanes(d, Iota(d, 0)));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllSumOfLanes() {
|
|
||||||
ForUIF3264(ForPartialVectors<TestSumOfLanes>());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestMinOfLanes {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto in_lanes = AllocateAligned<T>(N);
|
|
||||||
|
|
||||||
// Lane i = bit i, higher lanes = 2 (not the minimum)
|
|
||||||
T min = HighestValue<T>();
|
|
||||||
// Avoid setting sign bit and cap at double precision
|
|
||||||
constexpr size_t kBits = HWY_MIN(sizeof(T) * 8 - 1, 51);
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = i < kBits ? static_cast<T>(1ull << i) : 2;
|
|
||||||
min = HWY_MIN(min, in_lanes[i]);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, min), MinOfLanes(d, Load(d, in_lanes.get())));
|
|
||||||
|
|
||||||
// Lane i = N - i to include upper lanes
|
|
||||||
min = HighestValue<T>();
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = static_cast<T>(N - i); // no 8-bit T so no wraparound
|
|
||||||
min = HWY_MIN(min, in_lanes[i]);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, min), MinOfLanes(d, Load(d, in_lanes.get())));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
struct TestMaxOfLanes {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto in_lanes = AllocateAligned<T>(N);
|
|
||||||
|
|
||||||
T max = LowestValue<T>();
|
|
||||||
// Avoid setting sign bit and cap at double precision
|
|
||||||
constexpr size_t kBits = HWY_MIN(sizeof(T) * 8 - 1, 51);
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = i < kBits ? static_cast<T>(1ull << i) : 0;
|
|
||||||
max = HWY_MAX(max, in_lanes[i]);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, max), MaxOfLanes(d, Load(d, in_lanes.get())));
|
|
||||||
|
|
||||||
// Lane i = i to include upper lanes
|
|
||||||
max = LowestValue<T>();
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = static_cast<T>(i); // no 8-bit T so no wraparound
|
|
||||||
max = HWY_MAX(max, in_lanes[i]);
|
|
||||||
}
|
|
||||||
HWY_ASSERT_VEC_EQ(d, Set(d, max), MaxOfLanes(d, Load(d, in_lanes.get())));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllMinMaxOfLanes() {
|
|
||||||
const ForPartialVectors<TestMinOfLanes> test_min;
|
|
||||||
const ForPartialVectors<TestMaxOfLanes> test_max;
|
|
||||||
ForUIF3264(test_min);
|
|
||||||
ForUIF3264(test_max);
|
|
||||||
test_min(uint16_t());
|
|
||||||
test_max(uint16_t());
|
|
||||||
test_min(int16_t());
|
|
||||||
test_max(int16_t());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestAbsDiff {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
auto in_lanes_a = AllocateAligned<T>(N);
|
|
||||||
auto in_lanes_b = AllocateAligned<T>(N);
|
|
||||||
auto out_lanes = AllocateAligned<T>(N);
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes_a[i] = static_cast<T>((i ^ 1u) << i);
|
|
||||||
in_lanes_b[i] = static_cast<T>(i << i);
|
|
||||||
out_lanes[i] = std::abs(in_lanes_a[i] - in_lanes_b[i]);
|
|
||||||
}
|
|
||||||
const auto a = Load(d, in_lanes_a.get());
|
|
||||||
const auto b = Load(d, in_lanes_b.get());
|
|
||||||
const auto expected = Load(d, out_lanes.get());
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected, AbsDiff(a, b));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, expected, AbsDiff(b, a));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllAbsDiff() {
|
|
||||||
ForPartialVectors<TestAbsDiff>()(float());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestSumsOf8 {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
RandomState rng;
|
|
||||||
|
|
||||||
const size_t N = Lanes(d);
|
|
||||||
if (N < 8) return;
|
|
||||||
const Repartition<uint64_t, D> du64;
|
|
||||||
|
|
||||||
auto in_lanes = AllocateAligned<T>(N);
|
|
||||||
auto sum_lanes = AllocateAligned<uint64_t>(N / 8);
|
|
||||||
|
|
||||||
for (size_t rep = 0; rep < 100; ++rep) {
|
|
||||||
for (size_t i = 0; i < N; ++i) {
|
|
||||||
in_lanes[i] = Random64(&rng) & 0xFF;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (size_t idx_sum = 0; idx_sum < N / 8; ++idx_sum) {
|
|
||||||
uint64_t sum = 0;
|
|
||||||
for (size_t i = 0; i < 8; ++i) {
|
|
||||||
sum += in_lanes[idx_sum * 8 + i];
|
|
||||||
}
|
|
||||||
sum_lanes[idx_sum] = sum;
|
|
||||||
}
|
|
||||||
|
|
||||||
const Vec<D> in = Load(d, in_lanes.get());
|
|
||||||
HWY_ASSERT_VEC_EQ(du64, sum_lanes.get(), SumsOf8(in));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllSumsOf8() {
|
|
||||||
ForGEVectors<64, TestSumsOf8>()(uint8_t());
|
|
||||||
}
|
|
||||||
|
|
||||||
struct TestNeg {
|
|
||||||
template <typename T, class D>
|
|
||||||
HWY_NOINLINE void operator()(T /*unused*/, D d) {
|
|
||||||
const auto v0 = Zero(d);
|
|
||||||
const auto vn = Set(d, T(-3));
|
|
||||||
const auto vp = Set(d, T(3));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, v0, Neg(v0));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vp, Neg(vn));
|
|
||||||
HWY_ASSERT_VEC_EQ(d, vn, Neg(vp));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
HWY_NOINLINE void TestAllNeg() {
|
|
||||||
ForSignedTypes(ForPartialVectors<TestNeg>());
|
|
||||||
ForFloatTypes(ForPartialVectors<TestNeg>());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
// NOLINTNEXTLINE(google-readability-namespace-comments)
|
||||||
|
@ -1127,35 +435,12 @@ namespace hwy {
|
||||||
HWY_BEFORE_TEST(HwyArithmeticTest);
|
HWY_BEFORE_TEST(HwyArithmeticTest);
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllPlusMinus);
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllPlusMinus);
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllSaturatingArithmetic);
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllSaturatingArithmetic);
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMinMax);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMinMax128);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllAverage);
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllAverage);
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllAbs);
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllAbs);
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMul);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMulHigh);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMulEven);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMulAdd);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllReorderWidenMulAccumulate);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllDiv);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllApproximateReciprocal);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllSquareRoot);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllReciprocalSquareRoot);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllSumOfLanes);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMinMaxOfLanes);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllRound);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllNearestInt);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllTrunc);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllCeil);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllFloor);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllAbsDiff);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllSumsOf8);
|
|
||||||
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllNeg);
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllNeg);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMinMax);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMinMax128);
|
||||||
|
HWY_EXPORT_AND_TEST_P(HwyArithmeticTest, TestAllMinMax128Upper);
|
||||||
} // namespace hwy
|
} // namespace hwy
|
||||||
|
|
||||||
// Ought not to be necessary, but without this, no tests run on RVV.
|
|
||||||
int main(int argc, char** argv) {
|
|
||||||
::testing::InitGoogleTest(&argc, argv);
|
|
||||||
return RUN_ALL_TESTS();
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
Некоторые файлы не были показаны из-за слишком большого количества измененных файлов Показать больше
Загрузка…
Ссылка в новой задаче