| /third_party/boost/boost/mpi/collectives/ |
| D | all_gatherv.hpp | 36 T* out_values, int const* sizes, int const* displs, mpl::true_) in all_gatherv_impl() argument
39 scoped_array<int> new_offsets_mem(make_offsets(comm, sizes, displs, -1)); in all_gatherv_impl()
40 if (new_offsets_mem) displs = new_offsets_mem.get(); in all_gatherv_impl()
46 const_cast<int*>(displs), in all_gatherv_impl()
57 T* out_values, int const* sizes, int const* displs, in all_gatherv_impl() argument
61 scoped_array<int> skipped(make_skipped_slots(comm, sizes, displs)); in all_gatherv_impl()
104 const std::vector<int>& sizes, const std::vector<int>& displs) in all_gatherv() argument
108 assert(displs.size() == comm.size()); in all_gatherv()
110 c_data(sizes), c_data(displs), is_mpi_datatype<T>()); in all_gatherv()
116 const std::vector<int>& sizes, const std::vector<int>& displs) in all_gatherv() argument
[all …]
|
| D | scatterv.hpp | 29 const int* sizes, const int* displs, int root, mpl::true_) in scatterv_impl() argument
34 scoped_array<int> new_offsets_mem(make_offsets(comm, sizes, displs, root)); in scatterv_impl()
35 if (new_offsets_mem) displs = new_offsets_mem.get(); in scatterv_impl()
39 const_cast<int*>(displs), type, in scatterv_impl()
63 int const* sizes, int const* displs, int root, mpl::false_) in scatterv_impl() argument
73 if (displs) { in scatterv_impl()
75 offsets2skipped(sizes, displs, c_data(skipped), nproc); in scatterv_impl()
76 displs = c_data(skipped); in scatterv_impl()
100 const std::vector<int>& sizes, const std::vector<int>& displs, in scatterv() argument
104 detail::scatterv_impl(comm, in_values, out_values, out_size, c_data(sizes), c_data(displs), in scatterv()
[all …]
|
| D | gatherv.hpp | 32 T* out_values, const int* sizes, const int* displs, int root, mpl::true_) in gatherv_impl() argument
37 out_values, const_cast<int*>(sizes), const_cast<int*>(displs), in gatherv_impl()
62 T* out_values, const int* sizes, const int* displs, int root, mpl::false_) in gatherv_impl() argument
65 scoped_array<int> skipped(make_skipped_slots(comm, sizes, displs, root)); in gatherv_impl()
85 T* out_values, const std::vector<int>& sizes, const std::vector<int>& displs, in gatherv() argument
90 out_values, detail::c_data(sizes), detail::c_data(displs), in gatherv()
99 T* out_values, const std::vector<int>& sizes, const std::vector<int>& displs, in gatherv() argument
102 …::mpi::gatherv(comm, detail::c_data(in_values), in_values.size(), out_values, sizes, displs, root); in gatherv()
129 std::vector<int> displs( nprocs ); in gatherv() local
131 displs[rank] = aux; in gatherv()
[all …]
|
| /third_party/boost/libs/mpi/src/ |
| D | offsets.cpp | 54 make_offsets(communicator const& comm, int const* sizes, int const* displs, int root) in make_offsets() argument
58 if (!displs) { in make_offsets()
61 displs = offsets; in make_offsets()
78 make_skipped_slots(communicator const& comm, int const* sizes, int const* displs, int root) in make_skipped_slots() argument
82 if (displs) { in make_skipped_slots()
85 std::copy(displs, displs+nproc, skipped); in make_skipped_slots()
86 offsets2skipped(sizes, displs, skipped, nproc); in make_skipped_slots()
|
| /third_party/boost/boost/mpi/detail/ |
| D | offsets.hpp | 36 int* make_offsets(communicator const& comm, int const* sizes, int const* displs, int root = -1);
43 int* make_skipped_slots(communicator const& comm, int const* sizes, int const* displs, int root = -…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | all_to_all_v_calc_param.cc | 76 std::vector<int64_t> *displs) { in CalcMemOffset() argument
95 (*displs)[LongToSize(rank_ids[i])] = SizeToLong(mem_offset[i]); in CalcMemOffset()
110 (*displs)[i] = SizeToLong(offset); in CalcMemOffset()
|
| D | all_to_all_v_calc_param.h | 38 … const std::string &rank_ids_attr, std::vector<int64_t> *counts, std::vector<int64_t> *displs);
|
| /third_party/boost/boost/mpi/ |
| D | collectives.hpp | 364 T* out_values, const std::vector<int>& sizes, const std::vector<int>& displs,
373 T* out_values, const std::vector<int>& sizes, const std::vector<int>& displs,
504 const std::vector<int>& sizes, const std::vector<int>& displs,
513 const std::vector<int>& sizes, const std::vector<int>& displs,
|
| /third_party/boost/libs/mpi/test/ |
| D | scatter_test.cpp | 177 std::vector<int> displs(comm.size()); in scatterd_test() local
193 displs[p] = shift; in scatterd_test()
201 scatterv(comm, values, sizes, displs, &(myvalues[0]), mysize, root); in scatterd_test()
|