Skip to content

Vector pa kernels #4861

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.

Sign up for GitHub

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

Draft
wants to merge 46 commits into
base: master
Choose a base branch
from
Draft

Vector pa kernels #4861

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
46 commits
Select commit Hold shift + click to select a range
68ae66c
vecmass PA setup
camierjs May 9, 2025
93dbe14
PAVectorMassApply3D with Eval3d
camierjs May 9, 2025
3949dec
PAVectorMassApply2D with Eval2d
camierjs May 10, 2025
d2e2834
VectorMassIntegrator PA specializations
camierjs May 10, 2025
b08d12b
VectorMassIntegrator 2D with MatrixFunctionCoefficients
camierjs May 12, 2025
0892f19
VectorMassIntegrator 3D MatrixFunctionCoefficient
camierjs May 12, 2025
1c04593
VectorMassIntegrator cleanup
camierjs May 12, 2025
72d28e0
VectorDiffusionIntegrator AssemblePA vector coefficients
camierjs May 12, 2025
ba4befc
VectorDiffusionIntegrator wip Matrix coefficient
camierjs May 12, 2025
c707a4d
jj=ii
camierjs May 12, 2025
3086dae
WIP PAVectorDiffusionApply2D mcoeff
camierjs May 13, 2025
5827732
WIP 0 1
camierjs May 13, 2025
7f7e5f6
All c, d
camierjs May 13, 2025
031cd67
VectorDiffusionIntegrator 2D mcoeff
camierjs May 13, 2025
2a3c1c5
VectorDiffusionIntegrator 2D cleanup
camierjs May 13, 2025
3c2f98b
test_vector_pa_integrator all tests passed
camierjs May 13, 2025
3fb13a3
WIP 3D vector diffusion with sym mcoeff
camierjs May 13, 2025
d27fa84
Pre AssemblePA 2D vector diffusion
camierjs May 13, 2025
580dc6f
Simplify VectorDiffusionIntegrator::AssemblePA
camierjs May 13, 2025
6a04555
SmemPAVectorDiffusionApply2D
camierjs May 13, 2025
795a121
SmemPAVectorDiffusionApply3D
camierjs May 13, 2025
686feff
VectorDiffusionAddMultPA registered
camierjs May 13, 2025
51c28ad
Fix grad ii vs. jj matrix coeff
camierjs May 14, 2025
eb68434
Simplify regs_t kernels types on CPU
May 15, 2025
e41c782
Use explicit type names for registers
May 15, 2025
62dd1ae
Cleanup
May 16, 2025
ea25a9b
Remove PAVectorDiffusionApply kernels, fuse SDIM != DIM diffusion vec…
camierjs May 17, 2025
3b88332
Merge branch 'master' into vector-pa-kernels
camierjs May 17, 2025
6bb5f8a
Change pa data layout
camierjs May 18, 2025
190c4a0
WIP SmemPAVectorMassApply2D layouts
camierjs May 18, 2025
2ead488
Pre SmemPAVectorMassApply2D cleanup
camierjs May 18, 2025
82c2e01
Cleanup
camierjs May 18, 2025
7a37cc2
MSVC header guards
camierjs May 18, 2025
c50905e
MAX_T1D for WIN32
camierjs May 18, 2025
f3443f9
MSVC fix
camierjs May 18, 2025
b3e9ddd
WIN32 cmath
camierjs May 18, 2025
f9972a5
Simplify back SmemPAVectorMassApply2D
camierjs May 19, 2025
716c201
GPU vector kernels fix
camierjs May 19, 2025
0742475
test_pa_kernels tags
camierjs May 19, 2025
189c201
Fix shadowing
camierjs May 19, 2025
7dba460
Move kernel methods to fem/kernels.hpp
camierjs May 19, 2025
acbe258
CMake remove header
camierjs May 19, 2025
6145305
Merge branch 'master' into vector-pa-kernels
camierjs May 28, 2025
2e70688
Merge branch 'master' into vector-pa-kernels
camierjs Jun 13, 2025
ba70885
Merge branch 'master' into vector-pa-kernels
camierjs Jun 16, 2025
319b870
Use future namespace for tensor
camierjs Jun 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
VectorMassIntegrator 3D MatrixFunctionCoefficient
  • Loading branch information
