MirrorHub/construct
0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-09-26 02:18:53 +02:00
Code Issues Releases Wiki Activity

ircd::math: Add simd template wrapping for sqrt().

Browse source
This commit is contained in:
Jason Volk 2021-04-17 12:42:57 -07:00
parent b76d1f32ee
commit b6e2876af4
2 changed files with 68 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
include/ircd/math/math.h
View file

@ -20,3 +20,4 @@
#include "fmma.h"
#include "norm.h"
#include "smax.h"
#include "sqrt.h"

67
include/ircd/math/sqrt.h Normal file
View file

@ -0,0 +1,67 @@
// Matrix Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2021 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
#pragma once
#define HAVE_IRCD_MATH_SQRT_H
namespace ircd::math
{
template<class T>
typename std::enable_if<simd::is<T>(), T>::type
sqrt(T);
template<class T>
typename std::enable_if<simd::is<T>(), T>::type
sqrtf(T);
template<class T>
typename std::enable_if<simd::is<T>(), T>::type
sqrtl(T);
}
namespace ircd
{
using ::sqrt;
using ::sqrtf;
using ::sqrtl;
using math::sqrt;
using math::sqrtf;
using math::sqrtl;
}
template<class T>
inline typename std::enable_if<ircd::simd::is<T>(), T>::type
ircd::math::sqrtl(T a)
{
for(uint i(0); i < simd::lanes<T>(); ++i)
a[i] = ::sqrtl(a[i]);
return a;
}
template<class T>
inline typename std::enable_if<ircd::simd::is<T>(), T>::type
ircd::math::sqrtf(T a)
{
for(uint i(0); i < simd::lanes<T>(); ++i)
a[i] = ::sqrtf(a[i]);
return a;
}
template<class T>
inline typename std::enable_if<ircd::simd::is<T>(), T>::type
ircd::math::sqrt(T a)
{
for(uint i(0); i < simd::lanes<T>(); ++i)
a[i] = std::sqrt(a[i]);
return a;
}