Changed vtkm::Abs() to return input type
This commit is contained in:
parent
71b2348dd9
commit
f1300c6143
56
vtkm/Math.h
56
vtkm/Math.h
@ -2570,6 +2570,22 @@ vtkm::Int64 Abs(vtkm::Int64 x)
|
||||
#error Unknown size of Int64.
|
||||
#endif
|
||||
}
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Float32 Abs(vtkm::Float32 x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_32(fabs)(x);
|
||||
#else
|
||||
return std::fabs(x);
|
||||
#endif
|
||||
}
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Float64 Abs(vtkm::Float64 x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_64(fabs)(x);
|
||||
#else
|
||||
return std::fabs(x);
|
||||
#endif
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
typename detail::FloatingPointReturnType<T>::Type
|
||||
@ -2580,31 +2596,10 @@ Abs(T x) {
|
||||
return std::fabs(static_cast<vtkm::Float64>(x));
|
||||
#endif
|
||||
}
|
||||
template<>
|
||||
inline VTKM_EXEC_CONT
|
||||
detail::FloatingPointReturnType<vtkm::Float32>::Type
|
||||
Abs(vtkm::Float32 x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_32(fabs)(x);
|
||||
#else
|
||||
return std::fabs(x);
|
||||
#endif
|
||||
}
|
||||
template<>
|
||||
inline VTKM_EXEC_CONT
|
||||
detail::FloatingPointReturnType<vtkm::Float64>::Type
|
||||
Abs(vtkm::Float64 x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_64(fabs)(x);
|
||||
#else
|
||||
return std::fabs(x);
|
||||
#endif
|
||||
}
|
||||
template<typename T, vtkm::IdComponent N>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,N>
|
||||
Abs(const vtkm::Vec<T,N> &x) {
|
||||
vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,N> result;
|
||||
vtkm::Vec<T,N> Abs(const vtkm::Vec<T,N> &x) {
|
||||
vtkm::Vec<T,N> result;
|
||||
for (vtkm::IdComponent index = 0; index < N; index++)
|
||||
{
|
||||
result[index] = vtkm::Abs(x[index]);
|
||||
@ -2613,26 +2608,23 @@ Abs(const vtkm::Vec<T,N> &x) {
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,4>
|
||||
Abs(const vtkm::Vec<T,4> &x) {
|
||||
return vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,4>(vtkm::Abs(x[0]),
|
||||
vtkm::Vec<T,4> Abs(const vtkm::Vec<T,4> &x) {
|
||||
return vtkm::Vec<T,4>(vtkm::Abs(x[0]),
|
||||
vtkm::Abs(x[1]),
|
||||
vtkm::Abs(x[2]),
|
||||
vtkm::Abs(x[3]));
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,3>
|
||||
Abs(const vtkm::Vec<T,3> &x) {
|
||||
return vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,3>(vtkm::Abs(x[0]),
|
||||
vtkm::Vec<T,3> Abs(const vtkm::Vec<T,3> &x) {
|
||||
return vtkm::Vec<T,3>(vtkm::Abs(x[0]),
|
||||
vtkm::Abs(x[1]),
|
||||
vtkm::Abs(x[2]));
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,2>
|
||||
Abs(const vtkm::Vec<T,2> &x) {
|
||||
return vtkm::Vec<typename detail::FloatingPointReturnType<T>::Type,2>(vtkm::Abs(x[0]),
|
||||
vtkm::Vec<T,2> Abs(const vtkm::Vec<T,2> &x) {
|
||||
return vtkm::Vec<T,2>(vtkm::Abs(x[0]),
|
||||
vtkm::Abs(x[1]));
|
||||
}
|
||||
|
||||
|
@ -994,7 +994,63 @@ vtkm::Int64 Abs(vtkm::Int64 x)
|
||||
#error Unknown size of Int64.
|
||||
#endif
|
||||
}
|
||||
$unary_math_function('Abs', 'fabs')\
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Float32 Abs(vtkm::Float32 x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_32(fabs)(x);
|
||||
#else
|
||||
return std::fabs(x);
|
||||
#endif
|
||||
}
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Float64 Abs(vtkm::Float64 x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_64(fabs)(x);
|
||||
#else
|
||||
return std::fabs(x);
|
||||
#endif
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
typename detail::FloatingPointReturnType<T>::Type
|
||||
Abs(T x) {
|
||||
#ifdef VTKM_CUDA
|
||||
return VTKM_CUDA_MATH_FUNCTION_64(fabs)(static_cast<vtkm::Float64>(x));
|
||||
#else
|
||||
return std::fabs(static_cast<vtkm::Float64>(x));
|
||||
#endif
|
||||
}
|
||||
template<typename T, vtkm::IdComponent N>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<T,N> Abs(const vtkm::Vec<T,N> &x) {
|
||||
vtkm::Vec<T,N> result;
|
||||
for (vtkm::IdComponent index = 0; index < N; index++)
|
||||
{
|
||||
result[index] = vtkm::Abs(x[index]);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<T,4> Abs(const vtkm::Vec<T,4> &x) {
|
||||
return vtkm::Vec<T,4>(vtkm::Abs(x[0]),
|
||||
vtkm::Abs(x[1]),
|
||||
vtkm::Abs(x[2]),
|
||||
vtkm::Abs(x[3]));
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<T,3> Abs(const vtkm::Vec<T,3> &x) {
|
||||
return vtkm::Vec<T,3>(vtkm::Abs(x[0]),
|
||||
vtkm::Abs(x[1]),
|
||||
vtkm::Abs(x[2]));
|
||||
}
|
||||
template<typename T>
|
||||
static inline VTKM_EXEC_CONT
|
||||
vtkm::Vec<T,2> Abs(const vtkm::Vec<T,2> &x) {
|
||||
return vtkm::Vec<T,2>(vtkm::Abs(x[0]),
|
||||
vtkm::Abs(x[1]));
|
||||
}
|
||||
|
||||
/// Returns a nonzero value if \p x is negative.
|
||||
///
|
||||
|
@ -86,7 +86,7 @@ namespace detail {
|
||||
template<typename T>
|
||||
VTKM_EXEC_CONT
|
||||
typename detail::FloatingPointReturnType<T>::Type
|
||||
MagnitudeTemplate(T &x, vtkm::TypeTraitsScalarTag)
|
||||
MagnitudeTemplate(T x, vtkm::TypeTraitsScalarTag)
|
||||
{
|
||||
return vtkm::Abs(x);
|
||||
}
|
||||
@ -125,7 +125,7 @@ VTKM_EXEC_CONT
|
||||
typename detail::FloatingPointReturnType<T>::Type
|
||||
RMagnitudeTemplate(T x, vtkm::TypeTraitsScalarTag)
|
||||
{
|
||||
return 1.0/vtkm::Abs(x);
|
||||
return T(1)/vtkm::Abs(x);
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
|
Loading…
Reference in New Issue
Block a user