adjust usages of read_distributed

ginkgo-project · Apr 30, 2024 · 26678b3 · 26678b3
1 parent 6f0ed3a
commit 26678b3
Show file tree

Hide file tree

Showing 6 changed files with 111 additions and 200 deletions.
diff --git a/benchmark/utils/generator.hpp b/benchmark/utils/generator.hpp
@@ -223,9 +223,11 @@ struct DistributedDefaultSystemGenerator {
         const gko::matrix_data<value_type, index_type>& data,
         json* spmv_case = nullptr) const
     {
-        auto part = gko::experimental::distributed::
-            Partition<itype, global_itype>::build_from_global_size_uniform(
-                exec, comm.size(), static_cast<global_itype>(data.size[0]));
+        auto part = gko::share(
+            gko::experimental::distributed::Partition<itype, global_itype>::
+                build_from_global_size_uniform(
+                    exec, comm.size(),
+                    static_cast<global_itype>(data.size[0])));
         auto formats = split(format_name, '-');
         if (formats.size() != 2) {
             throw std::runtime_error{"Invalid distributed format specifier " +