Merge pull request #14085 from peterrum/read_write_operation_3_masking

FEEvaluationBase::read_write_operation case 3: enable masking
dealii · Jul 3, 2022 · 196b8d1 · 196b8d1
2 parents 6da2e5d + a460634
commit 196b8d1
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/include/deal.II/matrix_free/fe_evaluation.h b/include/deal.II/matrix_free/fe_evaluation.h
@@ -3342,13 +3342,14 @@ FEEvaluationBase<dim, n_components_, Number, is_face, VectorizedArrayType>::
   // Case 3: standard operation with one index per degree of freedom -> go on
   // here
   constexpr unsigned int n_lanes = VectorizedArrayType::size();
-  Assert(mask.count() == n_lanes,
-         ExcNotImplemented("Masking currently not implemented for "
-                           "non-contiguous DoF storage"));
 
-  const std::array<unsigned int, VectorizedArrayType::size()> &cells =
+  std::array<unsigned int, VectorizedArrayType::size()> cells =
     this->get_cell_ids();
 
+  for (unsigned int v = 0; v < n_lanes; ++v)
+    if (mask[v] == false)
+      cells[v] = numbers::invalid_unsigned_int;
+
   bool has_hn_constraints = false;
 
   if (is_face == false)