Nektar++
VDmath.hpp
Go to the documentation of this file.
1 ///////////////////////////////////////////////////////////////////////////////
2 //
3 // File DVmath.hpp
4 //
5 // For more information, please see: http://www.nektar.info
6 //
7 // The MIT License
8 //
9 // Copyright (c) 2006 Division of Applied Mathematics, Brown University (USA),
10 // Department of Aeronautics, Imperial College London (UK), and Scientific
11 // Computing and Imaging Institute, University of Utah (USA).
12 //
13 // License for the specific language governing rights and limitations under
14 // Permission is hereby granted, free of charge, to any person obtaining a
15 // copy of this software and associated documentation files (the "Software"),
16 // to deal in the Software without restriction, including without limitation
17 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
18 // and/or sell copies of the Software, and to permit persons to whom the
19 // Software is furnished to do so, subject to the following conditions:
20 //
21 // The above copyright notice and this permission notice shall be included
22 // in all copies or substantial portions of the Software.
23 //
24 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
25 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
26 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
27 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
28 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
29 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
30 // DEALINGS IN THE SOFTWARE.
31 //
32 // Description: Collection of templated functions for distributed vector
33 // mathematics
34 //
35 ///////////////////////////////////////////////////////////////////////////////
36 
37 #ifndef NEKTAR_LIB_LIBUTILITIES_BASSICUTILS_VECTORDISTMATH_HPP
38 #define NEKTAR_LIB_LIBUTILITIES_BASSICUTILS_VECTORDISTMATH_HPP
39 
41 
42 namespace VDmath
43 {
44  /// \brief vvtvp (vector times vector times vector): z = w*x*y
45  template<class T> T Ddot2( Nektar::LibUtilities::CommSharedPtr& pComm,
46  int n,
47  const T *w,
48  const T *x,
49  const int *y)
50  {
51  T sum = 0;
52 
53  while( n-- )
54  {
55  sum += (*y == 1 ? (*w) * (*x) : 0 );
56  ++w;
57  ++x;
58  ++y;
59  }
60  pComm->AllReduce(sum, Nektar::LibUtilities::ReduceSum);
61  return sum;
62  }
63 
64  /// \brief vvtvp (vector times vector times vector): z = w*x*y
65  template<class T> T Ddot2( Nektar::LibUtilities::CommSharedPtr& pComm,
66  int n,
67  const T *w, const int incw,
68  const T *x, const int incx,
69  const int *y, const int incy)
70  {
71  T sum = 0;
72 
73  while( n-- )
74  {
75  sum += (*y == 1 ? (*w) * (*x) : 0.0 );
76  w += incw;
77  x += incx;
78  y += incy;
79  }
80  pComm->AllReduce(sum, Nektar::LibUtilities::ReduceSum);
81  return sum;
82  }
83 }
84 #endif /* DVMATH_HPP_ */
boost::shared_ptr< Comm > CommSharedPtr
Pointer to a Communicator object.
Definition: Comm.h:53
T Ddot2(Nektar::LibUtilities::CommSharedPtr &pComm, int n, const T *w, const T *x, const int *y)
vvtvp (vector times vector times vector): z = w*x*y
Definition: VDmath.hpp:45