nagai/fune
3
0
Fork 0
forked from mirrors/gecko-dev
Code Pull requests Activity

Bug 1891990 - Upgrade gemmology to match xsimd version upgrade r=padenot

Browse source 
Differential Revision: https://phabricator.services.mozilla.com/D207723
This commit is contained in:
serge-sans-paille 2024-04-20 16:38:48 +00:00
parent e02cc14f5a
commit b5a6a756cb
2 changed files with 5 additions and 8 deletions
repo.diff.show_diff_stats Download patch file Download diff file Expand all files Collapse all files

9
third_party/gemmology/gemmology.h vendored
View file

@ -786,7 +786,6 @@ public:
using batch8 = xsimd::batch<int8_t, Arch>; using batch8 = xsimd::batch<int8_t, Arch>;
using batch16 = xsimd::batch<int16_t, Arch>; using batch16 = xsimd::batch<int16_t, Arch>;
using batch32 = xsimd::batch<int32_t, Arch>; using batch32 = xsimd::batch<int32_t, Arch>;
using ubatch32 = xsimd::batch<uint32_t, Arch>;
// Put higher rows in the second half of the register. These will jumble // Put higher rows in the second half of the register. These will jumble
// around in the same way then conveniently land in the right place. // around in the same way then conveniently land in the right place.
@ -814,7 +813,7 @@ public:
return xsimd::bitwise_cast<int8_t>( return xsimd::bitwise_cast<int8_t>(
xsimd::swizzle(xsimd::bitwise_cast<int32_t>(packed), xsimd::swizzle(xsimd::bitwise_cast<int32_t>(packed),
xsimd::make_batch_constant<ubatch32, Tiler<Arch>>())); xsimd::make_batch_constant<uint32_t, Arch, Tiler<Arch>>()));
} else if constexpr (batchf32::size == 8) } else if constexpr (batchf32::size == 8)
return Tile(quant_mult, input, input + 2 * cols, input + 16 * cols, return Tile(quant_mult, input, input + 2 * cols, input + 16 * cols,
input + 18 * cols); input + 18 * cols);
@ -833,7 +832,6 @@ public:
using batch8 = xsimd::batch<int8_t, Arch>; using batch8 = xsimd::batch<int8_t, Arch>;
using batch16 = xsimd::batch<int16_t, Arch>; using batch16 = xsimd::batch<int16_t, Arch>;
using batch32 = xsimd::batch<int32_t, Arch>; using batch32 = xsimd::batch<int32_t, Arch>;
using ubatch32 = xsimd::batch<uint32_t, Arch>;
const batch8 neg127(-127); const batch8 neg127(-127);
// Grab 4 registers at a time in 32-bit format. // Grab 4 registers at a time in 32-bit format.
@ -857,7 +855,7 @@ public:
// and the values are only used for GEMM. // and the values are only used for GEMM.
return xsimd::bitwise_cast<int8_t>( return xsimd::bitwise_cast<int8_t>(
xsimd::swizzle(xsimd::bitwise_cast<int32_t>(packed), xsimd::swizzle(xsimd::bitwise_cast<int32_t>(packed),
xsimd::make_batch_constant<ubatch32, Tiler<Arch>>())); xsimd::make_batch_constant<uint32_t, Arch, Tiler<Arch>>()));
} }
private: private:
@ -869,7 +867,6 @@ private:
using batch8 = xsimd::batch<int8_t, Arch>; using batch8 = xsimd::batch<int8_t, Arch>;
using batch16 = xsimd::batch<int16_t, Arch>; using batch16 = xsimd::batch<int16_t, Arch>;
using batch32 = xsimd::batch<int32_t, Arch>; using batch32 = xsimd::batch<int32_t, Arch>;
using ubatch32 = xsimd::batch<uint32_t, Arch>;
const batch8 neg127 = -127; const batch8 neg127 = -127;
const batch8 pos127 = +127; const batch8 pos127 = +127;
@ -894,7 +891,7 @@ private:
// and the values are only used for GEMM. // and the values are only used for GEMM.
return xsimd::bitwise_cast<uint8_t>( return xsimd::bitwise_cast<uint8_t>(
xsimd::swizzle(xsimd::bitwise_cast<int32_t>(packed), xsimd::swizzle(xsimd::bitwise_cast<int32_t>(packed),
xsimd::make_batch_constant<ubatch32, Tiler<Arch>>())); xsimd::make_batch_constant<uint32_t, Arch, Tiler<Arch>>()));
} }
}; };

4
third_party/gemmology/moz.yaml vendored
View file

@ -10,8 +10,8 @@ origin:
url: https://github.com/mozilla/gemmology url: https://github.com/mozilla/gemmology
release: dbcd029c3bc6e183355ea597216d379677ff9b19 (2024-02-20T12:36:14Z). release: a37d922a88ded67a0b741f16390821000fde99ee (2024-04-17T11:18:45Z).
revision: dbcd029c3bc6e183355ea597216d379677ff9b19 revision: a37d922a88ded67a0b741f16390821000fde99ee
license: MIT license: MIT