mirror of
https://github.com/aicodix/code.git
synced 2026-04-27 14:30:36 +00:00
added vclamp()
This commit is contained in:
parent
09945a18b5
commit
e51425e121
4 changed files with 174 additions and 0 deletions
54
simd.hh
54
simd.hh
|
|
@ -1010,6 +1010,60 @@ static inline SIMD<int64_t, WIDTH> vmax(SIMD<int64_t, WIDTH> a, SIMD<int64_t, WI
|
|||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<float, WIDTH> vclamp(SIMD<float, WIDTH> x, float a, float b)
|
||||
{
|
||||
SIMD<float, WIDTH> tmp;
|
||||
for (int i = 0; i < WIDTH; ++i)
|
||||
tmp.v[i] = std::min(std::max(x.v[i], a), b);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<double, WIDTH> vclamp(SIMD<double, WIDTH> x, double a, double b)
|
||||
{
|
||||
SIMD<double, WIDTH> tmp;
|
||||
for (int i = 0; i < WIDTH; ++i)
|
||||
tmp.v[i] = std::min(std::max(x.v[i], a), b);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<int8_t, WIDTH> vclamp(SIMD<int8_t, WIDTH> x, int8_t a, int8_t b)
|
||||
{
|
||||
SIMD<int8_t, WIDTH> tmp;
|
||||
for (int i = 0; i < WIDTH; ++i)
|
||||
tmp.v[i] = std::min(std::max(x.v[i], a), b);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<int16_t, WIDTH> vclamp(SIMD<int16_t, WIDTH> x, int16_t a, int16_t b)
|
||||
{
|
||||
SIMD<int16_t, WIDTH> tmp;
|
||||
for (int i = 0; i < WIDTH; ++i)
|
||||
tmp.v[i] = std::min(std::max(x.v[i], a), b);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<int32_t, WIDTH> vclamp(SIMD<int32_t, WIDTH> x, int32_t a, int32_t b)
|
||||
{
|
||||
SIMD<int32_t, WIDTH> tmp;
|
||||
for (int i = 0; i < WIDTH; ++i)
|
||||
tmp.v[i] = std::min(std::max(x.v[i], a), b);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<int64_t, WIDTH> vclamp(SIMD<int64_t, WIDTH> x, int64_t a, int64_t b)
|
||||
{
|
||||
SIMD<int64_t, WIDTH> tmp;
|
||||
for (int i = 0; i < WIDTH; ++i)
|
||||
tmp.v[i] = std::min(std::max(x.v[i], a), b);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
template <int WIDTH>
|
||||
static inline SIMD<float, WIDTH> vadd(SIMD<float, WIDTH> a, SIMD<float, WIDTH> b)
|
||||
{
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue