|
Isorropia: Partitioning, Load Balancing and more
|
00001 //@HEADER 00002 //************************************************************************ 00003 // 00004 // Isorropia: Partitioning and Load Balancing Package 00005 // Copyright (2006) Sandia Corporation 00006 // 00007 //Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive 00008 //license for use of this work by or on behalf of the U.S. Government. 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 //************************************************************************ 00038 //@HEADER 00039 00040 #ifndef _Isorropia_TpetraLevelScheduler_hpp_ 00041 #define _Isorropia_TpetraLevelScheduler_hpp_ 00042 00043 #include <Isorropia_ConfigDefs.hpp> 00044 #include <Isorropia_TpetraOperator.hpp> 00045 #include <Isorropia_LevelScheduler.hpp> 00046 #include <Teuchos_RCP.hpp> 00047 00048 #ifdef HAVE_ISORROPIA_TPETRA 00049 00050 #include <Tpetra_CrsGraph_decl.hpp> 00051 #include <Kokkos_DefaultNode.hpp> 00052 00053 namespace Isorropia { 00054 00055 namespace Tpetra { 00056 00063 template <class Node=KokkosClassic::DefaultNode::DefaultNodeType> 00064 class LevelScheduler : public Isorropia::LevelScheduler, public Isorropia::Tpetra::Operator<Node> { 00065 00066 public: 00067 00075 LevelScheduler(Teuchos::RCP<const ::Tpetra::CrsGraph<int,int,Node> > input_graph, 00076 const Teuchos::ParameterList& paramlist=Teuchos::ParameterList("EmptyParameterList"), 00077 bool compute_now=true); 00078 00080 ~LevelScheduler(); // {} ; 00081 00088 void schedule(bool force_scheduling=false); 00089 00096 void compute(bool force_compute=false) { 00097 schedule(force_compute); 00098 } 00099 00100 private: 00101 Teuchos::RCP<const ::Tpetra::CrsGraph<int,int,Node> > input_graph_; 00102 00103 };//class LevelScheduler 00104 00105 }//namespace Tpetra 00106 }//namespace Isorropia 00107 00108 #endif //HAVE_ISORROPIA_TPETRA 00109 00110 #endif 00111