usearch 2.10.3 → 2.10.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (112) hide show
  1. package/README.md +1 -1
  2. package/binding.gyp +101 -0
  3. package/c/CMakeLists.txt +27 -0
  4. package/c/README.md +34 -0
  5. package/c/lib.cpp +362 -0
  6. package/c/test.c +382 -0
  7. package/c/usearch.h +356 -0
  8. package/cmake/clipp.cmake +7 -0
  9. package/cmake/config.cmake.in +15 -0
  10. package/cmake/emscripten.cmake +8 -0
  11. package/cmake/jemalloc.cmake +62 -0
  12. package/cmake/pkg-config.pc.in +4 -0
  13. package/cmake/usearchConfigVersion.cmake.in +20 -0
  14. package/cpp/CMakeLists.txt +14 -0
  15. package/cpp/README.md +161 -0
  16. package/cpp/bench.cpp +616 -0
  17. package/cpp/test.cpp +324 -0
  18. package/fp16/CMakeLists.txt +176 -0
  19. package/fp16/LICENSE +11 -0
  20. package/fp16/README.md +20 -0
  21. package/fp16/bench/alt-element.cc +89 -0
  22. package/fp16/bench/from-alt-array.cc +175 -0
  23. package/fp16/bench/from-ieee-array.cc +399 -0
  24. package/fp16/bench/ieee-element.cc +272 -0
  25. package/fp16/bench/to-alt-array.cc +91 -0
  26. package/fp16/bench/to-ieee-array.cc +303 -0
  27. package/fp16/cmake/DownloadGoogleBenchmark.cmake +15 -0
  28. package/fp16/cmake/DownloadGoogleTest.cmake +15 -0
  29. package/fp16/cmake/DownloadPSimd.cmake +15 -0
  30. package/fp16/configure.py +65 -0
  31. package/fp16/confu.yaml +8 -0
  32. package/fp16/include/fp16/__init__.py +0 -0
  33. package/fp16/include/fp16/avx.py +50 -0
  34. package/fp16/include/fp16/avx2.py +53 -0
  35. package/fp16/include/fp16/bitcasts.h +92 -0
  36. package/fp16/include/fp16/fp16.h +451 -0
  37. package/fp16/include/fp16/psimd.h +131 -0
  38. package/fp16/include/fp16.h +11 -0
  39. package/fp16/jni/Android.mk +6 -0
  40. package/fp16/jni/Application.mk +4 -0
  41. package/fp16/test/alt-from-fp32-value.cc +498 -0
  42. package/fp16/test/alt-to-fp32-bits.cc +263 -0
  43. package/fp16/test/alt-to-fp32-psimd.cc +145 -0
  44. package/fp16/test/alt-to-fp32-value.cc +348 -0
  45. package/fp16/test/alt-to-fp32x2-psimd.cc +245 -0
  46. package/fp16/test/bitcasts.cc +56 -0
  47. package/fp16/test/ieee-from-fp32-value.cc +523 -0
  48. package/fp16/test/ieee-to-fp32-bits.cc +333 -0
  49. package/fp16/test/ieee-to-fp32-psimd.cc +312 -0
  50. package/fp16/test/ieee-to-fp32-value.cc +420 -0
  51. package/fp16/test/ieee-to-fp32x2-psimd.cc +541 -0
  52. package/fp16/test/peachpy/alt-xmm-to-fp32-xmm-avx.cc +96 -0
  53. package/fp16/test/peachpy/alt-xmm-to-fp32-ymm-avx2.cc +96 -0
  54. package/fp16/test/peachpy/stubs.py +38 -0
  55. package/fp16/test/tables.cc +530 -0
  56. package/fp16/test/tables.h +13 -0
  57. package/fp16/third-party/THHalf.h +92 -0
  58. package/fp16/third-party/eigen-half.h +170 -0
  59. package/fp16/third-party/float16-compressor.h +80 -0
  60. package/fp16/third-party/half.hpp +2916 -0
  61. package/fp16/third-party/npy-halffloat.h +328 -0
  62. package/include/usearch/index.hpp +3838 -0
  63. package/include/usearch/index_dense.hpp +2004 -0
  64. package/include/usearch/index_plugins.hpp +2255 -0
  65. package/javascript/dist-package-cjs.json +3 -0
  66. package/javascript/dist-package-esm.json +3 -0
  67. package/javascript/lib.cpp +350 -0
  68. package/javascript/node-gyp-build.d.ts +1 -0
  69. package/javascript/tsconfig-base.json +25 -0
  70. package/javascript/tsconfig-cjs.json +8 -0
  71. package/javascript/tsconfig-esm.json +8 -0
  72. package/javascript/usearch.test.js +84 -0
  73. package/javascript/usearch.ts +640 -0
  74. package/package.json +10 -3
  75. package/prebuilds/darwin-arm64+x64/usearch.node +0 -0
  76. package/prebuilds/linux-arm64/usearch.node +0 -0
  77. package/prebuilds/linux-x64/usearch.node +0 -0
  78. package/prebuilds/win32-ia32/usearch.node +0 -0
  79. package/prebuilds/win32-x64/usearch.node +0 -0
  80. package/simsimd/.releaserc +87 -0
  81. package/simsimd/CONTRIBUTING.md +114 -0
  82. package/simsimd/Cargo.lock +709 -0
  83. package/simsimd/Cargo.toml +44 -0
  84. package/simsimd/LICENSE +201 -0
  85. package/simsimd/README.md +708 -0
  86. package/simsimd/binding.gyp +36 -0
  87. package/simsimd/build.rs +51 -0
  88. package/simsimd/c/lib.c +110 -0
  89. package/simsimd/include/simsimd/binary.h +258 -0
  90. package/simsimd/include/simsimd/dot.h +1237 -0
  91. package/simsimd/include/simsimd/geospatial.h +37 -0
  92. package/simsimd/include/simsimd/probability.h +563 -0
  93. package/simsimd/include/simsimd/simsimd.h +1129 -0
  94. package/simsimd/include/simsimd/spatial.h +1019 -0
  95. package/simsimd/include/simsimd/types.h +309 -0
  96. package/simsimd/javascript/bench.js +127 -0
  97. package/simsimd/javascript/dist-package-cjs.json +3 -0
  98. package/simsimd/javascript/dist-package-esm.json +3 -0
  99. package/simsimd/javascript/fallback.ts +203 -0
  100. package/simsimd/javascript/lib.c +107 -0
  101. package/simsimd/javascript/node-gyp-build.d.ts +1 -0
  102. package/simsimd/javascript/simsimd.ts +135 -0
  103. package/simsimd/javascript/test.mjs +179 -0
  104. package/simsimd/javascript/tsconfig-base.json +37 -0
  105. package/simsimd/javascript/tsconfig-cjs.json +8 -0
  106. package/simsimd/javascript/tsconfig-esm.json +8 -0
  107. package/simsimd/package-lock.json +1709 -0
  108. package/simsimd/package.json +61 -0
  109. package/simsimd/rust/benches/cosine.rs +27 -0
  110. package/simsimd/rust/benches/native.rs +30 -0
  111. package/simsimd/rust/benches/sqeuclidean.rs +27 -0
  112. package/simsimd/rust/lib.rs +778 -0
