Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix MatrixFree + FE_Q_iso_Q1(2) #14197

Merged
merged 1 commit into from
Aug 18, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
14 changes: 8 additions & 6 deletions include/deal.II/matrix_free/evaluation_kernels.h
Original file line number Diff line number Diff line change
Expand Up @@ -2303,7 +2303,8 @@ namespace internal
values_dofs,
fe_eval);
}
else if (fe_degree >= 0 && element_type <= ElementType::tensor_symmetric)
else if (fe_degree >= 0 &&
element_type <= ElementType::tensor_symmetric_no_collocation)
{
FEEvaluationImpl<ElementType::tensor_symmetric,
dim,
Expand Down Expand Up @@ -2438,7 +2439,8 @@ namespace internal
fe_eval,
sum_into_values_array);
}
else if (fe_degree >= 0 && element_type <= ElementType::tensor_symmetric)
else if (fe_degree >= 0 &&
element_type <= ElementType::tensor_symmetric_no_collocation)
{
FEEvaluationImpl<ElementType::tensor_symmetric,
dim,
Expand Down Expand Up @@ -5174,7 +5176,7 @@ namespace internal
{
Assert(fe_degree > -1, ExcInternalError());
Assert(fe_eval.get_shape_info().element_type <=
MatrixFreeFunctions::tensor_symmetric,
MatrixFreeFunctions::tensor_symmetric_no_collocation,
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Here, I think we also need to play safe and only enable it for the tensor_symmetric case.

ExcInternalError());

const unsigned int dofs_per_face = Utilities::pow(fe_degree + 1, dim - 1);
Expand Down Expand Up @@ -5293,7 +5295,7 @@ namespace internal
(evaluation_flag & EvaluationFlags::hessians) ||
vector_ptr == nullptr ||
shape_info.data.front().element_type >
MatrixFreeFunctions::tensor_symmetric ||
MatrixFreeFunctions::tensor_symmetric_no_collocation ||
storage <
MatrixFreeFunctions::DoFInfo::IndexStorageVariants::contiguous)
return false;
Expand Down Expand Up @@ -5394,7 +5396,7 @@ namespace internal
{
Assert(fe_degree > -1, ExcInternalError());
Assert(fe_eval.get_shape_info().element_type <=
MatrixFreeFunctions::tensor_symmetric,
MatrixFreeFunctions::tensor_symmetric_no_collocation,
ExcInternalError());

const unsigned int dofs_per_face = Utilities::pow(fe_degree + 1, dim - 1);
Expand Down Expand Up @@ -5601,7 +5603,7 @@ namespace internal

Assert(dim >= 1 || dim <= 3, ExcNotImplemented());
Assert(fe_eval.get_shape_info().element_type <=
MatrixFreeFunctions::tensor_symmetric,
MatrixFreeFunctions::tensor_symmetric_no_collocation,
ExcNotImplemented());

EvaluatorTensorProduct<evaluate_evenodd,
Expand Down
16 changes: 11 additions & 5 deletions include/deal.II/matrix_free/shape_info.h
Original file line number Diff line number Diff line change
Expand Up @@ -81,35 +81,41 @@ namespace internal
*/
tensor_symmetric = 2,

/**
* For function spaces that are not equivalent to a polynom of degree p,
* which the assumption of collocation is.
*/
tensor_symmetric_no_collocation = 3,

/**
* Tensor product shape functions without further particular properties
*/
tensor_general = 3,
tensor_general = 4,

/**
* Polynomials of complete degree rather than tensor degree which can be
* described by a truncated tensor product
*/
truncated_tensor = 4,
truncated_tensor = 5,

/**
* Tensor product shape functions that are symmetric about the midpoint
* of the unit interval 0.5 that additionally add a constant shape
* function according to FE_Q_DG0.
*/
tensor_symmetric_plus_dg0 = 5,
tensor_symmetric_plus_dg0 = 6,

/**
* Special case of the FE_RaviartThomasNodal element with anisotropic
* tensor product shape functions, i.e degree (k + 1) in normal direction,
* and k in tangential direction.
*/
tensor_raviart_thomas = 6,
tensor_raviart_thomas = 7,

/**
* Shape functions without a tensor product properties.
*/
tensor_none = 7
tensor_none = 8


};
Expand Down
7 changes: 6 additions & 1 deletion include/deal.II/matrix_free/shape_info.templates.h
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@
#include <deal.II/fe/fe_pyramid_p.h>
#include <deal.II/fe/fe_q.h>
#include <deal.II/fe/fe_q_dg0.h>
#include <deal.II/fe/fe_q_iso_q1.h>
#include <deal.II/fe/fe_raviart_thomas.h>
#include <deal.II/fe/fe_simplex_p.h>
#include <deal.II/fe/fe_simplex_p_bubbles.h>
Expand Down Expand Up @@ -913,10 +914,14 @@ namespace internal
if (element_type == tensor_general &&
check_1d_shapes_symmetric(univariate_shape_data))
{
if (check_1d_shapes_collocation(univariate_shape_data))
if (dynamic_cast<const FE_Q_iso_Q1<dim> *>(&fe) &&
fe.tensor_degree() > 1)
element_type = tensor_symmetric_no_collocation;
else if (check_1d_shapes_collocation(univariate_shape_data))
element_type = tensor_symmetric_collocation;
else
element_type = tensor_symmetric;

if (n_dofs_1d > 2 && element_type == tensor_symmetric)
{
// check if we are a Hermite type
Expand Down
98 changes: 98 additions & 0 deletions tests/matrix_free/fe_q_iso_q1_01.cc
Original file line number Diff line number Diff line change
@@ -0,0 +1,98 @@
// ---------------------------------------------------------------------
//
// Copyright (C) 2022 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------



// Test the evaluated values and gradients at the quadrature points for
// FE_Q_iso_Q1.

#include <deal.II/base/quadrature_lib.h>

#include <deal.II/fe/fe_q_iso_q1.h>

#include <deal.II/grid/grid_generator.h>

#include <deal.II/matrix_free/fe_evaluation.h>
#include <deal.II/matrix_free/matrix_free.h>

#include "../tests.h"


template <int dim>
void
test(const unsigned int n_subdivisions)
{
const FE_Q_iso_Q1<dim> fe(n_subdivisions);
const QIterated<1> quad(QGauss<1>(2), n_subdivisions);

Triangulation<dim> tria;
GridGenerator::hyper_cube(tria);

DoFHandler<dim> dof(tria);
dof.distribute_dofs(fe);

AffineConstraints<double> constraints;
constraints.close();

using number = double;
MatrixFree<dim, number> matrix_free;
typename MatrixFree<dim, number>::AdditionalData data;
data.mapping_update_flags = update_values | update_gradients;
matrix_free.reinit(MappingQ1<dim>(), dof, constraints, quad, data);

FEEvaluation<dim, -1, 0, 1, number> fe_eval(matrix_free);
fe_eval.reinit(0);

for (unsigned int i = 0; i <= dim; ++i)
{
for (unsigned int j = 0; j < fe_eval.dofs_per_cell; ++j)
{
for (unsigned int k = 0; k < fe_eval.dofs_per_cell; ++k)
fe_eval.begin_dof_values()[k] = static_cast<number>(j == k);

fe_eval.evaluate(EvaluationFlags::values |
EvaluationFlags::gradients);

for (unsigned int q = 0; q < fe_eval.n_q_points; ++q)
{
const auto temp = (i == 0) ? fe_eval.get_value(q) :
fe_eval.get_gradient(q)[i - 1];
deallog << ((std::abs(temp[0]) > 1e-8) ? 1 : 0) << " ";
}

deallog << std::endl;
}

deallog << std::endl;
}
}

int
main(int argc, char **argv)
{
Utilities::MPI::MPI_InitFinalize mpi(argc, argv, 1);
MPILogInitAll log;

{
deallog.push("2d");
for (unsigned int i = 1; i <= 4; ++i)
test<2>(i);
deallog.pop();
deallog.push("3d");
for (unsigned int i = 1; i <= 4; ++i)
test<2>(i);
deallog.pop();
}
}