add example4 - dmdarray adding

Author: haichangsi

src/CMakeLists.txt

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

src/example4.cpp

@@ -0,0 +1,39 @@
+// SPDX-FileCopyrightText: Intel Corporation
+//
+// SPDX-License-Identifier: BSD-3-Clause
+
+#include <dr/mhp.hpp>
+#include <fmt/core.h>
+
+namespace mhp = dr::mhp;
+using T = int;
+
+int main(int argc, char **argv) {
+
+  mhp::init(sycl::default_selector_v);
+  std::size_t xdim = 9, ydim = 5;
+
+  std::array<std::size_t, 2> extents2d = {xdim, ydim};
+
+  mhp::distributed_mdarray<T, 2> a(extents2d);
+  mhp::distributed_mdarray<T, 2> b(extents2d);
+  mhp::distributed_mdarray<T, 2> c(extents2d);
+  mhp::iota(a, 100);
+  mhp::iota(b, 200);
+
+  auto copy_op = [](auto v) {
+    auto [in1, in2, out] = v;
+    out = in1 + in2;
+  };
+  mhp::for_each(copy_op, a, b, c);
+
+  if (mhp::rank() == 0) {
+    fmt::print("A:\n{}\n", a.mdspan());
+    fmt::print("B:\n{}\n", b.mdspan());
+    fmt::print("C:\n{}\n", c.mdspan());
+  }
+
+  mhp::finalize();
+
+  return 0;
+}