Function megray_send_tensor

imperative/src/impl/ops/io_remote.cpp:68–79

}

void megray_send_tensor(
        std::shared_ptr<MegRay::Communicator> megray_comm, const TensorPtr& src,
        uint32_t rank_to) {
    auto&& tensor = src->dev_tensor();
    auto&& ishp = src->shape();
    size_t data_size = ishp.total_nr_elems();
    auto megray_ctx = mgb::opr::get_megray_context(src->comp_node());
    auto status = megray_comm->send(
            src->dev_tensor().raw_ptr(), data_size,
            mgb::opr::get_megray_dtype(src->layout().dtype), rank_to, megray_ctx);
    mgb_assert(status == MegRay::MEGRAY_OK, "MegRay send failed");
}

TensorLayout create_layout(const std::vector<int32_t>& shape, DType dtype) {
    TensorShape tshape;