@camierjs
camierjs committed May 12, 2025
commit 0892f190cf74481fd1b68fbf38d0d411e7ad2c1e
162 changes: 96 additions & 66 deletions fem/integ/bilininteg_vecmass_pa.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,19 +29,20 @@ namespace mfem
void VectorMassIntegrator::AssemblePA(const FiniteElementSpace &fes)
{
Mesh *mesh = fes.GetMesh();
const FiniteElement &el = *fes.GetTypicalFE();
ElementTransformation &Trans = *mesh->GetTypicalElementTransformation();
const int spaceDim = Trans.GetSpaceDim();

const FiniteElement &el = *fes.GetTypicalFE();
const int nd = el.GetDof();
dbg("nd: {} spaceDim: {}", nd, spaceDim);

dbg("vdim: {} spaceDim: {}", vdim, spaceDim);

// If vdim is not set, set it to the space dimension
vdim = (vdim == -1) ? Trans.GetSpaceDim() : vdim;
dbg("vdim: {}", vdim);

MFEM_VERIFY(vdim == fes.GetVDim(), "vdim != fes.GetVDim()");

const MemoryType mt = pa_mt == MemoryType::DEFAULT
? Device::GetDeviceMemoryType()
: pa_mt;
Expand All @@ -60,8 +61,11 @@ void VectorMassIntegrator::AssemblePA(const FiniteElementSpace &fes)
dim = mesh->Dimension();
ne = mesh->GetNE();
nq = ir->GetNPoints();
const int flags = GeometricFactors::COORDINATES | GeometricFactors::JACOBIANS;
const int sdim = mesh->SpaceDimension();
MFEM_VERIFY(sdim == dim, "sdim != dim");
const int flags = GeometricFactors::JACOBIANS;
geom = mesh->GetGeometricFactors(*ir, flags, mt);

maps = &el.GetDofToQuad(*ir, DofToQuad::TENSOR);
dofs1D = maps->ndof;
quad1D = maps->nqpt;
Expand Down Expand Up @@ -105,83 +109,84 @@ void VectorMassIntegrator::AssemblePA(const FiniteElementSpace &fes)

if (!(dim == 2 || dim == 3)) { MFEM_ABORT("Dimension not supported."); }

const int NE = ne, NQ = nq;
const auto W = ir->GetWeights().Read();
const auto C = Reshape(coeff.Read(), coeff_vdim, NQ, NE);
const auto J = Reshape(geom->J.Read(), NQ, dim, dim, NE);

auto DE = Reshape(pa_data.Write(), NQ, coeff_vdim, NE);
const int q1d = quad1D;
const auto w_r = ir->GetWeights().Read();

if (dim == 2)
{
const auto W = Reshape(w_r, q1d, q1d);
const auto C = Reshape(coeff.Read(), coeff_vdim, q1d, q1d, ne);
const auto J = Reshape(geom->J.Read(), q1d, q1d, sdim, dim, ne);
// const auto J = Reshape(geom->J.Read(), dim, dim, NQ, NE);
auto DE = Reshape(pa_data.Write(), q1d, q1d, coeff_vdim, ne);

mfem::forall(NE, [=] MFEM_HOST_DEVICE(int e)
mfem::forall_2D(ne, q1d, q1d, [=] MFEM_HOST_DEVICE(int e)
{
for (int q = 0; q < NQ; ++q)
MFEM_FOREACH_THREAD(qy, y, q1d)
{
const real_t J11 = J(q, 0, 0, e);
const real_t J12 = J(q, 1, 0, e);
const real_t J21 = J(q, 0, 1, e);
const real_t J22 = J(q, 1, 1, e);
const real_t detJ = (J11 * J22) - (J21 * J12);
const real_t w_det = W[q] * detJ;
DE(q, 0, e) = C(0, q, e) * w_det;
if (const_coeff) { continue; }
DE(q, 1, e) = C(1, q, e) * w_det;
if (vector_coeff) { continue; }
assert(matrix_coeff);
DE(q, 2, e) = C(2, q, e) * w_det;
DE(q, 3, e) = C(3, q, e) * w_det;
MFEM_FOREACH_THREAD(qx, x, q1d)
{
// const real_t *Jtr = &J(0, 0, qx, qy, e);
// const real_t detJ = kernels::Det<2>(Jtr);
const real_t J11 = J(qx, qy, 0, 0, e), J12 = J(qx, qy, 1, 0, e);
const real_t J21 = J(qx, qy, 0, 1, e), J22 = J(qx, qy, 1, 1, e);
const real_t detJ = (J11 * J22) - (J21 * J12);
const real_t w_det = W(qx, qy) * detJ;
DE(qx, qy, 0, e) = C(0, qx, qy, e) * w_det;
if (const_coeff) { continue; }
DE(qx, qy, 1, e) = C(1, qx, qy, e) * w_det;
if (vector_coeff) { continue; }
assert(matrix_coeff);
DE(qx, qy, 2, e) = C(2, qx, qy, e) * w_det;
DE(qx, qy, 3, e) = C(3, qx, qy, e) * w_det;
}
}
});
}

if (dim == 3)
{
assert(false);
/*mfem::forall(NE, [=] MFEM_HOST_DEVICE(int e)
const auto W = Reshape(w_r, q1d, q1d, q1d);
const auto C = Reshape(coeff.Read(), coeff_vdim, q1d, q1d, q1d, ne);
const auto J = Reshape(geom->J.Read(), q1d, q1d, q1d, dim, dim, ne);
auto DE = Reshape(pa_data.Write(), q1d, q1d, q1d, coeff_vdim, ne);

mfem::forall_3D(ne, q1d, q1d, q1d, [=] MFEM_HOST_DEVICE(int e)
{
for (int q = 0; q < NQ; ++q)
MFEM_FOREACH_THREAD(qz, z, q1d)
{
const real_t J11 = J(q, 0, 0, e), J12 = J(q, 0, 1, e), J13 = J(q, 0, 2, e);
const real_t J21 = J(q, 1, 0, e), J22 = J(q, 1, 1, e), J23 = J(q, 1, 2, e);
const real_t J31 = J(q, 2, 0, e), J32 = J(q, 2, 1, e), J33 = J(q, 2, 2, e);
const real_t detJ = J11 * (J22 * J33 - J32 * J23) -
J21 * (J12 * J33 - J32 * J13) +
J31 * (J12 * J23 - J22 * J13);
const real_t w_det = W[q] * detJ;
y(q, 0, e) = C(0, q, e) * w_det;
y(q, 1, e) = C(const_coeff ? 0 : 1, q, e) * w_det;
y(q, 2, e) = C(const_coeff ? 0 : 2, q, e) * w_det;
MFEM_FOREACH_THREAD(qy, y, q1d)
{
MFEM_FOREACH_THREAD(qx, x, q1d)
{
const real_t J11 = J(qx, qy, qz, 0, 0, e);
const real_t J12 = J(qx, qy, qz, 0, 1, e),
J13 = J(qx, qy, qz, 0, 2, e);
const real_t J21 = J(qx, qy, qz, 1, 0, e), J22 = J(qx, qy, qz, 1, 1, e),
J23 = J(qx, qy, qz, 1, 2, e);
const real_t J31 = J(qx, qy, qz, 2, 0, e), J32 = J(qx, qy, qz, 2, 1, e),
J33 = J(qx, qy, qz, 2, 2, e);
const real_t detJ = J11 * (J22 * J33 - J32 * J23) -
J21 * (J12 * J33 - J32 * J13) +
J31 * (J12 * J23 - J22 * J13);
const real_t w_det = W(qx, qy, qz) * detJ;
DE(qx, qy, qz, 0, e) = C(0, qx, qy, qz, e) * w_det;
if (const_coeff) { continue; }
DE(qx, qy, qz, 1, e) = C(1, qx, qy, qz, e) * w_det;
DE(qx, qy, qz, 2, e) = C(2, qx, qy, qz, e) * w_det;
if (vector_coeff) { continue; }
assert(matrix_coeff);
DE(qx, qy, qz, 3, e) = C(3, qx, qy, qz, e) * w_det;
DE(qx, qy, qz, 4, e) = C(4, qx, qy, qz, e) * w_det;
DE(qx, qy, qz, 5, e) = C(5, qx, qy, qz, e) * w_det;
DE(qx, qy, qz, 6, e) = C(6, qx, qy, qz, e) * w_det;
DE(qx, qy, qz, 7, e) = C(7, qx, qy, qz, e) * w_det;
DE(qx, qy, qz, 8, e) = C(8, qx, qy, qz, e) * w_det;
}
}
}
});*/
});
}

