|
Tpetra Matrix/Vector Services
Version of the Day
|
00001 // @HEADER 00002 // *********************************************************************** 00003 // 00004 // Tpetra: Templated Linear Algebra Services Package 00005 // Copyright (2008) Sandia Corporation 00006 // 00007 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation, 00008 // the U.S. Government retains certain rights in this software. 00009 // 00010 // Redistribution and use in source and binary forms, with or without 00011 // modification, are permitted provided that the following conditions are 00012 // met: 00013 // 00014 // 1. Redistributions of source code must retain the above copyright 00015 // notice, this list of conditions and the following disclaimer. 00016 // 00017 // 2. Redistributions in binary form must reproduce the above copyright 00018 // notice, this list of conditions and the following disclaimer in the 00019 // documentation and/or other materials provided with the distribution. 00020 // 00021 // 3. Neither the name of the Corporation nor the names of the 00022 // contributors may be used to endorse or promote products derived from 00023 // this software without specific prior written permission. 00024 // 00025 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY 00026 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00027 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 00028 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE 00029 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 00030 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 00031 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 00032 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 00033 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 00034 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00035 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00036 // 00037 // Questions? Contact Michael A. Heroux (maherou@sandia.gov) 00038 // 00039 // ************************************************************************ 00040 // @HEADER 00041 00042 #ifndef TPETRA_VIEWACCEPTER_HPP 00043 #define TPETRA_VIEWACCEPTER_HPP 00044 00045 #include <Teuchos_ArrayRCP.hpp> 00046 #include "Tpetra_ConfigDefs.hpp" 00047 00048 #include <Kokkos_SerialNode.hpp> 00049 #ifdef HAVE_KOKKOSCLASSIC_TBB 00050 #include <Kokkos_TBBNode.hpp> 00051 #endif 00052 #ifdef HAVE_KOKKOSCLASSIC_THREADPOOL 00053 #include <Kokkos_TPINode.hpp> 00054 #endif 00055 #ifdef HAVE_KOKKOSCLASSIC_OPENMP 00056 #include <Kokkos_OpenMPNode.hpp> 00057 #endif 00058 00059 namespace Tpetra { 00060 00061 namespace details { 00062 00064 template <class Node> 00065 class ViewAccepter { 00066 private: 00067 // no support for object instantiation 00068 ViewAccepter(); 00069 public: 00070 template <class T> 00071 inline static ArrayRCP<T> acceptView(const ArrayRCP<T> &view) { 00072 Node::this_node_type_does_not_support_object_construction_from_user_views(); 00073 return null; 00074 } 00075 }; 00076 00078 class ViewAccepterSupportedNode { 00079 private: 00080 // no support for object instantiation 00081 ViewAccepterSupportedNode(); 00082 public: 00083 template <class T> 00084 inline static ArrayRCP<T> acceptView(const ArrayRCP<T> &view) { 00085 return view; 00086 } 00087 }; 00088 00089 template <> 00090 class ViewAccepter<Kokkos::SerialNode> : public ViewAccepterSupportedNode {}; 00091 #ifdef HAVE_KOKKOSCLASSIC_TBB 00092 template <> 00093 class ViewAccepter<Kokkos::TBBNode> : public ViewAccepterSupportedNode {}; 00094 #endif 00095 #ifdef HAVE_KOKKOSCLASSIC_THREADPOOL 00096 template <> 00097 class ViewAccepter<Kokkos::TPINode> : public ViewAccepterSupportedNode {}; 00098 #endif 00099 #ifdef HAVE_KOKKOSCLASSIC_OPENMP 00100 template <> 00101 class ViewAccepter<Kokkos::OpenMPNode> : public ViewAccepterSupportedNode {}; 00102 #endif 00103 00104 } // end of namespace Tpetra::details 00105 00106 } // end of namespace Tpetra 00107 00108 #endif
1.7.6.1