@@ -0,0 +1,175 @@
1
+ #include <benchmark/benchmark.h>
2
+
3
+ #include <fp16.h>
4
+ #ifndef EMSCRIPTEN
5
+ #include <fp16/psimd.h>
6
+ #endif
7
+
8
+ #include <vector>
9
+ #include <random>
10
+ #include <chrono>
11
+ #include <functional>
12
+ #include <algorithm>
13
+
14
+ #if defined(__ARM_NEON__) || defined(__aarch64__)
15
+ #include <arm_neon.h>
16
+ #endif
17
+
18
+
19
+ static void fp16_alt_to_fp32_bits(benchmark::State& state) {
20
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
21
+ auto rng = std::bind(std::uniform_int_distribution<uint16_t>(0, 0x7BFF), std::mt19937(seed));
22
+
23
+ std::vector<uint16_t> fp16(state.range(0));
24
+ std::vector<uint32_t> fp32(state.range(0));
25
+ std::generate(fp16.begin(), fp16.end(),
26
+ [&rng]{ return fp16_alt_from_fp32_value(rng()); });
27
+
28
+ while (state.KeepRunning()) {
29
+ uint16_t* input = fp16.data();
30
+ benchmark::DoNotOptimize(input);
31
+
32
+ uint32_t* output = fp32.data();
33
+ const size_t n = state.range(0);
34
+ for (size_t i = 0; i < n; i++) {
35
+ output[i] = fp16_alt_to_fp32_bits(input[i]);
36
+ }
37
+
38
+ benchmark::DoNotOptimize(output);
39
+ }
40
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
41
+ }
42
+ BENCHMARK(fp16_alt_to_fp32_bits)->RangeMultiplier(2)->Range(1<<10, 64<<20);
43
+
44
+ static void fp16_alt_to_fp32_value(benchmark::State& state) {
45
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
46
+ auto rng = std::bind(std::uniform_int_distribution<uint16_t>(0, 0x7BFF), std::mt19937(seed));
47
+
48
+ std::vector<uint16_t> fp16(state.range(0));
49
+ std::vector<float> fp32(state.range(0));
50
+ std::generate(fp16.begin(), fp16.end(),
51
+ [&rng]{ return fp16_alt_from_fp32_value(rng()); });
52
+
53
+ while (state.KeepRunning()) {
54
+ uint16_t* input = fp16.data();
55
+ benchmark::DoNotOptimize(input);
56
+
57
+ float* output = fp32.data();
58
+ const size_t n = state.range(0);
59
+ for (size_t i = 0; i < n; i++) {
60
+ output[i] = fp16_alt_to_fp32_value(input[i]);
61
+ }
62
+
63
+ benchmark::DoNotOptimize(output);
64
+ }
65
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
66
+ }
67
+ BENCHMARK(fp16_alt_to_fp32_value)->RangeMultiplier(2)->Range(1<<10, 64<<20);
68
+
69
+ #ifndef EMSCRIPTEN
70
+ static void fp16_alt_to_fp32_psimd(benchmark::State& state) {
71
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
72
+ auto rng = std::bind(std::uniform_int_distribution<uint16_t>(0, 0x7BFF), std::mt19937(seed));
73
+
74
+ std::vector<uint16_t> fp16(state.range(0));
75
+ std::vector<float> fp32(state.range(0));
76
+ std::generate(fp16.begin(), fp16.end(),
77
+ [&rng]{ return fp16_alt_from_fp32_value(rng()); });
78
+
79
+ while (state.KeepRunning()) {
80
+ uint16_t* input = fp16.data();
81
+ benchmark::DoNotOptimize(input);
82
+
83
+ float* output = fp32.data();
84
+ const size_t n = state.range(0);
85
+ for (size_t i = 0; i < n - 4; i += 4) {
86
+ psimd_store_f32(&output[i],
87
+ fp16_alt_to_fp32_psimd(
88
+ psimd_load_u16(&input[i])));
89
+ }
90
+ const psimd_u16 last_vector = { input[n - 4], input[n - 3], input[n - 2], input[n - 1] };
91
+ psimd_store_f32(&output[n - 4],
92
+ fp16_alt_to_fp32_psimd(last_vector));
93
+
94
+ benchmark::DoNotOptimize(output);
95
+ }
96
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
97
+ }
98
+ BENCHMARK(fp16_alt_to_fp32_psimd)->RangeMultiplier(2)->Range(1<<10, 64<<20);
99
+
100
+ static void fp16_alt_to_fp32x2_psimd(benchmark::State& state) {
101
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
102
+ auto rng = std::bind(std::uniform_int_distribution<uint16_t>(0, 0x7BFF), std::mt19937(seed));
103
+
104
+ std::vector<uint16_t> fp16(state.range(0));
105
+ std::vector<float> fp32(state.range(0));
106
+ std::generate(fp16.begin(), fp16.end(),
107
+ [&rng]{ return fp16_alt_from_fp32_value(rng()); });
108
+
109
+ while (state.KeepRunning()) {
110
+ uint16_t* input = fp16.data();
111
+ benchmark::DoNotOptimize(input);
112
+
113
+ float* output = fp32.data();
114
+ const size_t n = state.range(0);
115
+ for (size_t i = 0; i < n; i += 8) {
116
+ const psimd_f32x2 data =
117
+ fp16_alt_to_fp32x2_psimd(
118
+ psimd_load_u16(&input[i]));
119
+ psimd_store_f32(&output[i], data.lo);
120
+ psimd_store_f32(&output[i + 4], data.hi);
121
+ }
122
+
123
+ benchmark::DoNotOptimize(output);
124
+ }
125
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
126
+ }
127
+ BENCHMARK(fp16_alt_to_fp32x2_psimd)->RangeMultiplier(2)->Range(1<<10, 64<<20);
128
+ #endif
129
+
130
+ #if defined(__ARM_NEON_FP) && (__ARM_NEON_FP & 0x2) || defined(__aarch64__)
131
+ static void hardware_vcvt_f32_f16(benchmark::State& state) {
132
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
133
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
134
+
135
+ std::vector<uint16_t> fp16(state.range(0));
136
+ std::vector<float> fp32(state.range(0));
137
+ std::generate(fp16.begin(), fp16.end(),
138
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
139
+
140
+ while (state.KeepRunning()) {
141
+ uint16_t* input = fp16.data();
142
+ benchmark::DoNotOptimize(input);
143
+
144
+ float* output = fp32.data();
145
+ const size_t n = state.range(0);
146
+ #if defined(__aarch64__)
147
+ const unsigned int fpcr = __builtin_aarch64_get_fpcr();
148
+ /* Disable flush-to-zero (bit 24) and enable Alternative FP16 format (bit 26) */
149
+ __builtin_aarch64_set_fpcr((fpcr & 0xFEFFFFFFu) | 0x08000000u);
150
+ #else
151
+ unsigned int fpscr;
152
+ __asm__ __volatile__ ("VMRS %[fpscr], fpscr" : [fpscr] "=r" (fpscr));
153
+ /* Disable flush-to-zero (bit 24) and enable Alternative FP16 format (bit 26) */
154
+ __asm__ __volatile__ ("VMSR fpscr, %[fpscr]" :
155
+ : [fpscr] "r" ((fpscr & 0xFEFFFFFFu) | 0x08000000u));
156
+ #endif
157
+ for (size_t i = 0; i < n; i += 4) {
158
+ vst1q_f32(&output[i],
159
+ vcvt_f32_f16(
160
+ (float16x4_t) vld1_u16(&input[i])));
161
+ }
162
+ #if defined(__aarch64__)
163
+ __builtin_aarch64_set_fpcr(fpcr);
164
+ #else
165
+ __asm__ __volatile__ ("VMSR fpscr, %[fpscr]" :: [fpscr] "r" (fpscr));
166
+ #endif
167
+
168
+ benchmark::DoNotOptimize(output);
169
+ }
170
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
171
+ }
172
+ BENCHMARK(hardware_vcvt_f32_f16)->RangeMultiplier(2)->Range(1<<10, 64<<20);
173
+ #endif
174
+
175
+ BENCHMARK_MAIN();
@@ -0,0 +1,399 @@
1
+ #include <benchmark/benchmark.h>
2
+
3
+ #include <fp16.h>
4
+ #ifndef EMSCRIPTEN
5
+ #include <fp16/psimd.h>
6
+ #endif
7
+
8
+ #include <vector>
9
+ #include <random>
10
+ #include <chrono>
11
+ #include <functional>
12
+ #include <algorithm>
13
+
14
+ #if (defined(__i386__) || defined(__x86_64__)) && defined(__F16C__)
15
+ #include <immintrin.h>
16
+ #endif
17
+
18
+ #if defined(__ARM_NEON__) || defined(__aarch64__)
19
+ #include <arm_neon.h>
20
+ #endif
21
+
22
+ #ifdef FP16_COMPARATIVE_BENCHMARKS
23
+ #include <third-party/THHalf.h>
24
+ #include <third-party/npy-halffloat.h>
25
+ #include <third-party/eigen-half.h>
26
+ #include <third-party/float16-compressor.h>
27
+ #include <third-party/half.hpp>
28
+ #endif
29
+
30
+
31
+ static void fp16_ieee_to_fp32_bits(benchmark::State& state) {
32
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
33
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
34
+
35
+ std::vector<uint16_t> fp16(state.range(0));
36
+ std::vector<uint32_t> fp32(state.range(0));
37
+ std::generate(fp16.begin(), fp16.end(),
38
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
39
+
40
+ while (state.KeepRunning()) {
41
+ uint16_t* input = fp16.data();
42
+ benchmark::DoNotOptimize(input);
43
+
44
+ uint32_t* output = fp32.data();
45
+ const size_t n = state.range(0);
46
+ for (size_t i = 0; i < n; i++) {
47
+ output[i] = fp16_ieee_to_fp32_bits(input[i]);
48
+ }
49
+
50
+ benchmark::DoNotOptimize(output);
51
+ }
52
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
53
+ }
54
+ BENCHMARK(fp16_ieee_to_fp32_bits)->RangeMultiplier(2)->Range(1<<10, 64<<20);
55
+
56
+ static void fp16_ieee_to_fp32_value(benchmark::State& state) {
57
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
58
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
59
+
60
+ std::vector<uint16_t> fp16(state.range(0));
61
+ std::vector<float> fp32(state.range(0));
62
+ std::generate(fp16.begin(), fp16.end(),
63
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
64
+
65
+ while (state.KeepRunning()) {
66
+ uint16_t* input = fp16.data();
67
+ benchmark::DoNotOptimize(input);
68
+
69
+ float* output = fp32.data();
70
+ const size_t n = state.range(0);
71
+ for (size_t i = 0; i < n; i++) {
72
+ output[i] = fp16_ieee_to_fp32_value(input[i]);
73
+ }
74
+
75
+ benchmark::DoNotOptimize(output);
76
+ }
77
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
78
+ }
79
+ BENCHMARK(fp16_ieee_to_fp32_value)->RangeMultiplier(2)->Range(1<<10, 64<<20);
80
+
81
+ #ifndef EMSCRIPTEN
82
+ static void fp16_ieee_to_fp32_psimd(benchmark::State& state) {
83
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
84
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
85
+
86
+ std::vector<uint16_t> fp16(state.range(0));
87
+ std::vector<float> fp32(state.range(0));
88
+ std::generate(fp16.begin(), fp16.end(),
89
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
90
+
91
+ while (state.KeepRunning()) {
92
+ uint16_t* input = fp16.data();
93
+ benchmark::DoNotOptimize(input);
94
+
95
+ float* output = fp32.data();
96
+ const size_t n = state.range(0);
97
+ for (size_t i = 0; i < n - 4; i += 4) {
98
+ psimd_store_f32(&output[i],
99
+ fp16_ieee_to_fp32_psimd(
100
+ psimd_load_u16(&input[i])));
101
+ }
102
+ const psimd_u16 last_vector = { input[n - 4], input[n - 3], input[n - 2], input[n - 1] };
103
+ psimd_store_f32(&output[n - 4],
104
+ fp16_ieee_to_fp32_psimd(last_vector));
105
+
106
+ benchmark::DoNotOptimize(output);
107
+ }
108
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
109
+ }
110
+ BENCHMARK(fp16_ieee_to_fp32_psimd)->RangeMultiplier(2)->Range(1<<10, 64<<20);
111
+
112
+ static void fp16_ieee_to_fp32x2_psimd(benchmark::State& state) {
113
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
114
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
115
+
116
+ std::vector<uint16_t> fp16(state.range(0));
117
+ std::vector<float> fp32(state.range(0));
118
+ std::generate(fp16.begin(), fp16.end(),
119
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
120
+
121
+ while (state.KeepRunning()) {
122
+ uint16_t* input = fp16.data();
123
+ benchmark::DoNotOptimize(input);
124
+
125
+ float* output = fp32.data();
126
+ const size_t n = state.range(0);
127
+ for (size_t i = 0; i < n; i += 8) {
128
+ const psimd_f32x2 data =
129
+ fp16_ieee_to_fp32x2_psimd(
130
+ psimd_load_u16(&input[i]));
131
+ psimd_store_f32(&output[i], data.lo);
132
+ psimd_store_f32(&output[i + 4], data.hi);
133
+ }
134
+
135
+ benchmark::DoNotOptimize(output);
136
+ }
137
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
138
+ }
139
+ BENCHMARK(fp16_ieee_to_fp32x2_psimd)->RangeMultiplier(2)->Range(1<<10, 64<<20);
140
+ #endif
141
+
142
+ #if (defined(__i386__) || defined(__x86_64__)) && defined(__F16C__)
143
+ static void hardware_mm_cvtph_ps(benchmark::State& state) {
144
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
145
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
146
+
147
+ std::vector<uint16_t> fp16(state.range(0));
148
+ std::vector<float> fp32(state.range(0));
149
+ std::generate(fp16.begin(), fp16.end(),
150
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
151
+
152
+ while (state.KeepRunning()) {
153
+ uint16_t* input = fp16.data();
154
+ benchmark::DoNotOptimize(input);
155
+
156
+ float* output = fp32.data();
157
+ const size_t n = state.range(0);
158
+ for (size_t i = 0; i < n; i += 4) {
159
+ _mm_storeu_ps(&output[i],
160
+ _mm_cvtph_ps(
161
+ _mm_loadl_epi64(static_cast<const __m128i*>(static_cast<const void*>(&input[i])))));
162
+ }
163
+
164
+ benchmark::DoNotOptimize(output);
165
+ }
166
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
167
+ }
168
+ BENCHMARK(hardware_mm_cvtph_ps)->RangeMultiplier(2)->Range(1<<10, 64<<20);
169
+
170
+ static void hardware_mm256_cvtph_ps(benchmark::State& state) {
171
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
172
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
173
+
174
+ std::vector<uint16_t> fp16(state.range(0));
175
+ std::vector<float> fp32(state.range(0));
176
+ std::generate(fp16.begin(), fp16.end(),
177
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
178
+
179
+ while (state.KeepRunning()) {
180
+ uint16_t* input = fp16.data();
181
+ benchmark::DoNotOptimize(input);
182
+
183
+ float* output = fp32.data();
184
+ const size_t n = state.range(0);
185
+ for (size_t i = 0; i < n; i += 8) {
186
+ _mm256_storeu_ps(&output[i],
187
+ _mm256_cvtph_ps(
188
+ _mm_loadu_si128(static_cast<const __m128i*>(static_cast<const void*>(&input[i])))));
189
+ }
190
+
191
+ benchmark::DoNotOptimize(output);
192
+ }
193
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
194
+ }
195
+ BENCHMARK(hardware_mm256_cvtph_ps)->RangeMultiplier(2)->Range(1<<10, 64<<20);
196
+ #endif
197
+
198
+ #if defined(__ARM_NEON_FP) && (__ARM_NEON_FP & 0x2) || defined(__aarch64__)
199
+ static void hardware_vcvt_f32_f16(benchmark::State& state) {
200
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
201
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
202
+
203
+ std::vector<uint16_t> fp16(state.range(0));
204
+ std::vector<float> fp32(state.range(0));
205
+ std::generate(fp16.begin(), fp16.end(),
206
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
207
+
208
+ while (state.KeepRunning()) {
209
+ uint16_t* input = fp16.data();
210
+ benchmark::DoNotOptimize(input);
211
+
212
+ float* output = fp32.data();
213
+ const size_t n = state.range(0);
214
+ #if defined(__aarch64__)
215
+ const unsigned int fpcr = __builtin_aarch64_get_fpcr();
216
+ /* Disable flush-to-zero (bit 24) and Alternative FP16 format (bit 26) */
217
+ __builtin_aarch64_set_fpcr(fpcr & 0xF6FFFFFFu);
218
+ #else
219
+ unsigned int fpscr;
220
+ __asm__ __volatile__ ("VMRS %[fpscr], fpscr" : [fpscr] "=r" (fpscr));
221
+ /* Disable flush-to-zero (bit 24) and Alternative FP16 format (bit 26) */
222
+ __asm__ __volatile__ ("VMSR fpscr, %[fpscr]" :
223
+ : [fpscr] "r" (fpscr & 0xF6FFFFFFu));
224
+ #endif
225
+ for (size_t i = 0; i < n; i += 4) {
226
+ vst1q_f32(&output[i],
227
+ vcvt_f32_f16(
228
+ (float16x4_t) vld1_u16(&input[i])));
229
+ }
230
+ #if defined(__aarch64__)
231
+ __builtin_aarch64_set_fpcr(fpcr);
232
+ #else
233
+ __asm__ __volatile__ ("VMSR fpscr, %[fpscr]" :: [fpscr] "r" (fpscr));
234
+ #endif
235
+
236
+ benchmark::DoNotOptimize(output);
237
+ }
238
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
239
+ }
240
+ BENCHMARK(hardware_vcvt_f32_f16)->RangeMultiplier(2)->Range(1<<10, 64<<20);
241
+ #endif
242
+
243
+ #ifdef FP16_COMPARATIVE_BENCHMARKS
244
+ static void TH_halfbits2float(benchmark::State& state) {
245
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
246
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
247
+
248
+ std::vector<uint16_t> fp16(state.range(0));
249
+ std::vector<float> fp32(state.range(0));
250
+ std::generate(fp16.begin(), fp16.end(),
251
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
252
+
253
+ while (state.KeepRunning()) {
254
+ uint16_t* input = fp16.data();
255
+ benchmark::DoNotOptimize(input);
256
+
257
+ float* output = fp32.data();
258
+ const size_t n = state.range(0);
259
+ for (size_t i = 0; i < n; i++) {
260
+ TH_halfbits2float(&input[i], &output[i]);
261
+ }
262
+
263
+ benchmark::DoNotOptimize(output);
264
+ }
265
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
266
+ }
267
+ BENCHMARK(TH_halfbits2float)->RangeMultiplier(2)->Range(1<<10, 64<<20);
268
+
269
+ static void npy_halfbits_to_floatbits(benchmark::State& state) {
270
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
271
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
272
+
273
+ std::vector<uint16_t> fp16(state.range(0));
274
+ std::vector<uint32_t> fp32(state.range(0));
275
+ std::generate(fp16.begin(), fp16.end(),
276
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
277
+
278
+ while (state.KeepRunning()) {
279
+ uint16_t* input = fp16.data();
280
+ benchmark::DoNotOptimize(input);
281
+
282
+ uint32_t* output = fp32.data();
283
+ const size_t n = state.range(0);
284
+ for (size_t i = 0; i < n; i++) {
285
+ output[i] = npy_halfbits_to_floatbits(input[i]);
286
+ }
287
+
288
+ benchmark::DoNotOptimize(output);
289
+ }
290
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
291
+ }
292
+ BENCHMARK(npy_halfbits_to_floatbits)->RangeMultiplier(2)->Range(1<<10, 64<<20);
293
+
294
+ static void Eigen_half_to_float(benchmark::State& state) {
295
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
296
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
297
+
298
+ std::vector<uint16_t> fp16(state.range(0));
299
+ std::vector<float> fp32(state.range(0));
300
+ std::generate(fp16.begin(), fp16.end(),
301
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
302
+
303
+ while (state.KeepRunning()) {
304
+ uint16_t* input = fp16.data();
305
+ benchmark::DoNotOptimize(input);
306
+
307
+ float* output = fp32.data();
308
+ const size_t n = state.range(0);
309
+ for (size_t i = 0; i < n; i++) {
310
+ output[i] =
311
+ Eigen::half_impl::half_to_float(
312
+ Eigen::half_impl::raw_uint16_to_half(input[i]));
313
+ }
314
+
315
+ benchmark::DoNotOptimize(output);
316
+ }
317
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
318
+ }
319
+ BENCHMARK(Eigen_half_to_float)->RangeMultiplier(2)->Range(1<<10, 64<<20);
320
+
321
+ static void Float16Compressor_decompress(benchmark::State& state) {
322
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
323
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
324
+
325
+ std::vector<uint16_t> fp16(state.range(0));
326
+ std::vector<float> fp32(state.range(0));
327
+ std::generate(fp16.begin(), fp16.end(),
328
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
329
+
330
+ while (state.KeepRunning()) {
331
+ uint16_t* input = fp16.data();
332
+ benchmark::DoNotOptimize(input);
333
+
334
+ float* output = fp32.data();
335
+ const size_t n = state.range(0);
336
+ for (size_t i = 0; i < n; i++) {
337
+ output[i] = Float16Compressor::decompress(input[i]);
338
+ }
339
+
340
+ benchmark::DoNotOptimize(output);
341
+ }
342
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
343
+ }
344
+ BENCHMARK(Float16Compressor_decompress)->RangeMultiplier(2)->Range(1<<10, 64<<20);
345
+
346
+ static void half_float_detail_half2float_table(benchmark::State& state) {
347
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
348
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
349
+
350
+ std::vector<uint16_t> fp16(state.range(0));
351
+ std::vector<float> fp32(state.range(0));
352
+ std::generate(fp16.begin(), fp16.end(),
353
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
354
+
355
+ while (state.KeepRunning()) {
356
+ uint16_t* input = fp16.data();
357
+ benchmark::DoNotOptimize(input);
358
+
359
+ float* output = fp32.data();
360
+ const size_t n = state.range(0);
361
+ for (size_t i = 0; i < n; i++) {
362
+ output[i] = half_float::detail::half2float_impl(input[i],
363
+ half_float::detail::true_type());
364
+ }
365
+
366
+ benchmark::DoNotOptimize(output);
367
+ }
368
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
369
+ }
370
+ BENCHMARK(half_float_detail_half2float_table)->RangeMultiplier(2)->Range(1<<10, 64<<20);
371
+
372
+ static void half_float_detail_half2float_branch(benchmark::State& state) {
373
+ const uint_fast32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
374
+ auto rng = std::bind(std::uniform_real_distribution<float>(-1.0f, 1.0f), std::mt19937(seed));
375
+
376
+ std::vector<uint16_t> fp16(state.range(0));
377
+ std::vector<float> fp32(state.range(0));
378
+ std::generate(fp16.begin(), fp16.end(),
379
+ [&rng]{ return fp16_ieee_from_fp32_value(rng()); });
380
+
381
+ while (state.KeepRunning()) {
382
+ uint16_t* input = fp16.data();
383
+ benchmark::DoNotOptimize(input);
384
+
385
+ float* output = fp32.data();
386
+ const size_t n = state.range(0);
387
+ for (size_t i = 0; i < n; i++) {
388
+ output[i] = half_float::detail::half2float_impl(input[i],
389
+ half_float::detail::false_type());
390
+ }
391
+
392
+ benchmark::DoNotOptimize(output);
393
+ }
394
+ state.SetItemsProcessed(int64_t(state.iterations()) * int64_t(state.range(0)));
395
+ }
396
+ BENCHMARK(half_float_detail_half2float_branch)->RangeMultiplier(2)->Range(1<<10, 64<<20);
397
+ #endif
398
+
399
+ BENCHMARK_MAIN();