saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Updated permalinks for the reference code used.

Browse source 
Signed-off-by: maajidkhann <maajidkhan.n@fujitsu.com>
This commit is contained in:
maajidkhann 2025-02-05 10:29:15 +05:30 committed by PyTorch MergeBot
parent da6e22a216
commit 79615a3d2a
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aten/src/ATen/cpu/vec/sve/vec_float.h
View file

@ -85,6 +85,7 @@ public:
}
return b;
}
//Implementation is picked from https://github.com/ARM-software/ComputeLibrary/blob/v25.01/src/core/NEON/SVEMath.inl#L105
inline svfloat32_t svexp_f32_z(svbool_t pg, svfloat32_t x) const {
const auto c1 = svreinterpret_f32_u32(svdup_n_u32(0x3f7ffff6)); // x^1: 0x1.ffffecp-1f
const auto c2 = svreinterpret_f32_u32(svdup_n_u32(0x3efffedb)); // x^2: 0x1.fffdb6p-2f
@ -384,7 +385,7 @@ public:
Vectorized<float> tan() const {
return USE_SLEEF(Vectorized<float>(Sleef_tanfx_u10sve(values)),map(std::tan));
}
//Implementation is picked from https://github.com/ARM-software/ComputeLibrary/blob/main/src/core/NEON/SVEMath.inl
//Implementation is picked from https://github.com/ARM-software/ComputeLibrary/blob/v25.01/src/core/NEON/SVEMath.inl#L179
Vectorized<float> tanh() const {
// Constants used for the tanh calculation.
const svfloat32_t CONST_1 = svdup_n_f32(1.f); // Constant 1.0f for the tanh formula.