Define transfer type width in a single place

fmihpc · Dec 2, 2024 · 484438c · 484438c
1 parent 85abe09
commit 484438c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/spatial_cell_cpu.cpp b/spatial_cell_cpu.cpp
@@ -773,12 +773,12 @@ namespace spatial_cell {
             displacements.size(),
             block_lengths.data(),
             displacements.data(),
-            MPI_UINT32_T,
+            transferTypeMPI,
             &datatype
          );
       } else {
          count = 0;
-         datatype = MPI_UINT32_T;
+         datatype = transferTypeMPI;
       }
 
       const bool printMpiDatatype = false;

diff --git a/spatial_cell_cpu.hpp b/spatial_cell_cpu.hpp
@@ -172,6 +172,9 @@ namespace spatial_cell {
       vmesh::VelocityBlockContainer<vmesh::LocalID> blockContainer;  /**< Velocity block data.*/
    };
 
+   typedef uint8_t transferType;
+   constexpr MPI_Datatype transferTypeMPI MPI_BYTE;
+
    class SpatialCell {
    public:
       SpatialCell();
@@ -364,7 +367,7 @@ namespace spatial_cell {
          // if (sizeof(T) < 4) {
          //    abort_mpi("Pointer too small to cast to uint32_t!");
          // }
-         return reinterpret_cast<const uint8_t*>(p) - reinterpret_cast<const uint8_t*>(this);
+         return reinterpret_cast<const transferType*>(p) - reinterpret_cast<const transferType*>(this);
       }
 
       template<typename T>
@@ -373,7 +376,7 @@ namespace spatial_cell {
          // if (sizeof(T) < 4) {
          //    abort_mpi("Pointer too small to cast to uint32_t!");
          // }
-         return len * sizeof(T) / sizeof(uint8_t);
+         return len * sizeof(T) / sizeof(transferType);
       }
 
       bool compute_block_has_content(const vmesh::GlobalID& block,const uint popID) const;