1 // @HEADER 2 // ************************************************************************ 3 // 4 // Rapid Optimization Library (ROL) Package 5 // Copyright (2014) Sandia Corporation 6 // 7 // Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive 8 // license for use of this work by or on behalf of the U.S. Government. 9 // 10 // Redistribution and use in source and binary forms, with or without 11 // modification, are permitted provided that the following conditions are 12 // met: 13 // 14 // 1. Redistributions of source code must retain the above copyright 15 // notice, this list of conditions and the following disclaimer. 16 // 17 // 2. Redistributions in binary form must reproduce the above copyright 18 // notice, this list of conditions and the following disclaimer in the 19 // documentation and/or other materials provided with the distribution. 20 // 21 // 3. Neither the name of the Corporation nor the names of the 22 // contributors may be used to endorse or promote products derived from 23 // this software without specific prior written permission. 24 // 25 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY 26 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 28 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE 29 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 30 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 31 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 32 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 33 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 34 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 35 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 36 // 37 // Questions? Contact lead developers: 38 // Drew Kouri (dpkouri@sandia.gov) and 39 // Denis Ridzal (dridzal@sandia.gov) 40 // 41 // ************************************************************************ 42 // @HEADER 43 44 #ifndef ROL_TPETRATEUCHOSBATCHMANAGER_HPP 45 #define ROL_TPETRATEUCHOSBATCHMANAGER_HPP 46 47 #include "ROL_TeuchosBatchManager.hpp" 48 #include "ROL_TpetraMultiVector.hpp" 49 50 namespace ROL { 51 52 template<class Real, 53 class LO=Tpetra::Map<>::local_ordinal_type, 54 class GO=Tpetra::Map<>::global_ordinal_type, 55 class Node=Tpetra::Map<>::node_type > 56 class TpetraTeuchosBatchManager : public TeuchosBatchManager<Real,GO> { 57 typedef Tpetra::MultiVector<Real,LO,GO,Node> Tpetra_Vector; 58 typedef TpetraMultiVector<Real,LO,GO,Node> OptVector; 59 60 public: TpetraTeuchosBatchManager(const ROL::Ptr<const Teuchos::Comm<int>> & comm)61 TpetraTeuchosBatchManager(const ROL::Ptr<const Teuchos::Comm<int> > &comm) 62 : TeuchosBatchManager<Real,GO>(comm) {} 63 sumAll(Vector<Real> & input,Vector<Real> & output)64 void sumAll(Vector<Real> &input, Vector<Real> &output) { 65 ROL::Ptr<Tpetra_Vector> ivec = dynamic_cast<OptVector&>(input).getVector(); 66 ROL::Ptr<Tpetra_Vector> ovec = dynamic_cast<OptVector&>(output).getVector(); 67 68 size_t ilength = ivec->getLocalLength(), olength = ovec->getLocalLength(); 69 ROL_TEST_FOR_EXCEPTION(ilength != olength, std::invalid_argument, 70 ">>> (TpetraTeuchosBatchManager::sumAll): Inconsistent local lengths!"); 71 72 size_t invec = ivec->getNumVectors(), onvec = ovec->getNumVectors(); 73 ROL_TEST_FOR_EXCEPTION(invec != onvec, std::invalid_argument, 74 ">>> (TpetraTeuchosBatchManager::sumAll): Inconsistent number of vectors!"); 75 76 for (size_t i = 0; i < invec; ++i) { 77 TeuchosBatchManager<Real,GO>::sumAll((ivec->getDataNonConst(i)).getRawPtr(), 78 (ovec->getDataNonConst(i)).getRawPtr(), 79 ilength); 80 } 81 } 82 }; 83 84 } 85 86 #endif 87