// Add the VectorMassIntegrator specializations
static const auto add_vector_mass_kernel_specializations =
( // 2D
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,2,2>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,2,2>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,3,3>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,4,4>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,5,5>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,6,6>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,7,7>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,8,8>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,9,9>::Add(),
// 3D
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,2,2>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,2,3>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,3,4>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,4,5>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,4,6>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,5,6>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,5,8>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,6,7>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,7,8>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,8,9>::Add(),
true);
MFEM_CONTRACT_VAR(add_vector_mass_kernel_specializations);
}

void VectorMassIntegrator::AddMultPA(const Vector &x, Vector &y) const
Expand All @@ -190,9 +195,34 @@ void VectorMassIntegrator::AddMultPA(const Vector &x, Vector &y) const
if (DeviceCanUseCeed()) { ceedOp->AddMult(x, y); }
else
{
// Add the VectorMassIntegrator specializations
static const auto vector_mass_kernel_specializations =
( // 2D
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,2,2>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,2,2>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,3,3>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,4,4>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,5,5>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,6,6>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,7,7>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,8,8>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<2,9,9>::Add(),
// 3D
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,2,2>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,2,3>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,3,4>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,4,5>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,4,6>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,5,6>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,5,8>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,6,7>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,7,8>::Add(),
VectorMassIntegrator::VectorMassAddMultPA::Specialization<3,8,9>::Add(),
true);
MFEM_CONTRACT_VAR(vector_mass_kernel_specializations);

VectorMassAddMultPA::Run(dim, dofs1D, quad1D,
ne, //coeff.GetVDim(),
maps->B, pa_data, x, y,
ne, maps->B, pa_data, x, y,
dofs1D, quad1D);
}
}
Expand Down
86 changes: 58 additions & 28 deletions fem/integ/bilininteg_vecmass_pa.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
#include "../../general/array.hpp"
#include "../../general/forall.hpp"
#include "../../linalg/dtensor.hpp"
#include "../../linalg/kernels.hpp"
#include "../../linalg/vector.hpp"
#include "../bilininteg.hpp"

