import mindspore.ops as ops
import mindspore.nn as nn
from mindspore.communication import init
from mindspore import Tensor
import numpy as np

init()
class Net(nn.Cell):
    def __init__(self):
        super(Net, self).__init__()
        self.depend = ops.Depend()
        self.send = ops.Send(st_tag=0, dest_rank=8, group="hccl_world_group")

    def construct(self, x):
        out = self.depend(x, self.send(x))
        return out

input_ = Tensor(np.ones([2, 8]).astype(np.float32))
net = Net()
output = net(input_)