Update

Author: cbritopacheco

examples/PETSc/MPI_Poisson.cpp

@@ -56,6 +56,34 @@ int main(int argc, char** argv)
     gf.save("Poisson." + std::to_string(world.rank()) + ".gf");
   }
 
+  // Vec v;
+  // VecCreate(PETSC_COMM_WORLD, &v);
+
+  // size_t sz;
+  // if (world.rank() == 0)
+  // {
+  //   sz = 7;
+  // }
+  // else
+  // {
+  //   sz = 8;
+  // }
+  // VecSetSizes(v, sz, 10);
+  // VecSetFromOptions(v);
+
+  // std::vector<int> ghosts;
+  // if (world.rank() == 0)
+  // {
+  //   ghosts = { 2, 3 };
+  // }
+  // else
+  // {
+  //   ghosts = { 4, 5, 6 };
+  // }
+
+  // VecMPISetGhost(v, ghosts.size(), ghosts.data());
+
+
   PetscFinalize();
 }
 

src/Rodin/MPI/Variational/P1/P1.h

@@ -3,6 +3,7 @@
 
 #include <mpi.h>
 #include <boost/serialization/optional.hpp>
+#include <sys/mman.h>
 
 #include "Rodin/MPI/Geometry/Mesh.h"
 #include "Rodin/MPI/Variational/FiniteElementSpace.h"
@@ -179,7 +180,7 @@ namespace Rodin::Variational
             const auto local = mesh.getLocalIndex(0, requested[i].first);
             assert(local);
             const Index& global = requested[i].second;
-            m_loc2glob.right.insert({ global, *local});
+            m_loc2glob.right.insert({ global, *local });
           }
         }
 

src/Rodin/PETSc/Variational/GridFunction.h

@@ -90,10 +90,13 @@ namespace Rodin::Variational
         {
           ierr = VecCreate(PETSC_COMM_SELF, &data);
           assert(ierr == PETSC_SUCCESS);
+
           ierr = VecSetSizes(data, fes.getSize(), PETSC_DECIDE);
           assert(ierr == PETSC_SUCCESS);
+
           ierr = VecSetFromOptions(data);
           assert(ierr == PETSC_SUCCESS);
+
           ierr = VecZeroEntries(data);
           assert(ierr == PETSC_SUCCESS);
         }
@@ -239,19 +242,10 @@ for (int r = 0; r < comm.size(); ++r) {
         }
         else if constexpr (std::is_same_v<FESMeshContextType, Context::MPI>)
         {
-          Index local;
-          if (m_begin <= global && global < m_end)
-          {
-            local = global - m_begin;
-          }
-          else
-          {
-            const auto& fes = this->getFiniteElementSpace();
-            const Optional<Index> localIdx = fes.getLocalIndex(global);
-            assert(localIdx);
-            local = *localIdx;
-          }
-          return m_write.raw[local];
+          const auto& fes = this->getFiniteElementSpace();
+          const Optional<Index> localIdx = fes.getLocalIndex(global);
+          assert(localIdx);
+          return m_write.raw[*localIdx];
         }
         else
         {
@@ -431,7 +425,8 @@ for (int r = 0; r < comm.size(); ++r) {
       }
 
       template <class NestedDerived>
-      GridFunction& projectOnCells(const FunctionBase<NestedDerived>& fn, const FlatSet<Geometry::Attribute>& attrs)
+      GridFunction& projectOnCells(
+          const FunctionBase<NestedDerived>& fn, const FlatSet<Geometry::Attribute>& attrs)
       {
         if constexpr (std::is_same_v<FESMeshContextType, Context::Local>)
         {
@@ -489,7 +484,8 @@ for (int r = 0; r < comm.size(); ++r) {
             shard.getMapping({ d, i }, fn.template cast<RangeType>());
           for (Index local = 0; local < fe.getCount(); local++)
           {
-            const Index global = fes.getGlobalIndex(shard.getGlobalIndex({ d, i }, local));
+            const Index global =
+              fes.getGlobalIndex(shard.getGlobalIndex({ d, i }, local));
             this->operator[](global) = fe.getLinearForm(local)(mapping);
           }
         }