def compute(self):
collective.allreduce(self.send, "177")
return self.send
def do_allreduce(self, group_name="default", op=ReduceOp.SUM):
col.allreduce(self.buffer, group_name, op)
return self.buffer
def compute(self):
collective.allreduce(self.send, "default")
print(self.send)
return self.send