3e6555c967c3c695108eac8a11930c0c1f091132
[alexxy/gromacs.git] / src / gromacs / simd / tests / simd4_math.cpp
1 /*
2  * This file is part of the GROMACS molecular simulation package.
3  *
4  * Copyright (c) 2014, by the GROMACS development team, led by
5  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
6  * and including many others, as listed in the AUTHORS file in the
7  * top-level source directory and at http://www.gromacs.org.
8  *
9  * GROMACS is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public License
11  * as published by the Free Software Foundation; either version 2.1
12  * of the License, or (at your option) any later version.
13  *
14  * GROMACS is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with GROMACS; if not, see
21  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
22  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
23  *
24  * If you want to redistribute modifications to GROMACS, please
25  * consider that scientific software is very special. Version
26  * control is crucial - bugs must be traceable. We will be happy to
27  * consider code for inclusion in the official distribution, but
28  * derived work must not be called official GROMACS. Details are found
29  * in the README & COPYING files - if they are missing, get the
30  * official version at http://www.gromacs.org.
31  *
32  * To help us fund GROMACS development, we humbly ask that you cite
33  * the research papers on the package. Check out http://www.gromacs.org.
34  */
35 #include "gmxpre.h"
36
37 #include "config.h"
38
39 #include <vector>
40 #include "gromacs/math/utilities.h"
41 #include "gromacs/simd/simd.h"
42 #include "gromacs/simd/simd_math.h"
43 #include "gromacs/options/basicoptions.h"
44
45 #include "simd4.h"
46
47 namespace gmx
48 {
49 namespace test
50 {
51
52 /*! \cond internal */
53 /*! \addtogroup module_simd */
54 /*! \{ */
55
56 #ifdef GMX_SIMD4_HAVE_REAL
57
58 class Simd4MathTest : public Simd4Test
59 {
60     public:
61         ::testing::AssertionResult
62                              compareSimd4MathFunction(const char * refFuncExpr, const char *simd4FuncExpr,
63                                                       real refFunc(real x),     gmx_simd4_real_t gmx_simdcall simd4Func(gmx_simd4_real_t x));
64 };
65
66 /*! \brief Test approximate equality of SIMD4 vs reference version of a function.
67  *
68  * This macro takes vanilla C and SIMD flavors of a function and tests it with
69  * the number of points, range, and tolerances specified by the test fixture class.
70  */
71 #define GMX_EXPECT_SIMD4_FUNC_NEAR(refFunc, tstFunc) \
72     EXPECT_PRED_FORMAT2(compareSimd4MathFunction, refFunc, tstFunc)
73
74
75 /*! \brief Implementation routine to compare SIMD4 vs reference functions.
76  *
77  * \param refFuncExpr    Description of reference function expression
78  * \param simd4FuncExpr  Description of SIMD function expression
79  * \param refFunc        Reference math function pointer
80  * \param simd4Func      SIMD math function pointer
81  *
82  * The function will be tested with the range and tolerances specified in
83  * the SimdBaseTest class. You should not never call this function directly,
84  * but use the macro GMX_EXPECT_SIMD4_FUNC_NEAR(refFunc,tstFunc) instead.
85  */
86 ::testing::AssertionResult
87 Simd4MathTest::compareSimd4MathFunction(const char * refFuncExpr, const char *simd4FuncExpr,
88                                         real refFunc(real x),     gmx_simd4_real_t gmx_simdcall simd4Func(gmx_simd4_real_t x))
89 {
90     std::vector<real>            vx(GMX_SIMD4_WIDTH);
91     std::vector<real>            vref(GMX_SIMD4_WIDTH);
92     std::vector<real>            vtst(GMX_SIMD4_WIDTH);
93     real                         dx;
94     gmx_int64_t                  ulpDiff, maxUlpDiff;
95     real                         maxUlpDiffPos;
96     real                         refValMaxUlpDiff, simdValMaxUlpDiff;
97     bool                         eq, signOk;
98     int                          i, iter;
99     int                          niter   = s_nPoints/GMX_SIMD4_WIDTH;
100     int                          npoints = niter*GMX_SIMD4_WIDTH;
101 #    ifdef GMX_DOUBLE
102     union {
103         double r; gmx_int64_t i;
104     } conv0, conv1;
105 #    else
106     union {
107         float  r; gmx_int32_t i;
108     } conv0, conv1;
109 #    endif
110
111     maxUlpDiff = 0;
112     dx         = (range_.second-range_.first)/npoints;
113
114     for (iter = 0; iter < niter; iter++)
115     {
116         for (i = 0; i < GMX_SIMD4_WIDTH; i++)
117         {
118             vx[i]   = range_.first+dx*(iter*GMX_SIMD4_WIDTH+i);
119             vref[i] = refFunc(vx[i]);
120         }
121         vtst  = simd4Real2Vector(simd4Func(vector2Simd4Real(vx)));
122
123         for (i = 0, eq = true, signOk = true; i < GMX_SIMD4_WIDTH && eq == true; i++)
124         {
125             eq     = eq && ( fabs(vref[i]-vtst[i]) < absTol_ );
126             signOk = signOk && ( vref[i]*vtst[i] >= 0 );
127         }
128         if (eq == true)
129         {
130             // Go to next point if everything within absolute tolerance
131             continue;
132         }
133         else if (signOk == false)
134         {
135             return ::testing::AssertionFailure()
136                    << "Failing SIMD4 math function comparison due to sign differences." << std::endl
137                    << "Reference function: " << refFuncExpr << std::endl
138                    << "Simd function:      " << simd4FuncExpr << std::endl
139                    << "Test range is ( " << range_.first << " , " << range_.second << " ) " << std::endl
140                    << "First sign difference around x=" << std::setprecision(20) << ::testing::PrintToString(vx) << std::endl
141                    << "Ref values:   " << std::setprecision(20) << ::testing::PrintToString(vref) << std::endl
142                    << "SIMD4 values: " << std::setprecision(20) << ::testing::PrintToString(vtst) << std::endl;
143         }
144         /* We replicate the trivial ulp differences comparison here rather than
145          * calling the lower-level routine for comparing them, since this enables
146          * us to run through the entire test range and report the largest deviation
147          * without lots of extra glue routines.
148          */
149         for (i = 0; i < GMX_SIMD4_WIDTH; i++)
150         {
151             conv0.r = vref[i];
152             conv1.r = vtst[i];
153             ulpDiff = llabs(conv0.i-conv1.i);
154             if (ulpDiff > maxUlpDiff)
155             {
156                 maxUlpDiff        = ulpDiff;
157                 maxUlpDiffPos     = vx[i];
158                 refValMaxUlpDiff  = vref[i];
159                 simdValMaxUlpDiff = vtst[i];
160             }
161         }
162     }
163
164     if (maxUlpDiff <= ulpTol_)
165     {
166         return ::testing::AssertionSuccess();
167     }
168     else
169     {
170         return ::testing::AssertionFailure()
171                << "Failing SIMD4 math function ulp comparison between " << refFuncExpr << " and " << simd4FuncExpr << std::endl
172                << "Requested ulp tolerance: " << ulpTol_ << std::endl
173                << "Requested abs tolerance: " << absTol_ << std::endl
174                << "Largest Ulp difference occurs for x=" << std::setprecision(20) << maxUlpDiffPos << std::endl
175                << "Ref  values:  " << std::setprecision(20) << refValMaxUlpDiff << std::endl
176                << "SIMD4 values: " << std::setprecision(20) << simdValMaxUlpDiff << std::endl
177                << "Ulp diff.:   " << std::setprecision(20) << maxUlpDiff << std::endl;
178     }
179 }
180
181 /*! \} */
182 /*! \endcond */
183
184 // Actual math function tests below
185
186 namespace
187 {
188
189 /*! \cond internal */
190 /*! \addtogroup module_simd */
191 /*! \{ */
192
193 /*! \brief Function wrapper to evaluate reference 1/sqrt(x) */
194 static real
195 ref_invsqrt(real x)
196 {
197     return 1.0/sqrt(x);
198 }
199
200 TEST_F(Simd4MathTest, gmxSimd4InvsqrtR)
201 {
202     setRange(1e-10, 1e10);
203     GMX_EXPECT_SIMD4_FUNC_NEAR(ref_invsqrt, gmx_simd4_invsqrt_r);
204 }
205
206 }      // namespace
207
208 #endif // GMX_SIMD4_HAVE_REAL
209
210 /*! \} */
211 /*! \endcond */
212
213 }      // namespace
214 }      // namespace