Kokkos Node API and Local Linear Algebra Kernels Version of the Day
Tsqr_MpiDatatype.hpp
00001 /*
00002 //@HEADER
00003 // ************************************************************************
00004 // 
00005 //          Kokkos: Node API and Parallel Node Kernels
00006 //              Copyright (2008) Sandia Corporation
00007 // 
00008 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
00009 // the U.S. Government retains certain rights in this software.
00010 // 
00011 // Redistribution and use in source and binary forms, with or without
00012 // modification, are permitted provided that the following conditions are
00013 // met:
00014 //
00015 // 1. Redistributions of source code must retain the above copyright
00016 // notice, this list of conditions and the following disclaimer.
00017 //
00018 // 2. Redistributions in binary form must reproduce the above copyright
00019 // notice, this list of conditions and the following disclaimer in the
00020 // documentation and/or other materials provided with the distribution.
00021 //
00022 // 3. Neither the name of the Corporation nor the names of the
00023 // contributors may be used to endorse or promote products derived from
00024 // this software without specific prior written permission.
00025 //
00026 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
00027 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00028 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00029 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
00030 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00031 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00032 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00033 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00034 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00035 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00036 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00037 //
00038 // Questions? Contact Michael A. Heroux (maherou@sandia.gov) 
00039 // 
00040 // ************************************************************************
00041 //@HEADER
00042 */
00043 
00044 #ifndef __TSQR_MpiDatatype_hpp
00045 #define __TSQR_MpiDatatype_hpp
00046 
00047 #include <Teuchos_ConfigDefs.hpp> // HAVE_MPI
00048 
00049 #ifdef HAVE_MPI
00050 #  include <mpi.h>
00051 #  include <stdexcept>
00052 
00055 
00056 namespace TSQR {
00057   namespace MPI {
00058 
00061     MPI_Datatype 
00062     cloneRawDatatype (MPI_Datatype in, const bool needsFree);
00063 
00072     template< class Datum >
00073     class MpiDatatype {
00074     public:
00077       MpiDatatype ();
00078 
00081       MpiDatatype (const MpiDatatype& rhs) 
00082       {
00083   clone (this->type_, this->needsFree_, rhs);
00084       }
00085 
00088       MpiDatatype& operator= (const MpiDatatype& rhs)
00089       {
00090   if (this != &rhs)
00091     {
00092       if (needsFree_)
00093         {
00094     // Return value doesn't matter...
00095     (void) MPI_Type_free (&type_);
00096     needsFree_ = false;
00097         }
00098       clone (this->type_, this->needsFree_, rhs);
00099     }
00100   return *this;
00101       }
00102 
00105       ~MpiDatatype () 
00106       { 
00107   if (needsFree_)
00108     {
00109       // Return value doesn't matter...
00110       (void) MPI_Type_free (&type_);
00111       needsFree_ = false;
00112     }
00113       }
00114 
00115       MPI_Datatype get() const { return type_; }
00116 
00117     private:
00118       static void 
00119       clone (MPI_Datatype& newType, 
00120        bool& needsFree,
00121        const MpiDatatype& rhs)
00122       {
00123   newType = cloneRawDatatype (rhs.get(), rhs.needsFree_);
00124   needsFree = rhs.needsFree_;
00125       }
00126 
00129       MPI_Datatype type_;
00130 
00133       bool needsFree_;
00134     };
00135 
00136   } // namespace MPI
00137 } // namespace TSQR
00138 
00139 
00140 #endif // HAVE_MPI
00141 #endif // __TSQR_MpiDatatype_hpp
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends