add example5 - simple stencil2d

Author: haichangsi

.github/workflows/docker_workflow.yml

@@ -47,6 +47,7 @@ jobs:
           cmake --build build -j
       - name: Run examples
         run: |
-          mpirun -n 2 ./build/src/example1
-          mpirun -n 2 ./build/src/example2
-          mpirun -n 2 ./build/src/example3
+          mpirun -n 3 ./build/src/example1
+          mpirun -n 3 ./build/src/example2
+          mpirun -n 3 ./build/src/example3
+          mpirun -n 3 ./build/src/example5

scripts/build_run.sh

@@ -13,3 +13,4 @@ cmake --build build -j
 mpirun -n 2 ./build/src/example1
 mpirun -n 2 ./build/src/example2
 mpirun -n 2 ./build/src/example3
+mpirun -n 2 ./build/src/example5

src/CMakeLists.txt

@@ -31,3 +31,8 @@ add_executable(example4 example4.cpp)
 
 target_compile_definitions(example4 INTERFACE DR_FORMAT)
 target_link_libraries(example4 DR::mpi fmt::fmt)
+
+add_executable(example5 example5.cpp)
+
+target_compile_definitions(example5 INTERFACE DR_FORMAT)
+target_link_libraries(example5 DR::mpi fmt::fmt)

src/example5.cpp

@@ -0,0 +1,53 @@
+// SPDX-FileCopyrightText: Intel Corporation
+//
+// SPDX-License-Identifier: BSD-3-Clause
+
+#include <dr/mhp.hpp>
+#include <fmt/core.h>
+
+namespace mhp = dr::mhp;
+
+using T = uint16_t;
+using MDA = dr::mhp::distributed_mdarray<T, 2>;
+
+int main() {
+#ifdef SYCL_LANGUAGE_VERSION
+  mhp::init(sycl::default_selector_v);
+#else
+  mhp::init();
+#endif
+  std::size_t arr_size = 4;
+  std::size_t radius = 1;
+  std::array slice_starts{radius, radius};
+  std::array slice_ends{arr_size - radius, arr_size - radius};
+
+  auto dist = dr::mhp::distribution().halo(radius);
+  MDA a({arr_size, arr_size}, dist);
+  MDA b({arr_size, arr_size}, dist);
+  mhp::iota(a, 1);
+  mhp::iota(b, 1);
+
+  auto in = dr::mhp::views::submdspan(a.view(), slice_starts, slice_ends);
+  auto out = dr::mhp::views::submdspan(b.view(), slice_starts, slice_ends);
+  auto in_array = &a;
+  auto out_array = &b;
+
+  auto mdspan_stencil_op = [](auto &&v) {
+    auto [in, out] = v;
+    out(0, 0) = (in(-1, 0) + in(0, -1) + in(0, 0) + in(0, 1) + in(1, 0)) / 4;
+  };
+
+  mhp::halo(*in_array).exchange();
+  mhp::stencil_for_each(mdspan_stencil_op, in, out);
+
+  if (mhp::rank() == 0) {
+    fmt::print("a: \n{} \n", a.mdspan());
+    fmt::print("b: \n{} \n", b.mdspan());
+    fmt::print("in: \n{} \n", in.mdspan());
+    fmt::print("out: \n{} \n", out.mdspan());
+  }
+
+  mhp::finalize();
+
+  return 0;
+}