From e993ee077b50d3a6134309d465a4174b5c749596 Mon Sep 17 00:00:00 2001
From: Cedric Nugteren <web@cedricnugteren.nl>
Date: Sat, 4 Mar 2017 15:21:33 +0100
Subject: Added a proper data-preparation function for the TRSM tests

---
 src/utilities/utilities.hpp | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'src/utilities/utilities.hpp')

diff --git a/src/utilities/utilities.hpp b/src/utilities/utilities.hpp
index 044955ea..3c9be6a2 100644
--- a/src/utilities/utilities.hpp
+++ b/src/utilities/utilities.hpp
@@ -98,6 +98,13 @@ constexpr auto kArgNoAbbreviations = "no_abbrv";
 
 // =================================================================================================
 
+// Converts a regular or complex type to it's base type (e.g. float2 to float)
+template <typename T> struct BaseType { using Type = T; };
+template <> struct BaseType<float2> { using Type = float; };
+template <> struct BaseType<double2> { using Type = double; };
+
+// =================================================================================================
+
 // Returns a scalar with a default value
 template <typename T> T GetScalar();
 
@@ -105,11 +112,11 @@ template <typename T> T GetScalar();
 template <typename T> T ConstantZero();
 template <typename T> T ConstantOne();
 template <typename T> T ConstantNegOne();
-template <typename T> T ConstantTwo();
+template <typename T> T Constant(const double val);
 template <typename T> T SmallConstant();
 
-// Returns the absolute value of a scalar
-template <typename T> T AbsoluteValue(const T value);
+// Returns the absolute value of a scalar (modulus in case of complex numbers)
+template <typename T> typename BaseType<T>::Type AbsoluteValue(const T value);
 
 // Returns whether a scalar is close to zero
 template <typename T> bool IsCloseToZero(const T value);
-- 
cgit v1.2.3