64 const Teuchos::RCP<const panzer::GlobalIndexer> & indexer,
65 const Teuchos::ParameterList& p)
66 : globalIndexer_(indexer)
67 , useTimeDerivativeSolutionVector_(false)
68 , globalDataKey_(
"Tangent Gather Container")
70 const std::vector<std::string>& names =
71 *(p.get< Teuchos::RCP< std::vector<std::string> > >(
"DOF Names"));
73 indexerNames_ = p.get< Teuchos::RCP< std::vector<std::string> > >(
"Indexer Names");
76 Teuchos::RCP<const panzer::PureBasis> basis;
77 if(p.isType< Teuchos::RCP<panzer::PureBasis> >(
"Basis"))
78 basis = p.get< Teuchos::RCP<panzer::PureBasis> >(
"Basis");
80 basis = p.get< Teuchos::RCP<const panzer::PureBasis> >(
"Basis");
83 for (std::size_t fd = 0; fd < names.size(); ++fd) {
85 PHX::MDField<ScalarT,Cell,NODE>(names[fd],basis->functional);
93 if (p.isType<
bool>(
"Use Time Derivative Solution Vector"))
96 if (p.isType<std::string>(
"Global Data Key"))
99 this->setName(
"Gather Tangent");
108 TEUCHOS_ASSERT(gatherFields_.size() == indexerNames_->size());
110 fieldIds_.resize(gatherFields_.size());
112 gatherFieldsVoV_.initialize(
"GatherSolution_Teptra<Tangent>",gatherFields_.size());
114 for (std::size_t fd = 0; fd < gatherFields_.size(); ++fd) {
115 const std::string& fieldName = (*indexerNames_)[fd];
116 fieldIds_[fd] = globalIndexer_->getFieldNum(fieldName);
117 gatherFieldsVoV_.addView(gatherFields_[fd].get_static_view(),fd);
120 gatherFieldsVoV_.syncHostToDevice();
122 indexerNames_ = Teuchos::null;
131 using Teuchos::rcp_dynamic_cast;
136 if (d.gedc->containsDataObject(globalDataKey_)) {
137 RCP<GlobalEvaluationData> ged = d.gedc->getDataObject(globalDataKey_);
138 RCP<LOCPair_GlobalEvaluationData> loc_pair =
139 rcp_dynamic_cast<LOCPair_GlobalEvaluationData>(ged);
141 if(loc_pair!=Teuchos::null) {
142 Teuchos::RCP<LinearObjContainer> loc = loc_pair->getGhostedLOC();
143 tpetraContainer_ = rcp_dynamic_cast<LOC>(loc,
true);
146 if(tpetraContainer_==Teuchos::null) {
147 tpetraContainer_ = rcp_dynamic_cast<LOC>(ged,
true);
159 if (tpetraContainer_ == Teuchos::null)
164 std::string blockId = this->wda(workset).block_id;
166 Teuchos::RCP<typename LOC::VectorType> x;
167 if (useTimeDerivativeSolutionVector_)
170 x = tpetraContainer_->get_x();
172 auto cellLocalIdsKokkos = this->wda(workset).getLocalCellIDs();
173 auto lids = globalIndexer_->getLIDs();
174 auto vov = Teuchos::rcp_dynamic_cast<const panzer::DOFManager>(globalIndexer_,
true)->getGIDFieldOffsetsKokkos(blockId,fieldIds_);
175 auto gidFieldOffsets = vov.getViewDevice();
176 auto gatherFieldsDevice = gatherFieldsVoV_.getViewDevice();
177 auto x_view = x->getLocalViewDevice(Tpetra::Access::ReadWrite);
178 Kokkos::MDRangePolicy<PHX::Device::execution_space,Kokkos::Rank<2>> policy({0,0},{cellLocalIdsKokkos.extent(0),gidFieldOffsets.extent(0)});
179 Kokkos::parallel_for(
"GatherSolutionTpetra<Tangent>",policy,KOKKOS_LAMBDA(
const int worksetCellIndex,
const int fieldIndex) {
180 for(std::size_t basis=0;basis<gidFieldOffsets(fieldIndex).extent(0);basis++) {
181 int offset = gidFieldOffsets(fieldIndex)(basis);
182 LO lid =
lids(cellLocalIdsKokkos(worksetCellIndex),offset);
183 auto& gf_ref = (gatherFieldsDevice[fieldIndex])(worksetCellIndex,basis);
184 gf_ref = x_view(lid,0);