Expand Down Expand Up @@ -44,13 +45,7 @@ static void PAVectorMassApply2D(const int NE,
const bool const_coeff = coeff_vdim == 1;
const bool vector_coeff = coeff_vdim == VDIM;
const bool matrix_coeff = coeff_vdim == VDIM*VDIM;

mfem::out << "\x1b[33m"
<< "PAVectorMassApply2D: coeff_vdim = " << coeff_vdim
<< ", const_coeff = " << const_coeff
<< ", vector_coeff = " << vector_coeff
<< ", matrix_coeff = " << matrix_coeff
<< "\x1b[m" << std::endl;
MFEM_VERIFY(const_coeff + vector_coeff + matrix_coeff == 1, "");

const auto DE = Reshape(d.Read(), Q1D, Q1D, coeff_vdim, NE);
const auto XE = Reshape(x.Read(), D1D, D1D, VDIM, NE);
Expand All @@ -68,6 +63,7 @@ static void PAVectorMassApply2D(const int NE,
kernels::internal::LoadMatrix(D1D, Q1D, b, sB);
kernels::internal::LoadDofs2d(e, D1D, XE, r0);
kernels::internal::Eval2d(D1D, Q1D, smem, sB, r0, r1);

MFEM_FOREACH_THREAD(qy, y, Q1D)
{
MFEM_FOREACH_THREAD(qx, x, Q1D)
Expand All @@ -78,24 +74,24 @@ static void PAVectorMassApply2D(const int NE,

if (const_coeff)
{
r0[0][0][qy][qx] = D0*Qx;
r0[1][0][qy][qx] = D0*Qy;
// kernels::Mult(2, 1, D, Q, R);
r0[0][0][qy][qx] = D0 * Qx;
r0[1][0][qy][qx] = D0 * Qy;
}
else if (vector_coeff)
if (vector_coeff)
{
const real_t D1 = DE(qx, qy, 1, e);
r0[0][0][qy][qx] = D0*Qx;
r0[1][0][qy][qx] = D1*Qy;
r0[0][0][qy][qx] = D0 * Qx;
r0[1][0][qy][qx] = D1 * Qy;
}
else if (matrix_coeff)
if (matrix_coeff)
{
const real_t D1 = DE(qx, qy, 1, e);
const real_t D2 = DE(qx, qy, 2, e);
const real_t D3 = DE(qx, qy, 3, e);
r0[0][0][qy][qx] = D0*Qx + D1*Qy;
r0[1][0][qy][qx] = D2*Qx + D3*Qy;
r0[0][0][qy][qx] = D0 * Qx + D1 * Qy;
r0[1][0][qy][qx] = D2 * Qx + D3 * Qy;
}
else { assert(false); }
}
}
kernels::internal::EvalTranspose2d(D1D, Q1D, smem, sB, r0, r1);
Expand All @@ -105,20 +101,26 @@ static void PAVectorMassApply2D(const int NE,

template <int T_D1D = 0, int T_Q1D = 0>
static void PAVectorMassApply3D(const int NE,
const Array<real_t> &B_,
const Vector &op_,
const Vector &x_,
Vector &y_,
const Array<real_t> &b,
const Vector &d,
const Vector &x,
Vector &y,
const int d1d = 0,
const int q1d = 0)
{
constexpr int VDIM = 3;
const int D1D = T_D1D ? T_D1D : d1d;
const int Q1D = T_Q1D ? T_Q1D : q1d;

const auto DE = Reshape(op_.Read(), Q1D, Q1D, Q1D, VDIM, NE);
const auto XE = Reshape(x_.Read(), D1D, D1D, D1D, VDIM, NE);
auto YE = Reshape(y_.ReadWrite(), D1D, D1D, D1D, VDIM, NE);
const int coeff_vdim = d.Size() / (Q1D*Q1D*Q1D*NE);
const bool const_coeff = coeff_vdim == 1;
const bool vector_coeff = coeff_vdim == VDIM;
const bool matrix_coeff = coeff_vdim == VDIM*VDIM;
MFEM_VERIFY(const_coeff + vector_coeff + matrix_coeff == 1, "");

const auto DE = Reshape(d.Read(), Q1D, Q1D, Q1D, coeff_vdim, NE);
const auto XE = Reshape(x.Read(), D1D, D1D, D1D, VDIM, NE);
auto YE = Reshape(y.ReadWrite(), D1D, D1D, D1D, VDIM, NE);

mfem::forall_2D(NE, Q1D, Q1D, [=] MFEM_HOST_DEVICE(int e)
{
Expand All @@ -127,21 +129,49 @@ static void PAVectorMassApply3D(const int NE,

MFEM_SHARED real_t sB[MD1][MQ1];
MFEM_SHARED real_t smem[MQ1][MQ1];

kernels::internal::regs5d_t<VDIM, 1, MQ1> r0, r1;
kernels::internal::LoadMatrix(D1D, Q1D, B_, sB);
kernels::internal::LoadMatrix(D1D, Q1D, b, sB);
kernels::internal::LoadDofs3d(e, D1D, XE, r0);
kernels::internal::Eval3d(D1D, Q1D, smem, sB, r0, r1);

for (int qz = 0; qz < Q1D; qz++)
{
MFEM_FOREACH_THREAD(qy, y, Q1D)
{
MFEM_FOREACH_THREAD(qx, x, Q1D)
{
for (int c = 0; c < VDIM; ++c)
const real_t Qx = r1[0][0][qz][qy][qx];
const real_t Qy = r1[1][0][qz][qy][qx];
const real_t Qz = r1[2][0][qz][qy][qx];
const real_t D0 = DE(qx, qy, qz, 0, e);
if (const_coeff)
{
r0[0][0][qz][qy][qx] = D0 * Qx;
r0[1][0][qz][qy][qx] = D0 * Qy;
r0[2][0][qz][qy][qx] = D0 * Qz;
}
if (vector_coeff)
{
const real_t D1 = DE(qx, qy, qz, 1, e);
const real_t D2 = DE(qx, qy, qz, 2, e);
r0[0][0][qz][qy][qx] = D0 * Qx;
r0[1][0][qz][qy][qx] = D1 * Qy;
r0[2][0][qz][qy][qx] = D2 * Qz;
}
if (matrix_coeff)
{
const real_t r = r1[c][0][qz][qy][qx];
const real_t D = DE(qx, qy, qz, c, e);
r0[c][0][qz][qy][qx] = D * r;
const real_t D1 = DE(qx, qy, qz, 1, e);
const real_t D2 = DE(qx, qy, qz, 2, e);
const real_t D3 = DE(qx, qy, qz, 3, e);
const real_t D4 = DE(qx, qy, qz, 4, e);
const real_t D5 = DE(qx, qy, qz, 5, e);
const real_t D6 = DE(qx, qy, qz, 6, e);
const real_t D7 = DE(qx, qy, qz, 7, e);
const real_t D8 = DE(qx, qy, qz, 8, e);
r0[0][0][qz][qy][qx] = D0 * Qx + D1 * Qy + D2 * Qz;
r0[1][0][qz][qy][qx] = D3 * Qx + D4 * Qy + D5 * Qz;
r0[2][0][qz][qy][qx] = D6 * Qx + D7 * Qy + D8 * Qz;
}
}
}
Expand Down
12 changes: 6 additions & 6 deletions tests/unit/fem/test_pa_kernels.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -449,12 +449,12 @@ TEST_CASE("PA Vector Mass", "[PartialAssembly], [VectorPA], [MassPA], [CUDA]")
== MFEM_Approx(0.0));
}

// SECTION("3D")
// {
// dbg("Vector Mass test 3D");
// REQUIRE(test_vector_pa_integrator<VectorMassIntegrator>(3, true)
// == MFEM_Approx(0.0));
// }
SECTION("3D")
{
dbg("Vector Mass test 3D");
REQUIRE(test_vector_pa_integrator<VectorMassIntegrator>(3, true)
== MFEM_Approx(0.0));
}
}

TEST_CASE("PA Vector Diffusion",
Expand Down